a
    Dg|                     @   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   Z/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/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_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   sD   | j   | tj | j  W d    n1 s60    Y  d S r   )r    r%   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   