不使用mock.patch调用函数

Question

我正在尝试测试连接到BluetoothSocket的Class BluetoothClient。为了避免使用真正的套接字，我只想测试使用正确的参数调用套接字中的connect（）方法。使用mock.patch替换我的bluetooth_control模块中导入的蓝牙模块并没有像预期的那样工作。

正如我所看到的，调用了connect（）方法，但断言告诉我。

代码：

被测单位（bluetooth_control.py）：

import bluetooth

class BluetoothClient(object):
    def __init__(self):
        self.address="98:D3:31:B2:EF:32"
        self.port=1

    def establishConnection(self):
        self.createSocket()
        self.connect()

    def createSocket(self):
        self.sock=bluetooth.BluetoothSocket( bluetooth.RFCOMM )

    def connect(self):
        print "connect: sock="+str(self.sock)
        self.sock.connect((self.address, self.port))

测试（bluetooth_control_test.py）：

import unittest
import mock
import bluetooth_control
import bluetooth

class TestShelf(unittest.TestCase):

    def setUp(self):
        unittest.TestCase.setUp(self)
        self.bc = bluetooth_control.BluetoothClient()
        print "setUp"

    def tearDown(self):
        self.shelf = None
        print "tearDown"

    @mock.patch('bluetooth_control.bluetooth')
    def testEstablishConnection(self,mock_bluetooth):
        self.bc.establishConnection()
        print "testEstablishConnection sock="+str(self.bc.sock)
        mock_bluetooth.connect().assert_called_with(self.bc.sock,("98:D3:31:B2:EF:32",1))


if __name__ == "__main__":
    unittest.main()

输出：

setUp
connect: sock=<MagicMock name='bluetooth.BluetoothSocket()' id='140433322111504'>
testEstablishConnection sock=<MagicMock name='bluetooth.BluetoothSocket()' id='140433322111504'>
FtearDown

======================================================================
FAIL: testEstablishConnection (__main__.TestShelf)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/usr/lib/python2.7/site-packages/mock.py", line 1201, in patched
    return func(*args, **keywargs)
  File "bluetooth_control_test.py", line 21, in testEstablishConnection
    mock_bluetooth.connect().assert_called_with(self.bc.sock,("98:D3:31:B2:EF:32",1))
  File "/usr/lib/python2.7/site-packages/mock.py", line 831, in assert_called_with
    raise AssertionError('Expected call: %s\nNot called' % (expected,))
AssertionError: Expected call: mock(<MagicMock name='bluetooth.BluetoothSocket()' id='140433322111504'>, ('98:D3:31:B2:EF:32', 1))
Not called

----------------------------------------------------------------------
Ran 1 test in 0.003s

FAILED (failures=1)

Answer 1

两天后再次看到这个问题后，我发现了我犯的愚蠢错误。我必须修补实际的方法，并在断言中删除错误添加的括号。

我不打算删除这个问题，这样可能会帮助别人避免这些错误。

测试

import unittest
import mock
import bluetooth_control
import bluetooth

class TestShelf(unittest.TestCase):

    def setUp(self):
        unittest.TestCase.setUp(self)
        self.bc = bluetooth_control.BluetoothClient()
        print "setUp"

    def tearDown(self):
        self.shelf = None
        print "tearDown"

    @mock.patch('bluetooth_control.bluetooth.BluetoothSocket.connect')
    def testEstablishConnection(self,mock_connect):
        self.bc.establishConnection()
        print "testEstablishConnection sock="+str(self.bc.sock)
        mock_connect.assert_called_with(("98:D3:31:B2:EF:32",1))


if __name__ == "__main__":
    unittest.main()