是否可以使用python模拟C函数?

时间:2015-08-19 08:21:56

标签: python c unit-testing mocking pymock

我想知道是否可以使用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')]

1 个答案:

答案 0 :(得分:2)

如果你嘲笑它,你假装它是可靠的。或者至少不要使用mock测试测试中的模拟功能。

没有必要模拟C函数,不需要加载SO对象。

模拟self.event并添加Mock实例&#34; read_sensor&#34;对它,使用模拟。