Python單元測試模擬，得到模擬函數的輸入參數

Question

我想要一個單元測試來聲明一個函數內的變量action被設置爲它的期望值，這個變量被使用的唯一時間是當它在一個調用中傳遞時到圖書館。

Class Monolith(object): 
    def foo(self, raw_event): 
     action = # ... Parse Event 
     # Middle of function 
     lib.event.Event(METADATA, action) 
     # Continue on to use the build event. 

我的想法是，我可以嘲笑lib.event.Event，並獲取其輸入參數，並聲稱他們的具體數值。

>這是不是嘲笑工作？模擬文檔使我感到沮喪，因爲它的不一致性，半範例和與我想要做的事無關的很多例子。

Answer 1

你可以使用貼片裝飾，然後調用assert_called_with到像這樣的嘲笑對象：

如果你有這樣的結構：

example.py 
tests.py 
lib/__init__.py 
lib/event.py 

而且example.py是內容：

import lib 

METADATA = 'metadata_example' 

class Monolith(object): 

    def foo(self, raw_event): 
     action = 'action_example' # ... Parse Event 
     # Middle of function 
     lib.event.Event(METADATA, action) 
     # Continue on to use the build event. 

而lib中的內容/ event.py是：

class Event(object): 

    def __init__(self, metadata, action): 
     pass 

tests.py的代碼應該是這樣的：

import mock 
import unittest 

from lib.event import Event 
from example import Monolith 


class TestExample(unittest.TestCase): 

    @mock.patch('lib.event.Event') 
    def test_example1(self, event_mocked): 
     # Setup 
     m = Monolith() 

     # Exercise 
     m.foo('raw_event') 

     # Verify 
     event_mocked.assert_called_with('metadata_example', 'action_example')