如何在 Django 和 Python 的单元测试中正确模拟大 XML？

Question

我想在我的 XML 解析器中对一个方法进行单元测试。该方法接受一个 XML 元素，将其解析为 Django 模型对象并返回此对象。

我已经为解析器编写了单元测试，但它们需要一小部分 XML，我可以将这些位粘贴到字符串中，例如：

xml = ElementTree.fromstring('<xml><item>content</item></xml>')

但现在我必须传递一个似乎太大而无法将其存储在单元测试文件本身中的 XML 实体。

我正在考虑将其保存到文件中，然后从中加载，但我不知道将文件放在哪里，并且不违反 Django 关于应用程序结构的约定。

是否有“Django”或“pythonic”方式来模拟这个 XML？

Answer 1

我通常会创建一个fixtures 文件夹（你可以在你的Django 设置文件中进行配置）。 这通常用于 json 固定装置，但也可以在其中添加 XML 文件。 您可以通过 unittest 提供的 setUp 方法 (https://docs.python.org/3/library/unittest.html#module-unittest) 加载和读取这些 XML 文件。 然后就像在项目中一样使用它。 一个简单的例子：

import os
from django.test import TestCase
from django.conf import settings
import xml.etree.ElementTree as ET

# Configure your XML_FILE_DIR inside your settings, this can be the
# same dir as the FIXTURE_DIR that Django uses for testing.
XML_FILE_DIR = getattr(settings, 'XML_FILE_DIR')


class MyExampleTestCase(TestCase):

    def setUp(self):
        """
        Load the xml files and pass them to the parser.
        """
        test_file = os.path.join(XML_FILE_DIR, 'my-test.xml')
        if os.path.isfile(test_file):
            # Through this now you can reffer to the parser through
            # self.parser.
            self.parser = ET.parse(test_file)
            # And of course assign the root as well.
            self.root = self.parser.getroot()