Bir sahte yönteme ardışık çağrılar iddia etme

Sahte bir sahiptir yararlı assert_called_with()bir yöntem . Ancak, anladığım kadarıyla bu sadece bir yönteme son çağrıyı kontrol eder .
Alaycı yöntemi 3 kez art arda çağıran kodum varsa, her seferinde farklı parametrelerle, bu 3 çağrıyı belirli parametreleriyle nasıl onaylayabilirim?

assert_has_calls bu soruna başka bir yaklaşımdır.

Dokümanlardan:

assert_has_calls (çağrılar, any_order = Yanlış)

iddia belirtilen aramalarla çağrıldı iddia. Mock_calls listesi çağrılar için kontrol edilir.

Any_order Yanlış (varsayılan) ise, çağrılar sıralı olmalıdır. Belirtilen aramalardan önce veya sonra ekstra aramalar yapılabilir.

Any_order True ise, çağrılar herhangi bir sırada olabilir, ancak tümü mock_calls içinde görünmelidir.

Misal:

>>> from unittest.mock import call, Mock
>>> mock = Mock(return_value=None)
>>> mock(1)
>>> mock(2)
>>> mock(3)
>>> mock(4)
>>> calls = [call(2), call(3)]
>>> mock.assert_has_calls(calls)
>>> calls = [call(4), call(2), call(3)]
>>> mock.assert_has_calls(calls, any_order=True)

Kaynak: https://docs.python.org/3/library/unittest.mock.html#unittest.mock.Mock.assert_has_calls

Genellikle, aramaların sırasını umursamıyorum, sadece olduklarını. Bu durumda, assert_any_callhakkında bir iddia ile birleşiyorum call_count.

>>> import mock
>>> m = mock.Mock()
>>> m(1)
<Mock name='mock()' id='37578160'>
>>> m(2)
<Mock name='mock()' id='37578160'>
>>> m(3)
<Mock name='mock()' id='37578160'>
>>> m.assert_any_call(1)
>>> m.assert_any_call(2)
>>> m.assert_any_call(3)
>>> assert 3 == m.call_count
>>> m.assert_any_call(4)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "[python path]\lib\site-packages\mock.py", line 891, in assert_any_call
    '%s call not found' % expected_string
AssertionError: mock(4) call not found

Bunu tek bir yönteme aktarılan çağrıların büyük bir listesinden daha kolay okunur ve anlaşılır buluyorum.

Siparişe önem veriyorsanız veya birden fazla özdeş çağrı bekliyorsanız assert_has_callsdaha uygun olabilir.

Düzenle

Bu yanıtı yayınladığımdan beri, genel olarak test etme yaklaşımımı tekrar düşündüm. Testiniz bu kadar karmaşık hale geliyorsa, uygun olmayan testler yapıyor olabileceğinizi veya bir tasarım sorununuz olabileceğini belirtmek gerekir. Alaycılar, nesne yönelimli bir tasarımda nesneler arası iletişimi test etmek için tasarlanmıştır. Tasarımınıza itiraz edilmezse (daha prosedürel veya işlevsel olduğu gibi), sahte tamamen uygunsuz olabilir. Ayrıca, yöntemin içinde çok fazla şey olabilir veya en iyi kilitsiz bırakılan iç ayrıntıları test ediyor olabilirsiniz. Kodum çok nesne odaklı olmadığında bu yöntemde belirtilen stratejiyi geliştirdim ve ayrıca en iyi kilitsiz bırakılan iç detayları da test ettiğime inanıyorum.

Parametreleri önceki yöntem çağrılarıyla karşılaştırmak için Mock.call_args_listözniteliği kullanabilirsiniz. Bu Mock.call_countözellik ile birlikte size tam kontrol vermelidir.

Buna her zaman tekrar tekrar bakmak zorundayım, işte cevabım.

Aynı sınıftan farklı nesnelerde birden çok yöntem çağrısı iddia etme

Bir ağır hizmet sınıfımız olduğunu varsayalım (ki bunu yapmak istiyoruz):

In [1]: class HeavyDuty(object):
   ...:     def __init__(self):
   ...:         import time
   ...:         time.sleep(2)  # <- Spends a lot of time here
   ...:     
   ...:     def do_work(self, arg1, arg2):
   ...:         print("Called with %r and %r" % (arg1, arg2))
   ...:  

HeavyDutysınıfın iki örneğini kullanan bazı kodlar şunlardır:

In [2]: def heavy_work():
   ...:     hd1 = HeavyDuty()
   ...:     hd1.do_work(13, 17)
   ...:     hd2 = HeavyDuty()
   ...:     hd2.do_work(23, 29)
   ...:    

Şimdi, heavy_workfonksiyon için bir test örneği :

In [3]: from unittest.mock import patch, call
   ...: def test_heavy_work():
   ...:     expected_calls = [call.do_work(13, 17),call.do_work(23, 29)]
   ...:     
   ...:     with patch('__main__.HeavyDuty') as MockHeavyDuty:
   ...:         heavy_work()
   ...:         MockHeavyDuty.return_value.assert_has_calls(expected_calls)
   ...:  

HeavyDutySınıfı alay ediyoruz MockHeavyDuty. Yöneltmek yerine, her HeavyDutyörnekten gelen yöntem çağrılarını iddia etmek için . Buna ek olarak, listede hangi yöntem adını çağırmakla ilgilendiğimizi belirtmeliyiz. Bu yüzden listemiz, sadeceMockHeavyDuty.return_value.assert_has_callsMockHeavyDuty.assert_has_callsexpected_callscall.do_workcall .

Test senaryosunu kullanmak bize bunun başarılı olduğunu gösterir:

In [4]: print(test_heavy_work())
None

heavy_workİşlevi değiştirirsek, test başarısız olur ve yardımcı bir hata mesajı verir:

In [5]: def heavy_work():
   ...:     hd1 = HeavyDuty()
   ...:     hd1.do_work(113, 117)  # <- call args are different
   ...:     hd2 = HeavyDuty()
   ...:     hd2.do_work(123, 129)  # <- call args are different
   ...:     

In [6]: print(test_heavy_work())
---------------------------------------------------------------------------
(traceback omitted for clarity)

AssertionError: Calls not found.
Expected: [call.do_work(13, 17), call.do_work(23, 29)]
Actual: [call.do_work(113, 117), call.do_work(123, 129)]

Bir işleve birden çok çağrı iddia etme

Yukarıdakilerle karşılaştırmak için, bir işleve birden çok çağrının alay edilmesini gösteren bir örnek:

In [7]: def work_function(arg1, arg2):
   ...:     print("Called with args %r and %r" % (arg1, arg2))

In [8]: from unittest.mock import patch, call
   ...: def test_work_function():
   ...:     expected_calls = [call(13, 17), call(23, 29)]    
   ...:     with patch('__main__.work_function') as mock_work_function:
   ...:         work_function(13, 17)
   ...:         work_function(23, 29)
   ...:         mock_work_function.assert_has_calls(expected_calls)
   ...:    

In [9]: print(test_work_function())
None

İki ana fark vardır. Birincisi, bir işlevi alay ederken call, kullanmak yerine, beklenen çağrılarımızı kullanarak ayarladığımızdır call.some_method. İkincisi, assert_has_callson mock_work_functionyerine çağırıyoruz mock_work_function.return_value.