我有一个类,它有一个__exit__
和__enter__
函数,这样我就可以在with语句中使用它,例如:
with ClassName() as c:
c.do_something()
我现在正试图编写一个单元测试来测试这个。基本上,我试图测试do_something()
只被调用过一次。你知道吗
一个例子(我称之为testmocking1
):
class temp:
def __init__(self):
pass
def __enter__(self):
pass
def __exit__(self, exc_type, exc_val, exc_tb):
pass
def test_method(self):
return 1
def fun():
with temp() as t:
return t.test_method()
还有我的测试:
import unittest
import test_mocking1
from test_mocking1 import fun
import mock
from mock import patch
class MyTestCase(unittest.TestCase):
@patch('test_mocking1.temp', autospec = True)
def test_fun_enter_called_once(self, mocked_object):
fun()
mocked_object.test_method.assert_called_once()
if __name__ == '__main__':
unittest.main()
所以我希望它能通过,因为test\u方法在函数fun()
中只被调用过一次。但我得到的实际结果是:
======================================================================
FAIL: test_fun_enter_called_once (__main__.MyTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
File "<path_to_virtual_env>\lib\site-packages\mock\mock.py", line 1305, in patched
return func(*args, **keywargs)
File "<File_with_test>", line 11, in test_fun_enter_called_once
mocked_object.test_method.assert_called_once()
File "<path_to_virtual_env>\lib\site-
packages\mock\mock.py", line 915, in assert_called_once
raise AssertionError(msg)
AssertionError: Expected 'test_method' to have been called once. Called 0 times.
如何测试使用with
语句创建的类中的函数是否已被调用(一次或多次),以及(相关)如何设置这些调用的结果(使用.side_effect
或.return_value
)?你知道吗
with
语句接受__enter__
返回的内容,以绑定到as <name>
部分中的名称。您将其绑定到t
:注意,
temp()
被调用,因此with
语句以temp.return_value
开头。t
也不是temp.return_value
,它是temp().__enter__()
返回的任何值,因此需要使用该调用的返回值:在此基础上扩展,如果要更改
test_method()
返回的内容,请对mocked_object.return_value.__enter__.return_value
的返回值进行更改。你知道吗始终可以打印出对象的^{} attribute ,以查看对象发生了什么:
请注意,
temp.__enter__()
的实际实现返回None
,因此在不模拟fun()
函数的情况下,函数会失败并出现属性错误。你知道吗相关问题 更多 >
编程相关推荐