폼으로 댓글 기능 구현하기[3]

# /blog/tests.py
# (...생략...)
def test_comment_form(self):
    # setUp() 함수에서 생성한 Comment가 1개인 것을 확인한다.
    self.assertEqual(Comment.objects.count(), 1)
    # 이 댓글은 self.post_001에 달린 댓글이므로 post_001의 댓글 개수도 1개인 것을 확인한다.
    self.assertEqual(self.post_001.comment_set.count(), 1)

    # 로그인하지 않은 상태 테스트
    response = self.client.get(self.post_001.get_absolute_url())
    self.assertEqual(response.status_code, 200)
    soup = BeautifulSoup(response.content, 'html.parser')

    # id가 'comment-area'인 div 요소를 찾아 comment-area에 저장한다.
    comment_area = soup.find('div', id='comment-area')
    # 로그인하지 않은 상태이므로 'Log in and leave a comment'라는 문구가 보여야 한다.
    self.assertIn('Log in and leave a comment', comment_area.text)
    # 로그인하지 않은 상태이므로 id가 comment-form인 요소는 존재하지 않아야 한다.
    self.assertFalse(comment_area.find('form', id='comment-form'))

# /blog/test.py
# (...생략...)
def test_comment_form(self):
    # (...생략...)
    # 로그인한 상태 테스트
    self.client.login(username='obama', password='somepassword')
    response = self.client.get(self.post_001.get_absolute_url())
    self.assertEqual(response.status_code, 200)
    soup = BeautifulSoup(response.content, 'html.parser')

    comment_area = soup.find('div', id='comment-area')
    # 로그인한 상태이므로 로그인을 요구하는 문구는 보이지 않는다.
    self.assertNotIn('Log in and leave a comment', comment_area.text)

    # 로그인한 상태이므로 댓글 폼이 보이고, 그 안에 textare도 있다.
    comment_form = soup.find('form', id='comment-form')
    self.assertTrue(comment_form.find('textarea',id='id_content'))
    # POST 방식으로 댓글 내용을 서버에 보내고, 결과를 response에 담는다.
    # follow=True 옵션은 POST로 요청을 보내는 경우 서버에서 처리한 후 리다이렉트되는데, 이에 따라가도록 설정하는 역할을 한다.
    response = self.client.post(
        self.post_001.get_absolute_url() + 'new_comment/',
        {
            'content': "오바마의 댓글입니다.",
        },
        follow=True
    )

    self.assertEqual(response.status_code, 200)

    # 댓글이 하나 추가되었으므로 전체 댓글 개수는 2개임을 확인한다.
    self.assertEqual(Comment.objects.count(), 2)
    # 추가된 댓글이 self.post_001의 것이므로 해당 post의 댓글 개수도 2개임을 확인한다.
    self.assertEqual(self.post_001.comment_set.count(), 2)

    # 마지막으로 생성된 comment를 가져온다.
    new_comment = Comment.objects.last()

    # POST요청을 보낼 때 follow=True로 설정했기 때문에 서버에서 처리된 뒤 comment가 달린 포스트의 상세 페이지로 리다이렉트된다.
    # 그래서 웹 브라우저의 타이틀로 새로 만든 comment가 달린 포스트의 타이틀이 나타난다.
    soup = BeautifulSoup(response.content, 'html.parser')
    self.assertIn(new_comment.post.title, soup.title.text)

    # 새로 만든 댓글의 내용과 작성자가 표시되는 것을 확인한다.
    comment_area = soup.find('div', id='comment-area')
    new_comment_div = comment_area.find('div', id=f'comment-{new_comment.pk}')
    self.assertIn('obama', new_comment_div.text)
    self.assertIn('오바마의 댓글입니다.', new_comment_div.text)

<!-- /blog/templates/blog/post_detail.html-->
<!-- ...생략... -->
<!-- Comments section-->
<div class="mb-5" id="comment-area">
    <div class="card bg-light">
        <div class="card-body">
            <!-- Comment form-->
            {% if user.is_authenticated %}
                <form class="mb-4" id="comment-form" method="POST" action="{{ post.get_absolute_url }}new_comment/">
                    <textarea class="form-control" rows="3" placeholder="Join the discussion and leave a comment!"></textarea>
                </form>
            {% else %}
                <a role="button" class="btn btn-outline-dark btn btn-sm col-12" href="#" data-toggle="modal" data-target="#loginModal">
                Log in and leave a comment
                </a>
            {% endif %}
