一時ファイル用のモックオブジェクトの作成方法NamedTemporaryFileまたはオブジェクトのようなファイル

Question

以下は私のコードです：myfunction（）をテストする必要があります。ファイルの模擬機能を作成する方法。

def myfunction(self): 
    with tempfile.NamedTemporaryFile() as tf: 
     f.seek(0) 
     tf.write(f.read()) 
     tf.flush() 
     ocr_content_dict = self.ocr.ocr_document(tf.name, mimetype) or '' 
     ocr_content = ocr_content_dict['content'] 

Answer 1

あなたはモックファイルオブジェクトを作成することもできますし、また、使用することができBytesIO/StringIO：

from io import BytesIO 

class MockFileObject(object): 
    ''' 
     write mock functions for any that are needed 
     DONE: read 
     TODO: write 
     TODO: seek 
     TODO: fileno 
     TODO: flush 
     TODO: name 

     with context manangement use the dunder enter/exit 
     TODO: __enter__ 
     TODO: __exit__ 

     ... etc 
    ''' 

    def read(self): 
     ''' example mock ''' 
     return BytesIO('some stuff').read() 

使用法：

fo = MockFileObject() 
with fo as f: 
    print(f.read()) 

考え出力（未テスト）：

some stuff