似乎调用了Python补丁模拟,但断言失败

时间:2016-03-17 17:01:36

标签: python unit-testing python-unittest python-unittest.mock

我正在使用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.called报告'False'?

3 个答案:

答案 0 :(得分:3)

在此测试中永远不会调用

mockObj.fooself.uut.Start()调用mockObj,创建一个新的模拟,然后调用该模拟的 foo方法。如果要断言发生此调用,则需要访问正确的对象:

mockObj.return_value.foo.assert_called_with()

答案 1 :(得分:0)

这是链接: link

我的简单修复:

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

然后我可以毫不犹豫地使用'mockObj'。

上周,当我第一次了解嘲讽时,我遇到了链接的解决方案 - 有时候信息很快就会陷入其中。

答案 2 :(得分:0)

模拟断言语法有很多怪癖。为了解决这个问题,我写了一个helper library 为我生成断言。

这是您将其用于测试方法的方式:

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的调用相关。

因此,您不再需要自己找出确切的语法,只需应用该工具即可:)