使用 python 和 mock 模拟 ReviewBoard 第三方库

Question

我使用 ReviewBoard API 库，今天我将代码移到单独的 class 并想用一些测试来覆盖逻辑。我了解模拟和测试，但我显然对 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)

我做错了什么？我是否正确模拟了导入库的 "local copy"？问题是否完全出在不同的领域？

我也试过这样做：

@mock.patch('module_base.RBClient.__init__')

和/或这个：

self.assertTrue(mock_client.__init__.called)

Answer 1

在您 post 的示例中，模拟的顺序是相反的：

test_client_connect(self, mock_client, mock_method)

客户端实际上被模拟为第二个参数，方法调用被模拟为第一个参数。

但是，要正确模拟客户端，您需要模拟客户端调用的 return 值。模拟 return 值并对 return 值进行断言的示例如下所示：

class RbTestCase(unittest.TestCase):                                                                                                                                                                        

    @mock.patch('module_base.RBClient')                                                                                                                                                                     
    def test_client_connect(self, mock_client):                                                                                                                                                             
        client = mock.MagicMock()                                                                                                                                                                           
        mock_client.return_value = client                                                                                                                                                                   
        rb_client = rb('', '', '')                                                                                                                                                                          
        rb_client.Connect()                                                                                                                                                                                 
        self.assertTrue(client.get_root.called)                                                                                                                                                             
        self.assertTrue(mock_client.called)

使用 python 和 mock 模拟 ReviewBoard 第三方库

Mocking ReviewBoard third party library using python and mock

python

mocking

python-unittest

review-board