如何在Django中对保存用户输入进行单元测试

Question

我正在做一个小项目，我希望能够测试post方法，如果电子邮件不存在，则保存到数据库中，并为下面的视图呈现到index.html中。我看过YouTube上的视频和博客教程，似乎找不到我想要的答案。我已经包含了模型、视图和我当前的测试。

型号：

class EmailList(models.Model):
email = models.TextField(unique=True)

视图：

def home(request):
# Join Email List
if request.method == 'POST':
if request.POST.get('email'):
email = request.POST.get('email')
if not EmailList.objects.filter(email=email).exists():
emailInput = EmailList()
emailInput.email = email
emailInput.save()
return render(request, 'email/index.html', {})
else:
return render(request, 'email/index.html', {})

到目前为止我的测试：

from django.test import TestCase
from email.models import EmailList   
class HomeViewTest(TestCase):
@classmethod
def setUpTestData(cls):
# Create a new email to test
EmailList.objects.create(email="[e-mail]")

def test_home_view_url_exists_at_desired_location(self):
response = self.client.get('')
self.assertEqual(response.status_code, 200)

def test_home_view_post_request_method(self):
response = self.client.post('', {'email' : '[e-mail]'})
self.assertEqual(response.status_code, 200)

def test_home_view_save_email(self):
self.assertEqual(EmailList.objects.count(), 1)

Answer 1

您不应该在setUpTestData方法中创建EmailList的实例，因为您想检查您的视图是否正确地创建了一个新实例，现在您可以在测试之前手动创建一个新的实例。试试这个：

from django.test import TestCase
from tasckq.models import EmailList   
class HomeViewTest(TestCase):
def test_home_view(self):
# Tests the get method
response = self.client.get('')
self.assertEqual(response.status_code, 200)
self.assertTemplateUsed(response, template_name='email/index.html')

# Tests the post method

email = '[e-mail]'
response = self.client.post('', {'email' : email})
self.assertEqual(response.status_code, 200)
self.assertTemplateUsed(response, template_name='email/index.html')
# A new EmailList instance has been successfully created
self.assertTrue(EmailList.objects.filter(email=email).exists())

response = self.client.post('', {'email' : email})
self.assertEqual(response.status_code, 200)
self.assertTemplateUsed(response, template_name='email/index.html')
# A new EmailList instance has not been created because the email was already in use
self.assertEqual(EmailList.objects.count(), 1)