o
    h                     @   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   get_first_root_noderoot_collection	add_childholiday_photos_collectionevil_plans_collectionrefresh_from_dbself r   c/var/www/html/ndineBlogger/venv/lib/python3.10/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   assertNotEqual
assertLess)r   old_evil_pathold_holiday_path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	inclusive)r   listr
   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   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
   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   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
   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)django.testr   wagtail.modelsr   r   r   r   r   r   <module>   s    