觸發導入錯誤，以測試覆蓋率後備代碼

Question

鑑於這樣的代碼在somemod.py的：

try: 
    Import cStringIO as StringIO 
except ImportError: 
    import StringIO 

一個如何測試「回退」分支？

（用例：努力實現100％覆蓋一致認爲，這是一個有點傻目標:)）。

併發症：

1. 代碼test_somemod.py具有import somemod
運行
2. Mocking ImportError in Python涵蓋了一些這個，但是很複雜！

Answer 1

首先，創建一個用於測試使用的功能：

>>> def somecode(): 
... try: 
...  import cStringIO as StringIO 
...  print 'got cStringIO' 
... except ImportError: 
...  import StringIO 
...  print 'got StringIO' 
>>> somecode() 
got cStringIO 

現在，作爲解釋here，您可以鉤到導入功能：

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