summaryrefslogtreecommitdiff
path: root/packages/python-google-compute-engine/google_compute_engine/metadata_scripts/tests/script_retriever_test.py
diff options
context:
space:
mode:
authorZach Marano <zmarano@google.com>2019-11-13 15:55:28 -0800
committerZach Marano <zmarano@google.com>2019-11-13 15:55:28 -0800
commitbe2c905a643f4c14340b8efa182cb47726d3026c (patch)
treedaea282c8c6b1b3ee99a58a00408b852fd4bb838 /packages/python-google-compute-engine/google_compute_engine/metadata_scripts/tests/script_retriever_test.py
parent89740a3d710140a7145e6cd54910a23f7b71f6fc (diff)
parent5d69a336f71498f33887dce362c0fe2f19bc73a5 (diff)
downloadgoogle-compute-image-packages-be2c905a643f4c14340b8efa182cb47726d3026c.tar.gz
Merge branch 'development'v20191113
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.py4
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..4019bbe 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, retries=1)
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, retries=1)
mock_download_url.assert_called_once_with(auth_url, self.dest_dir)
self.assertIsNone(self.retriever.token)