0

我正在使用 Python 2.6.6

我想要做的是用 Mock 替换对象的创建,以确保进行正确的调用。应该直截了当。

我的模块:

import dir.SubModule

class Cls( object ):
    def Start( self ):
        self.__obj = dir.SubModule.SubCls()
        self.__obj.foo()

我的测试:

import MyModule
import unittest
from mock import Mock, MagicMock, patch

class MyTest( unittest.TestCase ):
    def setUp( self ):
        self.uut = MyModule.Cls()

    def test_one( self ):
        with patch( 'dir.SubModule.SubCls', spec=True ) as mockObj:
            print "mock calls before"
            print mockObj.mock_calls
            self.uut.Start()

            print "called: "      + str( mockObj.called )
            print "foo called: " + str( mockObj.foo.called )
            print "call_count: "  + str( mockObj.call_count )
            print "call_args: "   + str( mockObj.call_args )
            print "args_list: "   + str( mockObj.call_args_list )
            print "mock calls:\n" + str( mockObj.mock_calls )
            print "method calls:\n " + str( mockObj.method_calls )

输出是:

mock calls before:
[]
called: True
foo called: False
call_count: 1
call_args: call()
args_list: [call()]
mock calls:
[call(), call().foo()]
method calls:
[]

然而测试失败了:

AssertionError: Expected call: foo()
Not called

我不明白模拟是如何报告调用的,但我不能断言它们被调用了。我错过了什么?

编辑:在添加了所有指标的报告之后,我似乎对 python 模拟有一些基本的误解。如果 foo() 在调用列表中,那么为什么调用计数只有 1,为什么 foo.call 报告 'False'?

4

3 回答 3

6

mockObj.foo在此测试中从未调用过。self.uut.Start()调用mockObj,创建一个新的模拟,然后调用该模拟的 foo方法。如果你想断言这个调用发生了,你需要访问正确的对象:

mockObj.return_value.foo.assert_called_with()
于 2016-03-17T17:54:00.723 回答
0

模拟断言语法有很多怪癖。为了处理它,我编写了一个帮助库来为我生成断言。

以下是您将如何将其用于您的测试方法:

import MyModule
import unittest
from mock import patch

class MyTest( unittest.TestCase ):
    def setUp( self ):
        self.uut = MyModule.Cls()

    def test_one( self ):
        with patch( 'dir.SubModule.SubCls', spec=True ) as mockObj:
            self.uut.Start()

            # calls to generate_asserts, put this after the 'act'
            import mock_autogen.generator
            print(mock_autogen.generator.generate_asserts(mockObj))

你得到的输出是:

assert 1 == mockObj.call_count
mockObj.assert_called_once_with()
mockObj.return_value.foo.assert_called_once_with()

这与__init__调用相关,然后是foo对新创建对象的调用。

所以不再需要自己弄清楚确切的语法,只需应用该工具 :)

于 2019-09-06T07:17:54.450 回答
0

这是链接: 链接

我的简单修复:

with patch( 'class' ) as mockCreator:
    mockObj = mockCreator.return_value

然后我可以随意使用'mockObj'。

上周,当我第一次学习模拟时,我遇到了链接的解决方案——对于信息来说有点太早了。

于 2016-03-17T18:19:54.537 回答