環境変数を使用するデコレータの単体テスト方法は？

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

が、私はユニットそれをテストしたいと思います。 問題は、私のテストはすべての環境（ローカル、ステージング、プロード）で実行可能であることです。さらに、私のテストでは環境変数を変更することは安全ではないと思います。

この動作を "模擬する"べきですか？どうしますか？ありがとう！

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)