我想知道是否可以使用Python mock库来模拟C函数以及如何?
要在Python中对C函数进行单元测试,我使用共享库,但我无法模拟一个函数以通过测试。当不需要使用mock时,测试工作正常。
所以我的问题是:是否可以使用Python mock模拟C函数以及如何?
以下是源代码示例。
//read_sensor.c
#include <stdint.h>
#include "read_sensor.h"
uint8_t read_sensor (void)
{
return 0xff;
}
//event_report.c
include <stdint.h>
#include <string.h>
#include "event_report.h"
#include "read_sensor.h"
#define TRUE 1
#define FALSE 0
extern char status ;
extern uint8_t flag;
void read_distance(void)
{
uint8_t count;
uint8_t value;
uint8_t flag_count = 0;
for (count = 0; count < 3; count ++)
{
value = read_sensor();
if ( value < 100 )
{
flag_count++;
}
}
if ( flag_count == 3)
{
flag = TRUE;
}
else
{
flag = FALSE;
}
}
,测试在这里
import ctypes
from ctypes import *
import mock
import unittest
TRUE = 1
FALSE = 0
class TestClass(unittest.TestCase):
def setUp(self):
print "\nSetUp\n"
self.event=ctypes.CDLL('d:/test_pytest/test_sensor_report/event_report.so')
def tearDown(self):
print "tearDown\n"
def test_1_flag_should_be_true_when_readed_distance_is_0(self):
expected_flag = TRUE
self.event.read_sensor = MagicMock(return_value = 0)
#calling function under test
result = self.event.read_distance()
print result
#testing assertion
assert expected_flag == result
用inspect.getmembers()
检查后,我得到以下内容:
[('_FuncPtr', <class 'ctypes._FuncPtr'>), ('__class__', <class
'ctypes.CDLL'>), ('__delattr__', <method-wrapper '__delattr__' of
CDLL object at 0xffa87eec>), ('__dict__', {'_FuncPtr': <class
'ctypes._FuncPtr'>, '_handle': 1690304512, '_name':
'd:/test_pytest/test_sensor_report/event_report.so'}),
('__doc__', "An instance of this class represents a loaded
dll/shared\n library, exporting functions using the standard C
calling\n convention (named 'cdecl' on Windows).\n\n The
exported functions can be accessed as attributes, or by\n
indexing with the function name. Examples:\n\n <obj>.qsort ->
callable object\n <obj>['qsort'] -> callable object\n\n
Calling the functions releases the Python GIL during the call
and\n reacquires it afterwards.\n "), ('__format__',
<built-in method __format__ of CDLL object at 0xffa87eec>),
('__getattr__', <bound method CDLL.__getattr__ of <CDLL
'd:/test_pytest/test_sensor_report/event_report.so', handle
64c00000 at ffa87eec>>), ('__getattribute__', <method-wrapper
'__getattribute__' of CDLL object at 0xffa87eec>),
('__getitem__', <bound method CDLL.__getitem__ of <CDLL
'd:/test_pytest/test_sensor_report/event_report.so', handle
64c00000 at ffa87eec>>), ('__hash__', <method-wrapper
'__hash__' of CDLL object at 0xffa87eec>), ('__init__',
<bound method CDLL.__init__ of <CDLL
'd:/test_pytest/test_sensor_report/event_report.so',
handle 64c00000 at ffa87eec>>), ('__module__', 'ctypes'),
('__new__', <built-in method __new__ of type object at
0x3d35b600>), ('__reduce__', <built-in method __reduce__ of
CDLL object at 0xffa87eec>), ('__reduce_ex__', <built-in
method __reduce_ex__ of CDLL object at 0xffa87eec>),
('__repr__', <bound method CDLL.__repr__ of <CDLL
'd:/test_pytest/test_sensor_report/event_report.so',
handle 64c00000 at ffa87eec>>), ('__setattr__',
<method-wrapper '__setattr__' of CDLL object at
0xffa87eec>), ('__sizeof__', <built-in method
__sizeof__ of CDLL object at 0xffa87eec>),
('__str__', <method-wrapper '__str__' of CDLL object
at 0xffa87eec>), ('__subclasshook__', <built-in
method __subclasshook__ of type object at
0xffae8204>), ('__weakref__', None),
('_func_flags_', 1), ('_func_restype_', <class
'ctypes.c_long'>), ('_handle', 1690304512),
('_name',
'd:/test_pytest/test_sensor_report/event_report.so')]
答案 0 :(得分:2)
如果你嘲笑它,你假装它是可靠的。或者至少不要使用mock测试测试中的模拟功能。
没有必要模拟C函数,不需要加载SO对象。
模拟self.event并添加Mock实例&#34; read_sensor&#34;对它,使用模拟。