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