X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/d77bf0c67422a259afacc17660698729328a1ed3..ffc46b6c2fecf585f80c5e4513a088ad21a2185b:/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