我正在为脚本中的一个函数编写单元测试,但是我在创建模拟文件时遇到了麻烦。我的函数接受一个文件名并输出它的MD5值。
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()
目前我的单元测试:
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值的模拟文件的正确方法吗? 感谢您的建议!