我有以下測試代碼來測試Decorator。@mock.patch('a.b.c.KafkaProducer')def test_1(self, mocked): decorator = Decorator(....) @decorator() def test(): return 42 test() start_time = ??? v = {'type': 'batch', 'start_time': start_time.isoformat()} mocked.return_value.send.assert_called_once_with(value=v)但是，測試總是失敗，因為使用屬性分配給Decorator的字典參數模擬調用。這是一種比較除 之外的所有內容的方法嗎？或者有其他方法來測試通話嗎？start_timedatetime.now()start_time
1 回答
一只萌萌小番薯
TA貢獻1795條經驗 獲得超7個贊
兩種實用方法：
使用https://pypi.org/project/freezegun/凍結時間
import datetime
from freezegun import freeze_time
from unittest.mock import patch
import my_library
NOT_NOW = datetime.datetime.now()
@freeze_time("2020-01-01")
@patch("my_library.helper")
def test_foo(_helper):
my_library.under_test(NOT_NOW)
# kinda auto-magic
_helper.assert_called_once_with(datetime.datetime.now())
# or more explicitly
_helper.assert_called_once_with(datetime.datetime(2020, 1, 1))
或者，手動評估參數
@patch("my_library.helper", return_value=42)
def test_bar(_helper):
my_library.under_test(NOT_NOW)
assert _helper.call_count == 1
assert _helper.call_args[0][0]
assert _helper.call_args[0][0] != NOT_NOW
添加回答
舉報
0/150
提交
取消