class TestSubprocess(unittest.TestCase):
def setUp(self):
self.env = EnvironmentVarGuard()
def test_patch_container(self):
_subprocess.patch()
self.assertTrue(hasattr(subprocess, '_check_output'))
def test_unpatch_container(self):
_subprocess.unpatch()
self.assertFalse(hasattr(subprocess, '_check_output'))
def test_check_output_no_container_patched(self):
self.env.unset('CONTAINER_ID')
_subprocess.patch()
subprocess._check_output = Mock()
subprocess.check_output = Mock()
_subprocess.check_output(['ls', '/tmp'])
subprocess._check_output.assert_called_with(['ls', '/tmp'])
subprocess.check_output.assert_not_called()
def test_check_output_no_container_not_patched(self):
self.env.unset('CONTAINER_ID')
_subprocess.unpatch()
subprocess.check_output = Mock()
_subprocess.check_output(['ls', '/tmp'])
subprocess.check_output.assert_called_with(['ls', '/tmp'])
def AWSEnvironmentVarGuard():
env = EnvironmentVarGuard()
env.unset('AWS_ACCESS_KEY_ID')
env.unset('AWS_ACCESS_KEY')
env.unset('AWS_SECRET_ACCESS_KEY')
env.unset('AWS_SECRET')
env.unset('AWS_DEFAULT_REGION')
env.set('BODYLABS_CREDENTIAL_FILE', '/this_file_does_not_exist')
env.set('AWS_CONFIG_FILE', '/this_file_does_not_exist')
env.set('AWS_CREDENTIAL_FILE', '/this_file_does_not_exist')
return env
def test_environment_api_key_not_set_error(self):
def real_func():
self.fail()
env = EnvironmentVarGuard()
env.unset('CLOUDFLARE_API_KEY')
with env:
try:
bulk_dns.configured(real_func)()
self.fail()
except ValueError as e:
self.assertTrue('CLOUDFLARE_API_KEY' in e.message)
except:
self.fail()
class TestDockerHelpers(unittest.TestCase):
def setUp(self):
self.env = EnvironmentVarGuard()
def test_is_container_when_container_id(self):
self.env.set('CONTAINER_ID', CONTAINER_ID)
self.assertTrue(container.is_container())
def test_not_container_when_no_container_id(self):
self.env.unset('CONTAINER_ID')
self.assertFalse(container.is_container())
def test_container_id(self):
self.env.set('CONTAINER_ID', CONTAINER_ID)
self.assertEqual(CONTAINER_ID, container.get_container_id())
def test_container_ip(self):
self.env.set('CONTAINER_IP', CONTAINER_IP)
self.assertEqual(CONTAINER_IP, container.get_container_ip())
