嘲笑自己类的基类时停止迭代

Question

在一个相当复杂的测试场景中，我需要模拟我自己的一个类的基类，并多次实例化后者。当我这样做时，我的测试会出现StopIteration异常。以下是我在这方面的情景：

被测代码(my_class.py(：

from oauthlib.oauth2 import BackendApplicationClient
from requests_oauthlib import OAuth2Session
class MySession(OAuth2Session):
pass
class MyClass:
def init(self, x):
self.x = x
client = BackendApplicationClient(client_id=x)
self.session = MySession(client=client)

测试代码(test_mock.py(:

import unittest
from unittest.mock import patch
with patch('requests_oauthlib.OAuth2Session') as MockSession:
from my_class import MyClass
cls = MyClass()
class MockTest(unittest.TestCase):
def test_mock_1(self):
cls.init(1)
self.assertIsNotNone(cls.session)
def test_mock_2(self):
cls.init(2)
self.assertIsNotNone(cls.session)

测试结果：

$ python -m unittest test_mock
.E
======================================================================
ERROR: test_mock_2 (test_mock.MockTest)
----------------------------------------------------------------------
Traceback (most recent call last):
File "...test_mock.py", line 16, in test_mock_2
cls.init(2)
File "...my_class.py", line 11, in init
self.session = MySession(client=client)
File "C:Python39libunittestmock.py", line 1093, in __call__
return self._mock_call(*args, **kwargs)
File "C:Python39libunittestmock.py", line 1097, in _mock_call
return self._execute_mock_call(*args, **kwargs)
File "C:Python39libunittestmock.py", line 1154, in _execute_mock_call
result = next(effect)
StopIteration
----------------------------------------------------------------------
Ran 2 tests in 0.003s
FAILED (errors=1)

我已经调试到unittest.mock.MagicMock类中，但我不知道发生了什么。在MagicMock的_execute_mock_call()方法中，我注意到self.side_effect是一个元组迭代器对象，当在第二个测试(test_mock_2(中调用next()时，它会产生StopIteration。

两个测试都运行"；OK"；如果我不使用MySession子类，即MyClass的init()方法中的self.session = OAuth2Session(client=client)。(但测试中的真实代码并不是这样工作的…(

有什么想法吗？

Answer 1

您应该模拟直接使用的类，因为您的自定义类继承了mock，接下来会启动意外行为。

将path方法重写到自定义类。

import unittest
from unittest.mock import patch
with patch('my_class.MySession') as MockSession:
from my_class import MyClass