Django更新表单的单元测试

Question

我不知道如何管理表单和相关单元测试的更新，我非常感谢您的建议=）

我有一个公司模型和相关的非常简单的公司形式：

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 = []

视图也非常简单：

@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())

这个视图工作得很好，我可以更新信息（它实际上不用于创建，这要感谢Django管理面板）

不幸的是，我无法构建能够确保更新工作的单元测试
我试过两种方法，但都不管用。 我的第一次尝试是：

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")

在本例中，除了最新的断言之外，一切都正常。看起来更新尚未保存，事实并非如此。我之前尝试读取数据库，密钥存储在上下文中，但它保持不变

当我找到this topic时，我正在寻找其他信息，所以我尝试了另一种测试方法，即使这种方法让我有点吃惊（我不知道视图实际上是如何测试的）。
这是我的第二次尝试（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")

在这种情况下，表单就是空的

<>我可以考虑我的观点并继续下去，我的问题是我在另一个视图中有一个bug，我想确保我可以通过测试来发现bug。p>

非常感谢您的回答

编辑-8月30日
根据建议，我尝试使用self.company.refresh_from_db()，但没有改变结果。
已尝试传递self.client.post()中的所有字段，但一旦字段为空（'无法将任何字段编码为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")

以下是为测试创建“虚拟”copany的函数：

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

在这种情况下，在视图和表单更新完对象后，需要使用^{}来“刷新”对象。这意味着当您当前正在断言时，您正在使用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")

编辑：

找到了一个让这一切顺利的方法。由于表单要求您输入所有数据，您只需将公司实例传递给同一表单，并访问initial（它将作为您的请求数据）

然后，您可以使用所需的更改对其进行修改，在本例中为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")

这是我的作品，并通过与您相同的精确模型