Obiekt poprawki Pythona z parametrem side_effect

Question

Próbuję mieć obiekt Mock zwracający pewne wartości na podstawie podanego danych wejściowych. Poszukałem kilku przykładów SO i z jakiegoś powodu nadal nie mogę go uruchomić. Oto, co mam teraz.

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) 

Jednak nie wydaje się działać. Czego tu mi brakuje?

Answer 1

Zastosowanie patch.object jako dekorator lub kontekstowym menedżera, jak w poniższym kodzie:

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

UWAGA: Zastępuje **args z *args. Dodano brakujący argument self dla metody is_email_correct.

---

my_side_effect można uprościć następująco:

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