为什么读取 NamedTemporaryFile 时我的断言失败？

Question

我正在编写读取文件并根据该文件的内容创建字典的代码。代码非常简单，但我想测试边缘情况。

这是我的尝试：

from tempfile import NamedTemporaryFile
from nose.tools import *

def read_file(filename):
    with open(filename) as f:
        my_dict = { dict(line.strip().split(',')) for line in f }
    return my_dict

def test_read_file():
    file_contents = b"""Hello,World"""
    with NamedTemporaryFile() as fp:
        fp.write(file_contents)
        my_dict = read_file(fp.name)
    print(my_dict)
    assert my_dict == { "Hello" : "World" }

很遗憾，这个断言失败了，因为my_dict 是一个空字典。

我的理解是，一旦NamedTemporaryFile 被关闭，它就会被销毁，所以我不希望它在read_file 的my_dict 填充后直接 被销毁。 fp 被打开了两次：一次写一次读——这是麻烦制造者吗？

这是测试读取文件的函数的正确方法吗？如果是这样，为什么我的断言失败了？如果不是，那么编写此测试的更好机制是什么？

Answer 1

您需要刷新写入以确保在读取之前写入数据。

def test_read_file():
    file_contents = b"""Hello,World"""
    with NamedTemporaryFile() as fp:
        fp.write(file_contents)
        fp.flush()
        my_dict = read_file(fp.name)
    print(my_dict)
    assert my_dict == { "Hello" : "World" }

Answer 2

您可以使用内存中类似文件的对象，而不是使用真实文件。这需要模拟 open，或者更改您的 API 以采用类似文件的对象而不是文件名。

import unittest.mock
import io

def test_read_file():
    file_contents = io.BytesIO(b"""Hello,World""")
    m = unittest.mock.mock_open(read_data=file_contents)
    with unittest.mock.patch('__main__.open', m):
        my_dict = read_file("fake.txt")
    print(my_dict)
    assert my_dict == { "Hello" : "World" }