Question

我正在尝试在我的简单论坛应用程序中测试用于创建Post模型的表单。我遇到的问题是，尝试在测试中保存表单后，出现错误NOT NULL constraint failed: forum_web_post.user_id，因为我在视图的form_valid()方法中分配了用户。由于创建帖子的用户是发送请求的登录用户，因此不会通过表单传递用户。

models.py

class Post(models.Model):
    category = models.ForeignKey(Category, on_delete=models.PROTECT)
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    title = models.CharField(max_length=100)
    text = models.TextField()
    created_at = models.DateTimeField(auto_now=True)

用户从django.contrib.auth.models导入，类别模型如下所示。

class Category(models.Model):
    name = models.CharField(max_length=100)
    created_at = models.DateTimeField(auto_now=True)

在views.py中，用户提交表单后，他将被重定向到其个人资料页面

views.py

class PostCreate(generic.CreateView):
    model = Post
    form_class = PostForm
    template_name = 'forum_web/post_edit.html'

    def form_valid(self, form):
        post = form.save(commit=False)
        post.user = models.User.objects.get(id=self.request.user.id)
        post.save()
        return HttpResponseRedirect(reverse_lazy('forum:user_detail', kwargs={'pk': self.request.user.id}))

forms.py

class PostForm(ModelForm):
    class Meta:
        model = Post
        fields = ['category', 'title', 'text']

tests.py

    def test_post_form_create_should_pass(self):
        # this dict is in setUp() as well as the test_category but for simplicity I will include it in this method
        post_data = {
        'category': self.test_category.pk,
        'title': 'test_post',
        'text': 'test_post_text'
        }
        post_count = Post.objects.count()
        form = PostForm(data=self.post_data)
        self.assertTrue(form.is_valid())
        form.save()
        self.assertEqual(post_count + 1, Post.objects.count())

任何帮助将不胜感激！

Answer 1

您正在测试表单。因此，在测试中调用视图函数没有任何意义。在将表单保存到测试中之前，应该使用form.instance.user = ...手动将用户分配给表单的实例，因为这是使用表单的方式。

此外，您应该通过实际登录用户并将请求发布到PostCreate视图来测试您的视图。在这里，您正在测试视图是否正确保存了表单（并且您的form_valid()方法可以正常工作）。

注意：post.user = self.request.user比使用id从数据库重新获取用户更好。是多余的。

我可以以某种方式在Django测试中从form_valid（）调用逻辑吗？

1 个答案: