python - Django 更新表单的单元测试

Question

我不明白如何管理表单和相关单元测试的更新，我真的很感激一些建议 =)
我有一个 Company 模型，以及相关的非常简单的 CompanyForm:

class Company(models.Model):
    """
    Company informations
    - Detailed information for display purposes in the application
      but also used in documents built and sent by the application
    - Mail information to be able to send emails
    """
    company_name = models.CharField("nom", max_length=200)
    comp_slug = models.SlugField("slug")
    logo = models.ImageField(upload_to="img/", null=True, blank=True)
    use_groups = models.BooleanField("utilise les groupes", default=False)   # Company uses groups or not
    rules = [("MAJ", "Majorité"), ("PROP", "Proportionnelle")]   # Default management rule
    rule = models.CharField(
        "mode de scrutin", max_length=5, choices=rules, default="MAJ"
    )
    upd_rule = models.BooleanField("choisir la règle de répartition pour chaque événement", default=False)     # Event rule might change from one to another or always use default
    statut = models.CharField("forme juridique", max_length=50)
    siret = models.CharField("SIRET", max_length=50)
    street_num = models.IntegerField("N° de rue", null=True, blank=True)
    street_cplt = models.CharField("complément", max_length=50, null=True, blank=True)
    address1 = models.CharField("adresse", max_length=300)
    address2 = models.CharField(
        "complément d'adresse", max_length=300, null=True, blank=True
    )
    zip_code = models.IntegerField("code postal")
    city = models.CharField("ville", max_length=200)
    host = models.CharField("serveur mail", max_length=50, null=True, blank=True)
    port = models.IntegerField("port du serveur", null=True, blank=True)
    hname = models.EmailField("utilisateur", max_length=100, null=True, blank=True)
    fax = models.CharField("mot de passe", max_length=50, null=True, blank=True)
    use_tls = models.BooleanField("authentification requise", default=True, blank=True)

    class Meta:
        verbose_name = "Société"
        constraints = [
            models.UniqueConstraint(fields=["comp_slug"], name="unique_comp_slug")
        ]

    def __str__(self):
        return self.company_name

    @classmethod
    def get_company(cls, slug):
        """ Retreive company from its slug """
        return cls.objects.get(comp_slug=slug)


class CompanyForm(forms.ModelForm):
    company_name = forms.CharField(label="Société", disabled=True)

    class Meta:
        model = Company
        exclude = []

View 也很简单:

@user_passes_test(lambda u: u.is_superuser or u.usercomp.is_admin)
def adm_options(request, comp_slug):
    '''
        Manage Company options
    '''
    company = Company.get_company(comp_slug)
    comp_form = CompanyForm(request.POST or None, instance=company)

    if request.method == "POST":
        if comp_form.is_valid():
            comp_form.save()

    return render(request, "polls/adm_options.html", locals())

这个 View 工作正常，我可以更新信息(它实际上不用于创建，这要归功于 Django 管理面板)。
不幸的是，我无法构建单元测试来确保更新有效!
我尝试了 2 种方法，但都没有奏效。
我的第一次尝试如下:

class TestOptions(TestCase):
    def setUp(self):
        self.company = create_dummy_company("Société de test")
        self.user_staff = create_dummy_user(self.company, "staff", admin=True)
        self.client.force_login(self.user_staff.user)

    def test_adm_options_update(self):
        # Load company options page
        url = reverse("polls:adm_options", args=[self.company.comp_slug])
        response = self.client.get(url)
        self.assertEqual(response.status_code, 200)
        self.assertContains(response, "0123456789")
        self.assertEqual(self.company.siret, "0123456789")

        # Options update
        response = self.client.post(
            reverse("polls:adm_options", args=[self.company.comp_slug]),
            {"siret": "987654321"}
        )
        self.assertEqual(response.status_code, 200)
        self.assertContains(response, "987654321")
        self.assertNotContains(response, "0123456789")
        self.assertEqual(self.company.siret, "987654321")

