การยืนยันการเรียกไปยังเมธอด mock อย่างต่อเนื่อง

จำลองมีประโยชน์assert_called_with()วิธี อย่างไรก็ตามเท่าที่ฉันเข้าใจการตรวจสอบเพียงโทรครั้งสุดท้ายเพื่อวิธีการ
หากฉันมีรหัสที่เรียกใช้วิธีการเยาะเย้ย 3 ครั้งติดต่อกันแต่ละครั้งด้วยพารามิเตอร์ที่แตกต่างกันฉันจะยืนยันการโทร 3 ครั้งนี้ด้วยพารามิเตอร์เฉพาะได้อย่างไร

assert_has_calls เป็นอีกวิธีหนึ่งในการแก้ไขปัญหานี้

จากเอกสาร:

assert_has_calls (การโทร any_order = False)

ยืนยันการจำลองถูกเรียกด้วยการโทรที่ระบุ รายการ mock_calls ถูกตรวจสอบการโทร

ถ้า any_order เป็นเท็จ (ค่าเริ่มต้น) การโทรจะต้องต่อเนื่องกัน อาจมีการโทรเพิ่มเติมก่อนหรือหลังการโทรที่ระบุ

ถ้า any_order เป็น True การโทรสามารถอยู่ในลำดับใดก็ได้ แต่จะต้องปรากฏใน mock_calls

ตัวอย่าง:

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

ที่มา: https://docs.python.org/3/library/unittest.mock.html#unittest.mock.Mock.assert_has_calls

โดยปกติฉันไม่สนใจคำสั่งของการโทรเท่านั้นที่เกิดขึ้น ในกรณีที่ผมรวมกับการยืนยันเกี่ยวกับassert_any_callcall_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

ฉันพบว่าการทำเช่นนี้เพื่อให้อ่านและเข้าใจได้ง่ายกว่ารายการโทรจำนวนมากที่ส่งเป็นวิธีการเดียว

หากคุณใส่ใจในการสั่งซื้อหรือคุณคาดหวังว่าจะมีสายที่เหมือนกันหลายสายassert_has_callsอาจจะเหมาะสมกว่า

แก้ไข

ตั้งแต่ฉันโพสต์คำตอบนี้ฉันคิดใหม่เกี่ยวกับวิธีการทดสอบโดยทั่วไป ฉันคิดว่ามันคุ้มค่าที่จะกล่าวถึงว่าหากการทดสอบของคุณซับซ้อนนี้คุณอาจทำการทดสอบไม่เหมาะสมหรือมีปัญหาในการออกแบบ Mocks ถูกออกแบบมาสำหรับการทดสอบการสื่อสารระหว่างวัตถุในการออกแบบเชิงวัตถุ หากการออกแบบของคุณไม่ได้คัดค้าน (เช่นในขั้นตอนหรือการทำงานมากกว่า) การเยาะเย้ยอาจไม่เหมาะสมทั้งหมด นอกจากนี้คุณยังอาจใช้วิธีการมากเกินไปหรือคุณอาจทดสอบรายละเอียดภายในที่ยังไม่ได้ถอดออก ฉันพัฒนากลยุทธ์ที่กล่าวถึงในวิธีการนี้เมื่อรหัสของฉันไม่ได้มุ่งเน้นวัตถุมากนักและฉันเชื่อว่าฉันกำลังทดสอบรายละเอียดภายในที่จะไม่ได้รับการปลดจากตำแหน่งที่ดีที่สุด

คุณสามารถใช้Mock.call_args_listแอ็ตทริบิวต์เพื่อเปรียบเทียบพารามิเตอร์กับการเรียกใช้เมธอดก่อนหน้า ร่วมกับMock.call_countคุณสมบัติควรให้การควบคุมเต็มรูปแบบ

ฉันต้องดูครั้งนี้ซ้ำแล้วซ้ำอีกดังนั้นนี่คือคำตอบของฉัน

การยืนยันการเรียกใช้หลายวิธีบนวัตถุที่แตกต่างของคลาสเดียวกัน

สมมติว่าเรามีชั้นเรียนหนัก (ซึ่งเราต้องการเยาะเย้ย):

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))
   ...:  

นี่คือรหัสบางส่วนที่ใช้สองอินสแตนซ์ของ HeavyDutyคลาส:

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

ตอนนี้นี่คือกรณีทดสอบสำหรับheavy_workฟังก์ชั่น:

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)
   ...:  

เราจะเยาะเย้ยชั้นเรียนกับHeavyDuty MockHeavyDutyวิธีการที่จะยืนยันการโทรมาจากทุกHeavyDutyตัวอย่างเช่นเราต้องดูแทนMockHeavyDuty.return_value.assert_has_calls MockHeavyDuty.assert_has_callsนอกจากนี้ในรายการexpected_callsเราต้องระบุชื่อวิธีการที่เราสนใจในการยืนยันการโทร ดังนั้นรายการของเราจะทำจากการโทรไปเมื่อเทียบกับเพียงcall.do_workcall

การออกกำลังกายกรณีทดสอบแสดงให้เราเห็นว่ามันประสบความสำเร็จ:

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

หากเราปรับเปลี่ยนheavy_workฟังก์ชั่นการทดสอบล้มเหลวและสร้างข้อความแสดงข้อผิดพลาดที่เป็นประโยชน์:

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

การยืนยันการเรียกหลายฟังก์ชั่น

นี่คือตัวอย่างที่แสดงวิธีเยาะเย้ยการเรียกหลายฟังก์ชั่น:

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

มีความแตกต่างที่สำคัญสองประการ คนแรกคือว่าเมื่อเยาะเย้ยการตั้งค่าฟังก์ชั่นที่เราคาดว่าสายของเราโดยใช้แทนการใช้call call.some_methodคนที่สองคือการที่เราเรียกassert_has_callsบนแทนในmock_work_functionmock_work_function.return_value