[英]How can I pass a file to unittest.mock.mock_open()?
我有一个读取日志文件并筛选结果的功能,并且我想测试以确保其正确筛选。
我的密码
import os
import random
import unittest
from unittest.mock import patch, mock_open
__SAMPLE_LOG__ = os.path.join(settings.BASE_DIR, "apps/tests/log_viewer/sample_logs/sample_manager_log.log")
class FilterLog(unittest.TestCase):
def setUp(self):
with open(__SAMPLE_LOG__) as f:
self.sample_data = f.read()
@patch('builtins.open', new_callable = mock_open, read_data = self.sample_data)
def test_filterDate(self, mock_file):
day = '08'
month = '08'
year = '2019'
results = filter_log(filter_by = 'date', day = day, month = month, year = year)
self.assertEqual(open(settings.ACTIVITY_LOG_FILE).read(), self.sample_data)
错误
@patch('builtins.open',new_callable =模拟打开,read_data = self.sample_data)
NameError:名称“ self”未定义
我的问题
我应该如何将数据传递给mock_open()
? 我觉得在文件顶部with open() ... read()
是一种不好的做法,也不能将其设置为类变量(可以吗?),那我有什么选择呢?
文件说什么
从文档中 read_data
接受一个字符串,因此我需要以某种方式将文件读取到一个变量中并传递给它。但是读取文件在哪里合适呢? 在模块的顶部,在类的开头,还是在setUp()
?
这应该工作。 我将sample_data带出课堂。
import os
import random
import unittest
from unittest.mock import patch, mock_open
__SAMPLE_LOG__ = os.path.join(settings.BASE_DIR, "apps/tests/log_viewer/sample_logs/sample_manager_log.log")
# read your test data in sample_data
with open(__SAMPLE_LOG__) as f:
sample_data = f.read()
class FilterLog(unittest.TestCase):
@patch('builtins.open', new_callable = mock_open, read_data = sample_data)
def test_filterDate(self, mock_file):
day = '08'
month = '08'
year = '2019'
results = filter_log(filter_by = 'date', day = day, month = month, year = year)
self.assertEqual(open(settings.ACTIVITY_LOG_FILE).read(), sample_data)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.