我有一堂课有一个__exit__
and __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_method 在函数中被调用了一次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
or .return_value
)?