0
我想模拟从我的类中使用的pyserial中的Serial对象的读写函数来检查调用参数并编辑返回值,但似乎没有工作。在python中模仿成员对象的方法
目前,我有一个文件 'serialdevice.py',像这样:
import serial
class SerialDevice:
def __init__(self):
self._serial = serial.Serial(port='someport')
def readwrite(self, msg):
self._serial.write(msg)
return self._serial.read(1024)
然后,我有一个文件 'test_serialdevice.py',像这样:
import mock
from serialdevice import SerialDevice
@mock.patch('serialdevice.serial.Serial.read')
@mock.patch('serialdevice.serial.Serial.write')
@mock.patch('serialdevice.serial.Serial')
def test_write(mock_serial, mock_write, mock_read):
mock_read.return_value='hello'
sd = SerialDevice()
resp = sd.readwrite('test')
mock_write.assert_called_once_with('test')
assert resp == 'hello'
但两者断言失败。不知怎的,mock_write不是用参数'test'
调用的,write方法返回一个模拟对象而不是'hello'
字符串。我也试过使用@patch('serial.Serial.write)
等,结果相同。还使用mock_serial
的返回对象,例如, mock_read = mock_serial.read()
似乎不起作用。
构造函数,即mock_serial
,似乎确实被调用了期望的参数。
在这种情况下,我怎么能达到我想要的?
的Python版本是2.7.9