diff options
Diffstat (limited to 'tests/unit/utils/test_utils.py')
| -rw-r--r-- | tests/unit/utils/test_utils.py | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/tests/unit/utils/test_utils.py b/tests/unit/utils/test_utils.py index c170506a..d96978c1 100644 --- a/tests/unit/utils/test_utils.py +++ b/tests/unit/utils/test_utils.py @@ -272,7 +272,20 @@ class TestLazyLoadMetadata(unittest.TestCase): boto.utils.retry_url.assert_called_with( 'http://169.254.169.254/latest/user-data', - retry_on_404=False) + retry_on_404=False, + num_retries=5, timeout=None) + + def test_user_data_timeout(self): + self.set_normal_response(['foo']) + + userdata = get_instance_userdata(timeout=1, num_retries=2) + + self.assertEqual('foo', userdata) + + boto.utils.retry_url.assert_called_with( + 'http://169.254.169.254/latest/user-data', + retry_on_404=False, + num_retries=2, timeout=1) class TestStringToDatetimeParsing(unittest.TestCase): |