我正在使用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'?
答案 0 :(得分:3)
mockObj.foo
。 self.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
的调用相关。
因此,您不再需要自己找出确切的语法,只需应用该工具即可:)