def test_strict_exception(self, get):
get.side_effect = requests.exceptions.RequestException()
self._setup_mock(get)
ns = self._get_ns(['local-ipv4'], strict=True)
with self.assertRaises(requests.exceptions.RequestException):
ec2_meta_env.get_environ({}, ns)
self.assertEqual(get.call_count, 1)
def test_unstrict_exception(self, get):
get.side_effect = requests.exceptions.RequestException()
self._setup_mock(get)
ns = self._get_ns(['local-ipv4'], strict=False)
env = ec2_meta_env.get_environ({}, ns)
self.assertEqual(get.call_count, 1)
self.assertFalse('EC2_LOCAL_IPV4' in env)
def test_dont_override(self, get):
base_env = {'EC2_LOCAL_IPV4': 'original-data'}
self._setup_mock(get)
ns = self._get_ns(['local-ipv4'], override=False)
env = ec2_meta_env.get_environ(base_env, ns)
self.assertEqual(get.call_count, 1)
self.assertEqual(env['EC2_LOCAL_IPV4'], 'original-data')
def test_set_one_var(self, get):
test_value = self._setup_mock(get)
ns = self._get_ns(['local-ipv4'])
env = ec2_meta_env.get_environ({}, ns)
self.assertEqual(get.call_count, 1)
self.assertTrue('EC2_LOCAL_IPV4' in env)
self.assertEqual(env['EC2_LOCAL_IPV4'], test_value)
def test_set_many_var(self, get):
test_value = self._setup_mock(get)
ns = self._get_ns(['local-ipv4', 'local-hostname'])
env = ec2_meta_env.get_environ({}, ns)
self.assertEqual(get.call_count, 2)
self.assertTrue('EC2_LOCAL_IPV4' in env)
self.assertTrue('EC2_LOCAL_HOSTNAME' in env)
self.assertEqual(env['EC2_LOCAL_IPV4'], test_value)
self.assertEqual(env['EC2_LOCAL_HOSTNAME'], test_value)