side_effect가있는 Python 패치 객체

Question

Mock 객체가 지정된 입력을 기반으로 특정 값을 반환하도록하려고합니다. 나는 SO에서 몇 가지 예를 찾아 보았고 어떤 이유로 든 여전히 작동시킬 수 없습니다. 여기 내가 지금 가지고있는 것이있다.

class EmailChecker(): 
    def is_email_correct(email): 
     some regex to determine if email is valid, returns either True or False 

def my_side_effect(**args): 
    if args[0] == '1': 
     return True 
    else: 
     return False 

myMock = mock.patch.object(EmailChecker, 'is_email_correct', side_effect=my_side_effect) 

아직 작동하지 않는 것 같습니다. 내가 여기서 무엇을 놓치고 있니? 다음 코드에서와 같이 장식 또는 컨텍스트 매니저로

Answer 1

사용 patch.object :

>>> class EmailChecker(): 
...  def is_email_correct(self, email): 
...   pass 
... 
>>> def my_side_effect(*args): 
...  if args[0] == '1': 
...   return True 
...  else: 
...   return False 
... 
>>> with mock.patch.object(EmailChecker, 'is_email_correct', side_effect=my_side_effect): 
...  checker = EmailChecker() 
...  print(checker.is_email_correct('1')) 
...  print(checker.is_email_correct('2')) 
... 
True 
False 

참고은 : *args와 **args을 대체되었습니다. is_email_correct 메서드에 대한 인수가 누락되었습니다 (self).

def my_side_effect(email): 
    return email == '1'