Как высмеять возвращаемое значение метода класса, импортированного из другого класса в Python?

Question

У меня есть эти файлы:

/foo.py 
/foo2.py 
/test_food.py 

в foo.py, у меня есть это:

from foo2 import Foo2 

class Foo(object): 

    def GetFoo2(self): 
     f = Foo2() 
     return f.Get() 

и в foo2.py, у меня есть:

class Foo2(object): 

    def __init__(self): 
     self.myvar = "eeee" 

    def Get(self): 
     return self.myvar 

В test_foo.py ,

import os, sys, json, pytest 
from mock import * 
from foo import Foo 

def test_foo_ok(monkeypatch): 

    monkeypatch.setattr(Foo, "GetFoo2", lambda x:"abc") 
    f = Foo() 
    result = f.GetFoo2() 
    assert result == "abc" 

Итак, в test_foo.py, я могу высмеять или обезвредить метод GetFoo2(). Однако, вместо этого, как я могу высмеять возвращаемое значение метода Get() Foo2 из test_foo.py без изменения кода в foo.py и foo2.py?

Answer 1

Всякий раз, когда вы импортируете с помощью from, вы можете исправить пространство имен этого модуля. Вот пример для test_foo_ok функции:

import os, sys, json, pytest 
from mock import * 
from foo import Foo 

def test_foo_ok(monkeypatch): 
    monkeypatch.setattr("foo.Foo2.Get", lambda x:"abc") 
    f = Foo() 
    result = f.GetFoo2() 
    assert result == "abc" 

См where to patch раздел документации Ложная библиотеки для более подробной информации. Те же правила применяются к обезьянному патчу pytest.