a
    Dg                     @   s,   d dl mZ d dlmZ G dd deZdS )    )TestCase)
Collectionc                   @   sD   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dS )TestCollectionTreeOperationsc                 C   s8   t  | _| jjdd| _| jjdd| _| j  d S )NzHoliday photosnamez
Evil plans)r   Zget_first_root_noderoot_collection	add_childholiday_photos_collectionevil_plans_collectionrefresh_from_dbself r   b/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/wagtail/tests/test_collection_model.pysetUp   s    
z"TestCollectionTreeOperations.setUpc                 C   s   | j j}| jj}| jjdd}| || j j | || jj | j   | j  | || j j | || jj | |j| j j | |j| jj d S )NAlphar   )	r
   pathr	   r   r   assertEqualr   ZassertNotEqualZ
assertLess)r   Zold_evil_pathZold_holiday_pathZalpha_collectionr   r   r   test_alphabetic_sorting   s    

z4TestCollectionTreeOperations.test_alphabetic_sortingc                 C   sL   |  t| j d| jg |  t| jjddd| j| jg d S Nr   TZ	inclusive)r   listr	   Zget_ancestorsorder_byr   r   r   r   r   test_get_ancestors$   s    
z/TestCollectionTreeOperations.test_get_ancestorsc                 C   sT   |  t| j d| j| jg |  t| jjddd| j| j| jg d S r   )r   r   r   Zget_descendantsr   r
   r	   r   r   r   r   test_get_descendants2   s    
z1TestCollectionTreeOperations.test_get_descendantsc                 C   sL   |  t| j d| j| jg |  t| jjddd| jg d S )Nr   Fr   )r   r   r	   Zget_siblingsr   r
   r   r   r   r   test_get_siblings@   s    
z.TestCollectionTreeOperations.test_get_siblingsc                 C   sd   |  t| j d| jg |  t| jjddd| jg |  t| j dg  d S r   )r   r   r
   Zget_next_siblingsr   r	   r   r   r   r   test_get_next_siblingsN   s$    z3TestCollectionTreeOperations.test_get_next_siblingsc                 C   sd   |  t| j d| jg |  t| j dg  |  t| jjddd| jg d S r   )r   r   r	   Zget_prev_siblingsr   r
   r   r   r   r   test_get_prev_siblings`   s    z3TestCollectionTreeOperations.test_get_prev_siblingsN)
__name__
__module____qualname__r   r   r   r   r   r   r   r   r   r   r   r      s   r   N)Zdjango.testr   Zwagtail.modelsr   r   r   r   r   r   <module>   s   