为什么mocking在nosetests中存在代码多次测试后仍然有效？

Question

我有以下代码 test_A.py ，它模拟 MyClass.mymethod：

from unittest import main
from mocker import Mocker, MockerTestCase
Class test_A(MockerTestCase):
  def setUp(self):
    self.m=Mock()
    MyClass.mymethod = self.m.mock()
    self.m.result(None)
    self.m.count(0,None)
    self.m.replay()

  def test_me(self):
    #Do something about MyClass.method

  def tearDown(self):
    self.m.restore()
    self.m.verify()

我还有另一个代码 test_B.py ，它不模拟 MyClass.mymethod：

Class test_B(MockerTestCase):
  def setUp(self):
    pass

  def test_me(self):
    #Do something about MyClass.method

  def tearDown(self):
     pass

但是，当我执行“nosetests test_A.py test_B.py”时，测试后看起来像test_A.py 并进入 test_B.py，MyClass.mymethod 仍然是模拟的。不知道为什么以及如何解决它。谢谢！

Answer 1

该行：

MyClass.mymethod = self.m.mock()

确实用新对象替换了 MyClass.mymethod() 。对 MyClass.mymethod 的所有后续引用都将指向模拟对象，即使这些引用位于不同的类中。

您想要的是一种替换仅在类 test_A 中工作的 MyClass.mymethod() 的方法。实现此目的的最简单方法是在 tearDown 方法中恢复原始的 mymethod：

Class test_A():
    def setUp(self):
        self.originalMyMethod = MyClass.mymethod
        self.m=Mock()
        MyClass.mymethod = self.m.mock()
        self.m.result(None)
        self.m.count(0,None)
        self.m.replay()

    def test_me(self):
        # Do something about MyClass.mymethod

    def tearDown(self):
        self.m.restore()
        self.m.verify()
        MyClass.mymethod = self.originalMyMethod