Python：unittestingのためにファイルを偽装する

Question

私は自分のスクリプトに関数の単体テストを書いていますが、モックファイルの作成には問題があります。 私の関数はファイル名を取り、それの出力はmd5です。私はエラーを取得するしかし

import mock 
class Function_to_test_TestCase(unittest.TestCase): 

    def test_filename(self): 
     with mock.patch('__main__.open', mock.mock_open(read_data=''), create=True) as m: 
      md5_value=my_script.function_to_get_md5_value(m) 

：

with open(fname, "rb") as f: 
TypeError: coercing to Unicode: need string or buffer, MagicMock found 

はこのように、関連するモックMD5値とモックのファイルを作成するための正しい方法をされた時点での私のunittestの

def md5(file_name): 
    #Function to return the md5 value of a file 
    hash_md5 = hashlib.md5() 
    with open(fname, "rb") as f: 
     for chunk in iter(lambda: f.read(4096), b""): 
      hash_md5.update(chunk) 
    return hash_md5.hexdigest() 

？ アドバイスありがとうございます！

Answer 1

バイナリとしてread_dataを指定する必要がありますか？ここ あなたはファイルとユニットテストを実行するためにそれを貼り付けることができ、実施例である：

import hashlib 
from unittest import TestCase 

import mock 


def md5(file_name): 
    hash_md5 = hashlib.md5() 
    with open(file_name, "rb") as f: 
     for chunk in iter(lambda: f.read(4096), b""): 
      hash_md5.update(chunk) 
    return hash_md5.hexdigest() 


class MD5TestCase(TestCase): 
    def test(self): 
     with mock.patch('%s.open' % __name__, mock.mock_open(read_data=b'aaa'), create=True) as m: 
      result = md5("lalala") 
      self.assertEqual(result, '47bce5c74f589f4867dbd57e9ca9f808') 

答えはその答えに基づいています：関数my_script.function_to_get_md5_value（m）が配置されている場所How do I mock an open used in a with statement (using the Mock framework in Python)?