diff options
Diffstat (limited to 'packages/python-google-compute-engine/google_compute_engine/metadata_scripts/tests/script_retriever_test.py')
-rw-r--r-- | packages/python-google-compute-engine/google_compute_engine/metadata_scripts/tests/script_retriever_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/packages/python-google-compute-engine/google_compute_engine/metadata_scripts/tests/script_retriever_test.py b/packages/python-google-compute-engine/google_compute_engine/metadata_scripts/tests/script_retriever_test.py index 0df8830..558b0b0 100644 --- a/packages/python-google-compute-engine/google_compute_engine/metadata_scripts/tests/script_retriever_test.py +++ b/packages/python-google-compute-engine/google_compute_engine/metadata_scripts/tests/script_retriever_test.py @@ -89,7 +89,7 @@ class ScriptRetrieverTest(unittest.TestCase): # GetMetadata includes a prefix, so remove it. stripped_url = token_url.replace(metadata_prefix, '') mock_get_metadata.assert_called_once_with( - stripped_url, recursive=False, retry=False) + stripped_url, recursive=False, retry_limit=3) self.assertEqual(self.retriever.token, 'foo bar') @@ -119,7 +119,7 @@ class ScriptRetrieverTest(unittest.TestCase): prefix = 'http://metadata.google.internal/computeMetadata/v1/' stripped_url = token_url.replace(prefix, '') mock_get_metadata.assert_called_once_with( - stripped_url, recursive=False, retry=False) + stripped_url, recursive=False, retry_limit=3) mock_download_url.assert_called_once_with(auth_url, self.dest_dir) self.assertIsNone(self.retriever.token) |