X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/de9e183ceac9e5dc53cd4fbd089b752a75a4d5cf..c56098048193b9ba8597b55c761af07a7617c026:/services/fuse/tests/test_retry.py diff --git a/services/fuse/tests/test_retry.py b/services/fuse/tests/test_retry.py index b46ba7839f..b69707af4f 100644 --- a/services/fuse/tests/test_retry.py +++ b/services/fuse/tests/test_retry.py @@ -1,11 +1,18 @@ +# Copyright (C) The Arvados Authors. All rights reserved. +# +# SPDX-License-Identifier: AGPL-3.0 + +from __future__ import absolute_import +from future import standard_library +standard_library.install_aliases() import arvados import arvados_fuse.command import json import mock import os import pycurl -import Queue -import run_test_server +import queue +from . import run_test_server import tempfile import unittest @@ -46,11 +53,11 @@ class RetryPUT(IntegrationTest): def test_retry_write(self, sleep): mockedCurl = mock.Mock(spec=pycurl.Curl(), wraps=pycurl.Curl()) mockedCurl.perform.side_effect = Exception('mock error (ok)') - q = Queue.Queue() + q = queue.Queue() q.put(mockedCurl) q.put(pycurl.Curl()) q.put(pycurl.Curl()) - with mock.patch('arvados.keep.KeepClient.KeepService._get_user_agent', side_effect=lambda: q.get(block=None)): + with mock.patch('arvados.keep.KeepClient.KeepService._get_user_agent', side_effect=q.get_nowait): self.pool_test(os.path.join(self.mnt, 'zzz')) self.assertTrue(mockedCurl.perform.called) @staticmethod