我有以下測(cè)試代碼來(lái)測(cè)試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)但是，測(cè)試總是失敗，因?yàn)槭褂脤傩苑峙浣oDecorator的字典參數(shù)模擬調(diào)用。這是一種比較除 之外的所有內(nèi)容的方法嗎？或者有其他方法來(lái)測(cè)試通話嗎？start_timedatetime.now()start_time
1 回答
一只萌萌小番薯
TA貢獻(xiàn)1795條經(jīng)驗(yàn) 獲得超7個(gè)贊
兩種實(shí)用方法：
使用https://pypi.org/project/freezegun/凍結(jié)時(shí)間
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))
或者，手動(dòng)評(píng)估參數(shù)
@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
添加回答
舉報(bào)
0/150
提交
取消