대체 코드의 적용 범위를 테스트하기 위해 ImportError 트리거

Question

somemod.py에서 이렇게 주어진 코드 :

try: 
    Import cStringIO as StringIO 
except ImportError: 
    import StringIO 

어떻게 'fallback'분기를 테스트 할 수 있습니까?

:

import somemod import somemod import somemod import somemod import somemod import somemod

Mocking ImportError in Python은이 중 일부를 다루지 만 복잡합니다!

Answer 1

첫째, 테스트를 위해 사용하는 함수를 작성

테스트 케이스 후

>>> import __builtin__ 
>>> original_import = __builtin__.__import__ 
>>> def import_hook(name, *args, **kwargs): 
... if name == 'cStringIO': raise ImportError('test case module import failure') 
... else: return original_import(name, *args, **kwargs) 
... 
>>> 
>>> __builtin__.__import__ = import_hook 
>>> somecode() 
got StringIO 

, 다시 설정해야합니다 :

>>> __builtin__.__import__ = original_import