在Go语言中,如何为单元测试创建内存文件?
在Python中,我使用io.BytesIO
或io.StringIO
测试从文件阅读或写入文件。
def test_parse_function():
infile = io.StringIO('''\
line1
line2
line3
''')
parsed_contents = parse_function(infile)
expected_contents = ['line1', 'line2', 'line3'] # or whatever is appropriate
assert parsed_contents == expected_contents
类似地,对于文件输出,我将得到如下所示的内容:
def test_write_function():
outfile = io.StringIO()
write_function(outfile, ['line1', 'line2', 'line3'])
outfile.seek(0)
output = outfile.read()
expected_output = '''\
line1
line2
line3
'''
assert output == expected_output
3条答案
按热度按时间yxyvkwin1#
可以使用缓冲区。
一般来说,在代码中使用io.Reader和io.Writer接口(Buffer实现了这两个接口)来处理IO是一个好主意。这样,你就可以用同样的方式处理各种输入/输出方法(本地文件、内存缓冲区、网络连接......),而不必知道你在使用的特定函数中处理的是什么。这使得它更加抽象,测试也变得微不足道。
使用普通函数的示例:
功能定义:
程序中的函数使用:
试验:
qybjjes12#
如果您需要
io.ReadSeeker
,并且不需要写访问权限,请使用bytes.Reader
:这对于
http.ServeContent()
之类的东西很有用。或者,更简单地说:
wydwbb8l3#
正如在“Pragmatic and Effective Testing in Go“中提到的,您可以使用
spf13/afero
进行文件抽象,从而简化测试。该库有一个使用Afero进行测试:
使用模拟文件系统进行测试有很大的好处。
它在每次初始化时都处于完全空白状态,并且无论操作系统如何,都可以轻松再现。
你可以创建文件,你的心的内容和文件访问将是快速的,同时也节省了你从所有恼人的问题,删除临时文件,Windows文件锁定等。
MemMapFs后端非常适合测试。
rm -rf /
”信心十足它使用完全原子内存支持的文件系统(
MemMapFs
)。它是完全并发的,可以在go例程中安全地工作。