def test_write_clear(self): os.environ['TEST_ALREADY_SET'] = 'myvar' envitro.write('TEST_ALREADY_SET', None) self.assertEqual(os.environ.get('TEST_ALREADY_SET'), None) if 'TEST_ALREADY_SET_MISSING' in os.environ: del os.environ['TEST_ALREADY_SET_MISSING'] envitro.write('TEST_ALREADY_SET_MISSING', None) self.assertEqual(os.environ.get('TEST_ALREADY_SET_MISSING'), None)
def test_set_none(self): envitro.write('ENV_VAL', None) @envitro.decorators.write('ENV_VAL', 'myval') def myfunc(): self.assertEqual(os.environ['ENV_VAL'], 'myval') myfunc() with self.assertRaises(KeyError): os.environ['ENV_VAL']
def test_write(self): envitro.write('TEST_SET', 'setvar') self.assertEqual(os.environ['TEST_SET'], 'setvar') envitro.write('TEST_SET_SPACES', ' spacesvar ') self.assertEqual(os.environ['TEST_SET_SPACES'], ' spacesvar ') envitro.write('TEST_SET_INT', 123) self.assertEqual(os.environ['TEST_SET_INT'], '123') envitro.write('TEST_SET_BOOL', True) self.assertEqual(os.environ['TEST_SET_BOOL'], 'True')
def test_invalid_bool(self): envitro.write('INVALID_BOOL', 'nope') with self.assertRaises(ValueError): envitro.bool('INVALID_BOOL')
def test_port(self): envitro.write('DB_PORT', 'tcp://172.17.0.82:5432') self.assertEqual(envitro.docker.port('DB'), 5432)
def test_protocol(self): envitro.write('DB_PORT', 'tcp://172.17.0.82:5432') self.assertEqual(envitro.docker.protocol('DB'), 'tcp')
def test_read(self): envitro.write('DB_PORT', 'tcp://172.17.0.82:5432') self.assertEqual(envitro.docker.read('DB'), 'tcp://172.17.0.82:5432') envitro.write('NODB_PORT', None) self.assertEqual(envitro.docker.read('NODB', allow_none=True), None)
def test_isset_invalid_looking_port(self, mock_warn): envitro.write('BADPORT_PORT', 'tcp://172.17.0.82:notaport') self.assertFalse(envitro.docker.isset('BADPORT')) self.assertTrue(mock_warn.called)
def test_isset_invalid_looking(self, mock_warn): envitro.write('INVALIDDB_PORT', 'nodockerlinkhere') self.assertFalse(envitro.docker.isset('INVALIDDB')) self.assertTrue(mock_warn.called)
def test_isset(self): envitro.write('DB_PORT', 'tcp://172.17.0.82:5432') self.assertTrue(envitro.docker.isset('DB')) envitro.write('NODB_PORT', None) self.assertFalse(envitro.docker.isset('NODB'))