Django视图转CBV及测试求助：评论添加视图转换与测试疑问

嘿，我来帮你搞定这个问题！先从把添加评论的函数视图（FBV）转成类视图（CBV）开始，然后再聊聊对应的测试用例怎么写。

你的需求是创建Comment实例并关联到指定Article，用Django的CreateView最合适——它是专门用来处理模型实例创建的通用类视图，完美匹配你的场景。

替换原来的函数视图，改成下面的类视图：

from django.views.generic.edit import CreateView
from django.urls import reverse_lazy
from django.shortcuts import get_object_or_404
from .models import Article, Comment
from .forms import CommentForm

class CommentCreateView(CreateView):
    # 指定要操作的模型
    model = Comment
    # 使用你已经定义好的表单
    form_class = CommentForm
    # 指定渲染的模板（和原来FBV的模板一致，不用改模板文件）
    template_name = 'news/add_comment_to_article.html'

    def form_valid(self, form):
        # 表单验证通过后，关联当前评论到对应的文章
        article = get_object_or_404(Article, pk=self.kwargs['pk'])
        form.instance.post = article
        # 调用父类方法完成保存
        return super().form_valid(form)

    def get_success_url(self):
        # 评论保存成功后，跳回对应的文章详情页
        return reverse_lazy('news:article', kwargs={"article_id": self.kwargs['pk']})

关键细节说明： #

• form_valid：这是CreateView的钩子方法，用来在表单验证通过后、保存实例前做自定义操作——这里就是把评论和文章关联起来。
• get_success_url：指定保存成功后的跳转地址，用reverse_lazy而不是reverse，因为类视图加载时URL配置可能还没完全初始化。
• 你的URL配置已经写对了：path('<int:pk>/comment/', views.CommentCreateView.as_view(), name='add_comment_to_article')，刚好传递文章的pk参数给CBV。

测试主要覆盖三个核心场景：GET请求返回表单、POST有效数据创建评论、POST无效数据不创建评论。用Django的TestCase来实现：

from django.test import TestCase
from django.urls import reverse
from .models import Article, Comment, User

class CommentCreateViewTests(TestCase):
    def setUp(self):
        # 每个测试前先创建测试用的用户和文章，避免重复代码
        self.user = User.objects.create_user(username='testuser', password='12345')
        self.article = Article.objects.create(title='Test Article', content='Test content', author=self.user)

    def test_get_comment_form(self):
        # 测试GET请求是否正常返回表单页面
        url = reverse('news:add_comment_to_article', kwargs={'pk': self.article.pk})
        response = self.client.get(url)
        
        self.assertEqual(response.status_code, 200)
        self.assertTemplateUsed(response, 'news/add_comment_to_article.html')
        # 可以检查表单字段是否存在，比如author输入框
        self.assertContains(response, 'author')

    def test_post_valid_comment(self):
        # 测试提交有效数据是否能创建评论并跳转
        url = reverse('news:add_comment_to_article', kwargs={'pk': self.article.pk})
        test_data = {
            'author': 'Test Commenter',
            'text': '这是一条测试评论'
        }
        response = self.client.post(url, test_data)
        
        # 验证是否跳转到文章详情页
        self.assertRedirects(response, reverse('news:article', kwargs={'article_id': self.article.pk}))
        # 验证数据库中是否新增了一条评论
        self.assertEqual(Comment.objects.count(), 1)
        new_comment = Comment.objects.first()
        self.assertEqual(new_comment.author, 'Test Commenter')
        self.assertEqual(new_comment.text, '这是一条测试评论')
        self.assertEqual(new_comment.post, self.article)

    def test_post_invalid_comment(self):
        # 测试提交无效数据（比如text为空，假设text是必填字段）
        url = reverse('news:add_comment_to_article', kwargs={'pk': self.article.pk})
        invalid_data = {
            'author': 'Test Commenter',
            'text': ''  # 空文本，触发表单验证失败
        }
        response = self.client.post(url, invalid_data)
        
        # 验证页面返回200（停留在表单页）
        self.assertEqual(response.status_code, 200)
        # 验证数据库中没有新增评论
        self.assertEqual(Comment.objects.count(), 0)
        # 验证表单返回了错误信息
        self.assertTrue(response.context['form'].errors)

测试场景说明： #

• setUp：初始化测试数据，每个测试方法都会执行这个步骤，保证测试环境干净。
• test_get_comment_form：验证用户访问评论页面时，能正常加载表单。
• test_post_valid_comment：验证合法的评论数据能被正确保存，并且跳回文章页。
• test_post_invalid_comment：验证不合法的数据不会被保存，页面停留在表单页并显示错误。

内容的提问来源于stack exchange，提问作者Sandi