我不明白如何管理表单和相关单元测试的更新,我真的很感激一些建议 =)
我有一个 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",
)
最佳答案
在这种情况下,您需要使用 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")
这有效并以您拥有的完全相同的模型传递给我。
关于python - Django 更新表单的单元测试,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/68971787/