Как издеваются класс, чтобы он не работает на импорт

Question

У меня есть следующие файлы:

test_temp.py:

import pytest 
from mock import patch 

with patch('temp.TheEvilClass'), patch('temp2.TheEvilClass'): 
    from temp import TempClass 


def test_evil(): 
    assert True 

temp.py:

from temp2 import TheEvilClass 

class TempClass(object): 
    TheEvilClass() 

temp2.py:

class TheEvilClass(object): 
    print "I'm evil! ]:>" 
    raise Exception 

My i ntention был для патча в test_temp.py, чтобы высмеять TheEvilClass, чтобы исключение никогда не вызывалось. Я попытался исправить его в обоих местах, но исключение все еще бросается. У вас есть какие-то указания на то, что я делаю неправильно, и как тестировать temp.py, не вызывая фактического TheEvilClass? Я попытался насмехаться над всем модулем (с patch('temp2')), но это тоже не работает.

Answer 1

То, что вы пытаетесь сделать, невозможно с mock, так как сам патч запускает импорт. patch звонки _get_target, который использует _importer, который вызывает __import__.

https://github.com/testing-cabal/mock/blob/master/mock/mock.py#L1206

Более простой способ увидеть, что только путем применения пластыря непосредственно:

patch('temp.TheEvilClass').__enter__() 

Правильный путь об этом является создание тестовой среды, так что необработанное исключение Арен» во время импорта.