【发布时间】:2017-10-03 04:15:48
【问题描述】:
这是一些单元测试,我已将其简化为重现问题的最小示例。
import unittest
import zipfile
from unittest.mock import patch
"""My 'production' function"""
def list_txt_in_zipfile(zip_file_path):
with open(zip_file_path, "rb") as file:
zf = zipfile.ZipFile(file)
print("zipfile.ZipFile is", zipfile.ZipFile)
return [f_name for f_name in zf.namelist() if f_name.endswith(".txt")]
class UnzipperTest(unittest.TestCase):
"""My test method"""
@patch("__main__.zipfile.ZipFile")
def test_list_txt_in_zipfile(self, mock_ZipFile):
txt_list = list_txt_in_zipfile("my.txt.zip")
mock_ZipFile.assert_any_call() # AssertionError
print("mock_ZipFile is", mock_ZipFile)
if __name__ == '__main__':
unittest.main()
断言错误:
AssertionError: ZipFile() call not found
当我的测试方法调用list_txt_in_zipfile 时,它会记录一个模拟对象,而不是真正的zipfile.Zipfile(来自Python 的zipfile.py)。
如果我在生产函数中打印这“两个”对象zipfile.ZipFile,在我的测试方法中打印mock_ZipFile,它们就是同一个对象:
zipfile.ZipFile is <MagicMock name='ZipFile' id='3069297420'>
mock_ZipFile is <MagicMock name='ZipFile' id='3069297420'>
为什么mock_ZipFile 无法断言对它的任何调用?
我实际上已经成功测试了 list_txt_in_zipfile 函数,方法是将其拆分为两个函数(open() 部分和 zipfile.ZipFile() 的其余部分,但就目前而言,这不是一个巨大的函数,所以我会喜欢把这段逻辑放在一个地方。
更新:测试方法中的my.txt.zip 确实存在,我没有设法引入unittest.mock.mock_open 来模拟open(),因为我一直在模拟zipfile.ZipFile。
【问题讨论】:
标签: python unit-testing patch python-decorators assertion