<!-- ...생략... -->

#/blog/forms.py
from .models import Comment
from django import forms

class CommentForm(forms.ModelForm):
    class Meta:
        model = Comment
        fields = ('content', )
        # fields(포함시키기) 대신 exclude(제외시키기)를 사용할 수도 있다 

# /blog/views.py
from django.contrib.auth.mixins import LoginRequiredMixin, UserPassesTestMixin
from django.core.exceptions import PermissionDenied
from django.shortcuts import render, redirect
from django.utils.text import slugify
from django.views.generic import ListView, DetailView, CreateView, UpdateView

# CommentForm을 임포트한다.
from .forms import CommentForm
from .models import Post, Category, Tag

# (... 생략 ...)
# PostDetail 클래스의 get_context_data() 함수에서 CommentForm을 comment_form이라는 이름으로 넘긴다.
class PostDetail(DetailView):
    model = Post

    def get_context_data(self, **kwargs):
        context = super(PostDetail, self).get_context_data()
        context['categories'] = Category.objects.all()
        context['no_category_post_count'] = Post.objects.filter(category=None).count()
        context['comment_form'] = CommentForm

        return context
# (... 생략 ...)

<!-- /blog/templates/post_detail.html-->
{% extends 'blog/base.html' %}
{% load crispy_forms_tags %}
<!-- (...생략...) -->
<!-- Comment form-->
    {% if user.is_authenticated %}
        <form class="mb-4" id="comment-form" method="POST" action="{{ post.get_absolute_url }}new_comment/">
            {{ comment_form | crispy }}
            <button type="submit" class="btn btn-primary">Submit</button>
        </form>

# /blog/urls.py
from django.contrib import admin
from django.urls import path
from . import views

urlpatterns = [
    path('update_post/<int:pk>/', views.PostUpdate.as_view()),
    path('create_post/', views.PostCreate.as_view()),
    path('tag/<str:slug>/', views.tag_page),
    path('category/<str:slug>/', views.category_page),
    path('<int:pk>/new_comment/', views.new_comment),
    path('<int:pk>/', views.PostDetail.as_view()),
    path('', views.PostList.as_view()),
]

from django.contrib.auth.mixins import LoginRequiredMixin, UserPassesTestMixin
from django.core.exceptions import PermissionDenied
# get_object_or_404를 임포트한다.
from django.shortcuts import render, redirect, get_object_or_404
from django.utils.text import slugify
from django.views.generic import ListView, DetailView, CreateView, UpdateView

from .forms import CommentForm
from .models import Post, Category, Tag

# (... 생략 ...)

def new_comment(request, pk):
    # 로그인하지 않은 상태로 댓글을 작성하려 하면 PermissionDenied를 발생시킨다.
    if request.user.is_authenticated:
        # url에서 인자로 받은 pk로 댓글을 추가할 post를 가져오는데, 실패할 시 404에러를 발생시키도록 get_object_or_404를 사용한다.
        post = get_object_or_404(Post, pk=pk)

        # POST방식이 아닌 다른 method로 접근했을 경우 그냥 해당 포스트의 상세 페이지로 리다이렉트 되도록 한다.
        if request.method == 'POST':
            # POST방식으로 들어온 정보를 CommentForm의 형태로 가져온다.
            comment_form = CommentForm(request.POST)
            # 폼이 유효하게 작성되었다면 해당 내용으로 새로운 레코드를 만들어 DB에 저장한다.
            # 이때 commit=False 옵션으로 바로 저장하는 기능을 잠시 미루고 comment_form에 담긴 정보로 Comment 인스턴스만 가져온다.
            # 이때 CommentForm은 content 필드의 내용만 담고 있으므로 필요한 나머지 정보들을 모두 채운뒤 save()를 호출하여 저장한다.
            if comment_form.is_valid():
                comment = comment_form.save(commit=False)
                comment.post = post
                comment.author = request.user
                comment.save()
                # 처리가 끝나면 comment의 URL로 리다이렉트한다.
                return redirect(comment.get_absolute_url())
        else:
            return redirect(post.get_absolute_url())
    else:
        raise PermissionDenied

<!-- /blog/templates/post_detail.html-->
<!-- Comment form-->
{% if user.is_authenticated %}
    <form class="mb-4" id="comment-form" method="POST" action="{{ post.get_absolute_url }}new_comment/">
        {% csrf_token %}
        {{ comment_form | crispy }}