您可以使用patch decorator，然后對模擬對象調用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')

總結

以上是生活随笔為你收集整理的python单元测试mock_Python单元测试mock，获取mocked函数的输入参数的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。