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)