댓글 삭제 기능 추가하기

def test_delete_comment(self):
    # Trump로 로그인하여 댓글을 작성한다.
    comment_by_trump = Comment.objects.create(
        post=self.post_001,
        author=self.user_trump,
        content='트럼프의 댓글입니다.'
    )

    # setUp()에서 obama 계정으로 댓글을 하나 만들었으니, 총 댓글은 2개이며 post_001의 댓글 수도 2개이다.
    self.assertEqual(Comment.objects.count(), 2)
    self.assertEqual(self.post_001.comment_set.count(), 2)

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

    # 로그인하지 않은 상태에서는 삭제 버튼이 보이면 안된다.
    # 버튼의 id는 'comment-pk-delete-btn'으로 한다.
    comment_area = soup.find('div', id='comment-area')
    self.assertFalse(comment_area.find('a', id='comment-1-delete-btn'))
    self.assertFalse(comment_area.find('a', id='comment-2-delete-btn'))

    # trump로 로그인한 상태
    self.client.login(username='trump', password='somepassword')
    response = self.client.get(self.post_001.get_absolute_url())
    self.assertEqual(response.status_code, 200)
    soup = BeautifulSoup(response.content, 'html.parser')

    # trump로 로그인한 상태이므로 obama가 작성한 댓글에 대한 삭제 버튼은 보이지 않아야한다.
    comment_area = soup.find('div', id='comment-area')
    self.assertFalse(comment_area.find('a', id='comment-1-delete-btn'))
    # 반면 pk=2에 대한 삭제 버튼은 있어야 한다.
    comment_002_delete_modal_btn = comment_area.find(
        'a', id='comment-2-delete-modal-btn'
    )
    self.assertIn('delete', comment_002_delete_modal_btn)
    self.assertEqual(
        comment_002_delete_modal_btn.attrs['data-target'],
        '#deleteCommentModal-2'
    )

    # 삭제를 할지 물어보는 모달에는 'Are You Sure?'이라는 문구와 함께 delete 버튼이 있어야 한다.
    # 이 버튼의 링크는 '/blog/delete_comment/삭제할 comment의 pk'로 되어 있어야 한다.
    delete_comment_modal_002 = soup.find('div', id='deleteCommentModal-2')
    self.assertIn('Are You Sure?', delete_comment_modal_002.text)
    really_delete_btn_002 = delete_comment_modal_002.find('a')
    self.assertIn('Delete', really_delete_btn_002.text)
    self.assertEqual(
        really_delete_btn_002.attrs['href'],
        '/blog/delete_comment/2/'
    )

    # delete 버튼을 클릭하면 댓글이 삭제되고 post_001의 페이지로 리다이렉트된다.
    # 해당 페이지에는 더이상 트럼프의 댓글이 존재하지 않고, 댓글의 개수도 1로 줄어든다.
    response = self.client.get('/blog/delete_comment/2/', follow=True)
    self.assertEqual(response.status_code, 200)
    soup = BeautifulSoup(response.content, 'html.parser')
    self.assertIn(self.post_001.title, soup.title.text)
    comment_area = soup.find('div', id='comment-area')
    self.assertNotIn('트럼프의 댓글입니다.', comment_area.text)

    self.assertEqual(Comment.objects.count(), 1)
    self.assertEqual(self.post_001.comment_set.count(), 1)

<!-- /blog/templates/post_detail.html -->

<!-- ... 생략 ... -->

{% if user.is_authenticated and comment.author == user %}
    <div class="float-end">
        <a role="button" class="btn btn-sm btn-info float-end" id="comment-{{ comment.pk }}-update-btn" href="/blog/update_comment/{{ comment.pk }}/">
            edit
        </a>
        <a role="button" class="btn btn-sm btn-danger" id="comment-{{ comment.pk }}-delete-modal-btn" href="#" data-toggle="modal" data-target="#deleteCommentModal-{{ comment.pk }}">
            delete
        </a>
    </div>
    <!-- Modal -->
    <div class="modal fade" id="deleteCommentModal-{{ comment.pk }}" tabindex="-1" role="dialog" aria-labelledby="deleteCommentModalLabel" aria-hidden="true">
        <div class="modal-dialog" role="document">
            <div class="modal-content">
                <div class="modal-header">
                    <h5 class="modal-title" id="deleteModalLabel">Are You Sure?</h5>
                    <button type="button" class="btn-close" data-dismiss="modal" aria-label="Close">
                        <span aria-hidden="true">&times;</span>
                    </button>
                </div>
                <div class="modal-body">
                    <del>{{ comment | linebreaks }}</del>
                </div>
                <div class="modal-footer">
                    <button type="button" class="btn btn-secondary" data-dismiss="modal">Cancel</button>
                    <a role="button" class="btn btn-danger" href="/blog/delete_comment/{{ comment.pk }}/">Delete</a>
                </div>
            </div>
        </div>
    </div>
{% endif %}

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

urlpatterns = [
    path('delete_comment/<int:pk>/', views.delete_comment),
    path('update_comment/<int:pk>/', views.CommentUpdate.as_view()),
    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()),
]

# /blog/views.py

# (...생략...)

def delete_comment(request, pk):
    comment = get_object_or_404(Comment, pk=pk)
    post = comment.post
    if request.user.is_authenticated and request.user == comment.author:
        comment.delete()
        return redirect(post.get_absolute_url())
    else:
        raise PermissionDenied