Используйте mox, чтобы смоделировать метод by__init__

1

Я хотел бы исключить один метод в классе, который вызывается методом 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!

Есть ли другой способ заглушить этот метод?

Теги:
mox

2 ответа

3
Лучший ответ

Не могли бы вы подклассифицировать MyClass напрямую и переопределить method_with_side_effects?

1

Если вы хотите только заглушить метод, вы можете использовать модуль stubout:

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()

Если вам действительно не нужна проверка поведения, использование заглушек вместо mocks намного менее хрупкое.

Ещё вопросы

Сообщество Overcoder
Наверх
Меню