def test_read_none(self):
if 'TEST_DEFAULT_GET_NONE' in os.environ:
del os.environ['TEST_DEFAULT_GET_NONE']
self.assertEqual(envitro.read('TEST_DEFAULT_GET_NONE', allow_none=True), None)
if 'TEST_DEFAULT_GET_NONE_DEFAULT' in os.environ:
del os.environ['TEST_DEFAULT_GET_NONE_DEFAULT']
self.assertEqual(
envitro.read('TEST_DEFAULT_GET_NONE_DEFAULT', default='defaultval', allow_none=True), 'defaultval')
def test_read(self):
os.environ['TEST_GET'] = 'getvar'
self.assertEqual(envitro.read('TEST_GET'), 'getvar')
def test_invalid_read(self):
if 'TEST_INVALID_GET' in os.environ:
del os.environ['TEST_INVALID_GET']
self.assertRaises(KeyError, lambda: envitro.read('TEST_INVALID_GET'))
def test_read_default(self):
if 'TEST_DEFAULT_GET' in os.environ:
del os.environ['TEST_DEFAULT_GET']
self.assertEqual(envitro.read('TEST_DEFAULT_GET', 'defaultval'), 'defaultval')