1

在一个相当复杂的测试场景中，我需要模拟我自己的一个类的基类并多次实例化后者。当我这样做时，我的测试错误并出现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:\Python39\lib\unittest\mock.py", line 1093, in __call__
    return self._mock_call(*args, **kwargs)
  File "C:\Python39\lib\unittest\mock.py", line 1097, in _mock_call
    return self._execute_mock_call(*args, **kwargs)
  File "C:\Python39\lib\unittest\mock.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是一个元组迭代器对象，当next()在第二个测试（）中调用它时，test_mock_2它会导致StopIteration.

如果我不使用MySession子类，即self.session = OAuth2Session(client=client)在 MyClassinit()方法中，这两个测试都运行“OK”。（但这并不是真正的被测代码的工作方式......）

有什么想法吗？

4

1 回答 1

1

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

将路径方法重写为您的自定义类。

import unittest
from unittest.mock import patch

with patch('my_class.MySession') as MockSession:
    from my_class import MyClass
于 2021-12-21T14:33:12.687 回答