使用mox模拟一个名为by__init__的方法

Question

我想在由init方法调用的类中留出一个方法。

class MyClass(object):
  def __init__(self): 
    # Some initializer code here
    ...
    self.method_with_side_effects()

  def method_with_side_effects(self):
    ... # Load files, etc.

根据Mox文档，您可以通过实例化对象，然后使用StubOutWithMock方法来模拟方法。但在这种情况下，我不能这么做：

import mox
m = mox.Mox()
myobj = MyClass()
m.StubOutWithMock(myobj, "method_with_side_effects") # Too late!

还有其他方法可以消除这种方法吗？

Answer 1

您能直接子类MyClass并重写method_with_side_effects吗？

Answer 2

如果您只想对该方法进行存根处理，则可以使用“顽固”模块：

import stubout

s = stubout.StubOutForTesting()
s.Set(MyClass, 'method_with_side_effects', lambda self: None)

如果你真的想要模拟这个方法，那就更复杂了。您可以使用__new__()直接创建实例对象，以避免__init__()的副作用，并使用它记录预期的行为：

import mox

m = mox.Mox()
m.StubOutWithMock(MyClass, 'method_with_side_effects')

instance = MyClass.__new__(MyClass)
instance.method_with_side_effects().AndReturn(None)

m.ReplayAll()
MyClass()
m.VerifyAll()

如果您实际上不需要行为验证，那么使用存根代替模拟就不那么脆弱了。