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())