Merge "Fixed several shell tests on Python3."

1 files changed, 5 insertions, 0 deletions

diff --git a/tests/test_shell.py b/tests/test_shell.py
index b6f74f3..5708b97 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)