Pythonで擬似ファイルオブジェクトを使って単体テストを行う方法は？

Question

私はparser = ParserClass('/path/to/file')のようなファイル名を与えてインスタンス化するクラスを持っています。そして、私はparser.parse()というメソッドを呼び出してファイルを読み込みます。
今、私はユニットテストしたい悪い何かが内部で起こった場合、その：正しい例外が発生しますので、私はこのような__builtin__.openをモックとしたい

with open(filename, 'rb') as fp: 
    // do something 

：

from mock import MagicMock, patch 
from StringIO import StringIO 

test_lines = StringIO("""some test lines, emulating a real file content""") 
mock_open = MagicMock(return_value=test_lines) 
with patch('__builtin__.open', mock_open): 
    self.mock.parse() 

が、これは私を与えますAttributeError: StringIO instance has no attribute '__exit__'。
StringIOはファイルオブジェクトとまったく同じように動作しますが、そうではありません。

モックオブジェクトで特定のコンテンツ（test_lines）を使ってこのメソッドをテストするにはどうすればよいですか？代わりに私は何を使うべきですか？

Answer 1

あなたはコンテキストマネージャ提供するStringIOをサブクラス化することができます：

class ContextualStringIO(StringIO): 
    def __enter__(self): 
     return self 
    def __exit__(self, *args): 
     self.close() # icecrime does it, so I guess I should, too 
     return False # Indicate that we haven't handled the exception, if received 


test_lines = ContextualStringIO(...) 

---

総投機：StringIOオブジェクトがドロップイン代替品fileオブジェクトため除くコンテキストマネージャの不足のためであれば、Iこれがうまくいくかどうか疑問：何Fiの

class ContextualStringIO(StringIO, file): 
    pass 

ContextualStringIO継承le操作はStringIOから可能ですが、それ以外はすべてfileから継承されています。それはエレガントに見えますが、おそらく広範なテスト（またはこれがうまくいかない理由を説明するためにPythonの内部構造に詳しい方）が必要です。

Answer 2

これは、StringIOがコンテキストマネージャプロトコルを実装していないという既知の問題です。

common recipeは、以下である：

from contextlib import contextmanager 


@contextmanager 
def StringIO(): 
    """Add support for 'with' statement to StringIO - http://bugs.python.org/issue1286 
    """ 
    try: 
     from cStringIO import StringIO 
    except ImportError: 
     from StringIO import StringIO 

    sio = StringIO() 

    try: 
     yield sio 
    finally: 
     sio.close() 

それはStringIOのコンテキストマネージャプロトコルを実装し、それがwith文で使用されることを可能にします。

---

UPDATEまあ、私は直接文字列から読み取ることができるmock_openの存在を発見したので、それはおそらく行く方法です。 （注言うように）これだけunicodeタイプで使用することができますしかし

The io module provides the Python interfaces to stream handling. Under Python 2.x, this is proposed as an alternative to the built-in file object, but in Python 3.x it is the default interface to access files and streams.

：

Answer 3

また、あなたはio.StringIO from the Standard Libraryを使用することができます。

Answer 4

ありa provisionがmockライブラリに特にこの目的のために、である：それは（あなたがすぐfor line in opened_mock_fileを行うことができないので、すなわち__iter__方法。）デフォルトのイテレータをサポートして行方不明にissueを持ってい

が、記載された通りに処理することができる。here。

私は@ icecrimeの回答をupvotedしていますが、彼の更新部分は十分に強調表示されていないようです。