diff --git a/tests/test_shell.py b/tests/test_shell.py index b6f74f37..5708b973 100644 --- a/tests/test_shell.py +++ b/tests/test_shell.py @@ -185,6 +185,7 @@ class TestShell(unittest.TestCase): [None, [{'name': 'object'}]], [None, []], ] + connection.return_value.attempts = 0 argv = ["", "download", "container"] swiftclient.shell.main(argv) @@ -202,6 +203,7 @@ class TestShell(unittest.TestCase): def test_upload(self, connection, listdir): connection.return_value.head_object.return_value = { 'content-length': '0'} + connection.return_value.attempts = 0 argv = ["", "upload", "container", self.tmpfile] swiftclient.shell.main(argv) connection.return_value.put_object.assert_called_with( @@ -245,6 +247,7 @@ class TestShell(unittest.TestCase): [None, [{'name': 'object'}]], [None, []], ] + connection.return_value.attempts = 0 argv = ["", "delete", "--all"] connection.return_value.head_object.return_value = {} swiftclient.shell.main(argv) @@ -259,6 +262,7 @@ class TestShell(unittest.TestCase): [None, [{'name': 'object'}]], [None, []], ] + connection.return_value.attempts = 0 argv = ["", "delete", "container"] connection.return_value.head_object.return_value = {} swiftclient.shell.main(argv) @@ -271,6 +275,7 @@ class TestShell(unittest.TestCase): def test_delete_object(self, connection): argv = ["", "delete", "container", "object"] connection.return_value.head_object.return_value = {} + connection.return_value.attempts = 0 swiftclient.shell.main(argv) connection.return_value.delete_object.assert_called_with( 'container', 'object', query_string=None)