我使用ReviewBoard API庫,今天我將代碼移動到單獨的類,並希望用一些測試覆蓋邏輯。我瞭解嘲笑和測試,但我顯然沒有太多的經驗與Python和它的圖書館。這裏是真正的代碼塊:嘲笑ReviewBoard第三方庫使用Python和模擬
<!-- language: python -->
from rbtools.api.client import RBClient
class ReviewBoardWrapper():
def __init__(self, url, username, password):
self.url = url
self.username = username
self.password = password
pass
def Connect(self):
self.client = RBClient(self.url, username=self.username, password=self.password)
self.root = self.client.get_root()
pass
我想斷言初始化以及調用get_root()方法。下面是我如何努力來實現這一目標:
<!-- language: python -->
import unittest
import mock
from module_base import ReviewBoardWrapper as rb
class RbTestCase(unittest.TestCase):
@mock.patch('module_base.RBClient')
@mock.patch('module_base.RBClient.get_root')
def test_client_connect(self, mock_client, mock_method):
rb_client = rb('', '', '')
rb_client.Connect()
self.assertTrue(mock_method.called)
self.assertTrue(mock_client.called)
而且這裏是我堅持的錯誤:
$ python -m unittest module_base_tests
F.
======================================================================
FAIL: test_client_connect (module_base_tests.RbTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/usr/local/lib/python2.7/dist-packages/mock/mock.py", line 1305, in patched
return func(*args, **keywargs)
File "module_base_tests.py", line 21, in test_client_connect
self.assertTrue(mock_client.called)
AssertionError: False is not true
----------------------------------------------------------------------
Ran 2 tests in 0.002s
FAILED (failures=1)
我該怎麼辦錯了嗎?我是否正確地模擬了導入庫的「本地副本」?問題完全在於不同的領域嗎?
我也試着這樣做:
@mock.patch('module_base.RBClient.__init__')
和/或本:
self.assertTrue(mock_client.__init__.called)
正如DanGar如下所述,補丁裝飾的錯誤的順序旁邊的解決方案是斷言的返回值: assertTrue(mock_method.return_value.called) – qlf00n