在这种情况下，除了最新的断言，一切正常。看起来更新还没有保存，其实不然。我之前尝试读取数据库， key 存储在上下文中，但它保持不变。
当我找到 this topic 时，我正在寻找其他信息，所以我尝试了另一种测试方法，即使这种方法让我感到有些惊讶(我没有看到实际是如何测试 View 的)。
这是我的第二次尝试( setUp() 保持不变):

    def test_adm_options_update(self):
        # Load company options page
        url = reverse("polls:adm_options", args=[self.company.comp_slug])
        response = self.client.get(url)
        self.assertEqual(response.status_code, 200)
        self.assertContains(response, "0123456789")         # this is the default value in tests for this field
        self.assertEqual(self.company.siret, "0123456789")

        # Options update
        self.company.siret = "987654321"
        comp_form = CompanyForm(instance=self.company)
        self.assertTrue(comp_form.is_valid())
        comp_form.save()
        company = Company.get_company(self.company.comp_slug)
        self.assertEqual(company.siret, "987654321")

在这种情况下，表单只是空的!
我可以考虑我的 View 工作并继续，我的问题是我在另一个 View 中有一个错误，我想确保我可以构建测试来找出错误!
非常感谢您的回答!
编辑 - 8 月 30 日
按照建议，我尝试使用 self.company.refresh_from_db() 但它没有改变结果。
尝试传递 self.client.post() 中的所有字段，但一旦字段为空就会失败(“无法将 None 编码为 POST 数据”错误消息)
看来我还创建了一个“虚拟”公司用于测试，其中必填字段为空……无论如何它都有效。测试环境的问题？我改变了这一点，但我想知道问题是否不在其他任何地方......
编辑 - 9 月 15 日
寻找可以为我提供新想法的人，请=)
为了确保我理解最新的命题，这里是测试的完整代码:

def test_adm_options_update(self):
    # Load company options page
    url = reverse("polls:adm_options", args=[self.company.comp_slug])
    response = self.client.get(url)
    self.assertEqual(response.status_code, 200)
    self.assertContains(response, "0123456789")
    self.assertEqual(self.company.siret, "0123456789")

    # Apply changes
    company_data = copy.deepcopy(CompanyForm(instance=self.company).initial)
    company_data['siret'] = "987654321"
    response = self.client.post(
        reverse("polls:adm_options", args=[self.company.comp_slug]),
        company_data,
        )
    self.company.refresh_from_db()
    self.assertEqual(response.status_code, 200)
    self.assertContains(response, "987654321")
    self.assertNotContains(response, "0123456789")
    self.assertEqual(self.company.siret, "987654321")

这是为测试创建“虚拟”公司的函数:

def create_dummy_company(name):
    return Company.objects.create(
        company_name=name,
        comp_slug=slugify(name),
        logo=SimpleUploadedFile(name='logo.jpg', content=b'content', content_type='image/jpeg'),
        statut="SARL",
        siret="0123456789",
        address1="Rue des fauvettes",
        zip_code="99456",
        city='Somewhere',
        host="smtp.gmail.com",
        port=587,
        hname="test@polls.com",
        fax="toto",
    )

Answer 1

在这种情况下，您需要使用 refresh_from_db 一旦 View 和表单完成更新您的对象，就“刷新”您的对象。这意味着当您当前断言时，您使用的是 self.company 的“旧快照”。因此断言失败，所以你需要更新它:

        # Options update
        response = self.client.post(
            reverse("polls:adm_options", args=[self.company.comp_slug]),
            {"siret": "987654321"}
        )
        ...
        self.company.refresh_from_db()
        self.assertEqual(self.company.siret, "987654321")

编辑 :
想出了一种方法来完成这项工作。由于表单要求您输入所有数据，因此您只需将公司实例传递给同一个表单，然后访问初始值(将作为您的请求数据)。
然后，您可以使用所需的更改对其进行修改，在本例中为 siret和 logo :

from django.core.files.uploadedfile import SimpleUploadedFile

    def test(self):
        company_data = CompanyForm(instance=self.company).initial
        company_data['logo'] = SimpleUploadedFile(name='somefile', content=b'content', content_type='image/jpeg')
        company_data['siret'] = "987654321"

        response = self.client.post(
            reverse("polls:adm_options", args=[self.company.comp_slug]),
            company_data,
        )

        self.company.refresh_from_db()
        self.assertEqual(self.company.siret, "987654321")

这有效并以您拥有的完全相同的模型传递给我。