您如何在Django测试中处理文件

Question

我有一个具有文件字段的模型，并且我或多或少都在测试该模型，如下所示：

def test_importing_file_and_processing(self):
    file_path = os.path.join(settings.BASE_DIR,
                             'myapp/tests/reports/test_input_file')
    report = IngestedFile(
        filename='test_file'
    )
    report.document.save('test_file', File(open(file_path)))
    report.save()
    report.do_stuff()
    self.assertStuf....

有没有更好的方法来测试文件？ 我测试事物的方式确实有些古怪。

Answer 1

您可以使用tempfile模块

import tempfile

tmp_file = tempfile.NamedTemporaryFile(suffix='.pdf')

with open(tmp_file.name, 'w') as f:
    f.write('test_content')

report.document = tmp_file

或者您可以为此创建隔离功能

def generate_test_file(file_format='.pdf'):

    tmp_file = tempfile.NamedTemporaryFile(suffix=file_format)

    with open(tmp_file.name, 'w') as f:
        f.write('test_content')

    return tmp_file