X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/e61df48303291900581ef4a64dcf97864598c5f4..9d706659aa362728270f02b7d18d12b1f68f3e36:/sdk/python/tests/test_cache.py diff --git a/sdk/python/tests/test_cache.py b/sdk/python/tests/test_cache.py index ea34c8051d..259acd0a30 100644 --- a/sdk/python/tests/test_cache.py +++ b/sdk/python/tests/test_cache.py @@ -1,9 +1,13 @@ +# Copyright (C) The Arvados Authors. All rights reserved. +# +# SPDX-License-Identifier: Apache-2.0 + from __future__ import print_function from __future__ import absolute_import from builtins import str from builtins import range -import md5 +import hashlib import mock import os import random @@ -34,15 +38,16 @@ class CacheTestThread(threading.Thread): for x in range(16): try: data_in = _random(128) - data_in = md5.new(data_in).hexdigest() + "\n" + str(data_in) + data_in = hashlib.md5(data_in).hexdigest().encode() + b"\n" + data_in c.set(url, data_in) data_out = c.get(url) - digest, content = data_out.split("\n", 1) - if digest != md5.new(content).hexdigest(): + digest, _, content = data_out.partition(b"\n") + if digest != hashlib.md5(content).hexdigest().encode(): self.ok = False except Exception as err: self.ok = False - print("cache failed: {}".format(err), file=sys.stderr) + print("cache failed: {}: {}".format(type(err), err), file=sys.stderr) + raise class CacheTest(unittest.TestCase):