嘲弄一个在与政治家的谈话中使用的阶级

2024-03-29 12:55:39 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个类,它有一个__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)?你知道吗


Tags: 函数testimportselfreturndefwithmock
1条回答
网友
1楼 · 发布于 2024-03-29 12:55:39

with语句接受__enter__返回的内容,以绑定到as <name>部分中的名称。您将其绑定到t

with temp() as t:
    t.test_method()

注意,temp()被调用,因此with语句以temp.return_value开头。t也不是temp.return_value,它是temp().__enter__()返回的任何值,因此需要使用该调用的返回值:

entered = mocked_object.return_value.__enter__.return_value
entered.test_method.assert_called_once()

在此基础上扩展,如果要更改test_method()返回的内容,请对mocked_object.return_value.__enter__.return_value的返回值进行更改。你知道吗

始终可以打印出对象的^{} attribute,以查看对象发生了什么:

>>> from test_mocking1 import fun
>>> from mock import patch
>>> with patch('test_mocking1.temp', autospec = True) as mocked_object:
...     fun()
...
>>> print(mocked_object.mock_calls)
[call(),
 call().__enter__(),
 call().__enter__().test_method(),
 call().__exit__(None, None, None)]
>>> mocked_object.return_value.__enter__.return_value.test_method.called
True
>>> mocked_object.return_value.__enter__.return_value.test_method.call_count
1

请注意,temp.__enter__()的实际实现返回None,因此在不模拟fun()函数的情况下,函数会失败并出现属性错误。你知道吗

相关问题 更多 >