def env_var(self, name):
"""
Wrapper around env_var utility function with supplied current environ.
:param name: name of the environment variable
:return: value of the environment variable or None if it doesn't exist
"""
return env_var(self.environment, name)
def test_os_environ_valid_var(self):
key = 'KEY'
value = 'VALUE'
os.environ['{}{}'.format(ENV_BASE_PREFIX, key)] = value
self.assertEqual(env_var(os.environ, key), value)
def test_dict_environ_invalid_without_prefix(self):
environ = {}
key = 'KEY'
value = 'VALUE'
environ[key] = value
self.assertIsNone(env_var(os.environ, key))
def test_invalid_environ_raises_exception(self):
key = 'KEY'
with self.assertRaises(ImproperlyConfigured):
env_var([], key)
def test_dict_environ_valid_var(self):
environ = {}
key = 'KEY'
value = 'VALUE'
environ['{}{}'.format(ENV_BASE_PREFIX, key)] = value
self.assertEqual(env_var(environ, key), value)