o
    h|                     @   sL   d dl mZ d dlmZ d dlmZmZ G dd dZG dd deeZdS )	    )get_user_model)TestCase)CommentPagec                   @   s   e Zd Zdd Zdd ZdS )CommentTestingUtilsc                 C   s,   t jjdd| _| j | _| j | _d S )Nz!Welcome to the Wagtail test site!)title)r   objectsgetpagesave_revision
revision_1
revision_2self r   [/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/tests/test_comments.pysetUp   s   zCommentTestingUtils.setUpc                 C   s    t jj| jt j dd|dS )Ntestr   )r
   usertextcontentpathrevision_created)r   r   creater
   r   first)r   r   r   r   r   create_comment   s   
z"CommentTestingUtils.create_commentN)__name__
__module____qualname__r   r   r   r   r   r   r      s    r   c                       s2   e Zd ZdgZ fddZdd Zdd Z  ZS )TestRevisionDeletionz	test.jsonc                    s6   t    | j | _| | j| _| | j| _d S N)	superr   r
   r   
revision_3r   r   old_commentnew_commentr   	__class__r   r   r      s   
zTestRevisionDeletion.setUpc                 C   s*   | j   | j  | | jj| j d S r   )r   deleter"   refresh_from_dbassertEqualr   r   r   r   r   r   Btest_deleting_old_revision_moves_comment_revision_created_forwards    s   

zWTestRevisionDeletion.test_deleting_old_revision_moves_comment_revision_created_forwardsc                 C   sF   | j   | tj | j  W d    d S 1 sw   Y  d S r   )r!   r&   assertRaisesr   DoesNotExistr#   r'   r   r   r   r   ;test_deleting_most_recent_revision_deletes_created_comments'   s   
"zPTestRevisionDeletion.test_deleting_most_recent_revision_deletes_created_comments)r   r   r   fixturesr   r)   r,   __classcell__r   r   r$   r   r      s
    r   N)	django.contrib.authr   django.testr   wagtail.modelsr   r   r   r   r   r   r   r   <module>   s
    