Как выполнить модульное тестирование декоратора, который использует переменную окружения?

Question

Я создал декоратор, чтобы функции можно запускать только в определенных условиях:

def accepted_environments(*envs): 
    """ 
    The decorated function can be executed only in specified envs 
    """ 
    def my_decorator(func_to_be_decorated): 
     def wrapper(): 
      if settings.ENV_NAME not in envs: 
       raise EnvironmentException 
      return func_to_be_decorated() 
     return wrapper 
    return my_decorator 

# Usage example 
@accepted_environments('local', 'prod') 
def hello(): 
    print("hello") 

Это похоже на работу, но я хотел бы модульное тестирование его. Проблема в том, что мои тесты потенциально выполняются в каждой среде (локальная, промежуточная, prod). Плюс, я думаю, что это не безопасно, мои тесты могут изменять переменные среды.

Должен ли я «издеваться» над этим поведением? Как бы вы поступили? Благодаря!

Answer 1

Используйте mock, чтобы переопределить значение settings.ENV_NAME для испытания.

def test_not_in_dev(self): 
    with mock.patch.dict(settings.__dict__, ENV_NAME="dev"): 
     self.assertRaises(EnvironmentException, hello)