U
    |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   c/var/www/dating/data/www/fatepal.com/env/lib/python3.8/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textZ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_dbZ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   s0   | j   | tj | j  W 5 Q R X d S r   )r    r%   ZassertRaisesr   Z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   Zfixturesr   r'   r(   __classcell__r   r   r#   r   r      s   r   N)	Zdjango.contrib.authr   Zdjango.testr   Zwagtail.modelsr   r   r   r   r   r   r   r   <module>   s   