
    	]j                         d dl Z d dlZd dlmZ d dlmZ d dlmZ ej                  j                  Z
d Zd Zd Zd Zd	 Zy)
    N)Task)project_choices)make_projectc                 X   t        t               | j                  d      }|j                  }dD ]  }t	        ||d|i        |j                          | j                  d|j                   d      }|j                  dk(  sJ |j                          dD ]  }t        ||      i k(  rJ  y )NFuse_ml_backendcreated_labelscreated_labels_draftscreated_annotationsgarbled/api/projects//summary/reset   )r   r   usersummarysetattrsavepostidstatus_coderefresh_from_dbgetattr)business_clientprojectsfieldrs        W/root/env/lib/python3.12/site-packages/label_studio/tests/test_project_reset_summary.py test_reset_summary_empty_projectr       s    ?,o.B.BSXYGAS .59e,-.FFH~gjj\HIA==CS 'q% B&&&'    c                    t        t               | j                  d      }| j                  d|j                   dt        j                  dddii      d	      }|j                  d
k(  sJ t        j                  j                  |      j                         }|sJ |j                  }| j                  d|j                   dt        j                  ddddddgidgi      d	      }|j                  d
k(  sJ dD ]  }t        ||d|i        |j                          | j                  d|j                   d      }|j                  dk(  sJ |j                          dD ]  }t!        ||      i k(  rJ  |j"                  dddiik(  sJ y )NFr   r   /importdataimagekittens.jpgapplication/jsonr$   content_type   r   /api/tasks//draftsresultsomexnoneOpossum	from_nameto_nametypevaluer	   r   r   r   )r
   r      )r   r   r   r   r   jsondumpsr   r   objectsfilterfirstr   r   r   r   r   r   r   r   r   taskr   r   s         r   %test_reset_summary_project_has_draftsr@      s   ?,o.B.BSXYG
G,ZZ'=!9:;' 	 	A
 ==C<<w/557DK4A
dggYg&ZZffX^aj`kWlmno
 ( 	 	A ==CS .59e,-.FFH~gjj\HIA==C: 'q% B&&&' ""v	1~&>>>>r!   c                 4   t        t               | j                  d      }| j                  d|j                   dt        j                  dddii      d	      }|j                  d
k(  sJ t        j                  j                  |      j                         }|sJ |j                  }| j                  d|j                   dt        j                  ddddddgidgi      d	      }|j                  d
k(  sJ dD ]  }t        ||d|i        |j                          | j                  d|j                   d      }|j                  dk(  sJ |j                          |j                   i k(  sJ |j"                  ddik(  sJ |j$                  dddiik(  sJ y )NFr   r   r#   r$   r%   r&   r'   r(   r*   r+   r,   /annotationsr.   r/   r0   r1   r2   r3   r	   r   r   r   zsome|x|noner8   )r   r   r   r   r   r9   r:   r   r   r;   r<   r=   r   r   r   r   r   r   r
   r>   s         r   *test_reset_summary_project_has_annotationsrC   ?   s   ?,o.B.BSXYG
G,ZZ'=!9:;' 	 	A
 ==C<<w/557DK4A
dggYl+ZZffX^aj`kWlmno
 ( 	 	A ==CS .59e,-.FFH~gjj\HIA==C""b(((  ]A$6666A7777r!   c                    t        t               | j                  d      }| j                  d|j                   dt        j                  dddii      d	      }|j                  d
k(  sJ t        j                  j                  |      j                         }|sJ |j                  }| j                  d|j                   dt        j                  ddddddgidgi      d	      }|j                  d
k(  sJ | j                  d|j                   dt        j                  ddddddgidgi      d	      }|j                  d
k(  sJ dD ]  }t        ||d|i        |j                          | j                  d|j                         }|j                  dk(  sJ |j                          dD ]  }t!        ||      i k(  rJ  y )NFr   r   r#   r$   r%   r&   r'   r(   r*   r+   r,   r-   r.   r/   r0   r1   Mouser3   rB   r2   r	   r   z4/api/dm/actions?id=delete_tasks_annotations&project=r   )r   r   r   r   r   r9   r:   r   r   r;   r<   r=   r   r   r   r   r   r>   s         r   Ntest_delete_tasks_and_annotations_clears_created_drafts_annotations_and_labelsrF   b   s   ?,o.B.BSXYG
G,ZZ'=!9:;' 	 	A
 ==C<<w/557DK4A
dggYg&ZZffX^ah`iWjklm
 ( 	 	A ==C
dggYl+ZZffX^aj`kWlmno
 ( 	 	A ==CS .59e,-.FFHST[T^T^S_`aA==CS 'q% B&&&'r!   c                 (   t        t               | j                  d      }| j                  d      }|j                  dk(  sJ | j                  d|j                   d      }|j                  dk(  sJ d|j                         x}v sJ |d   d	k(  sJ y )
NFr   z/logouti.  r   r   i  detailz-Authentication credentials were not provided.)r   r   r   getr   r   r   r9   )r   r   r   r_jsons       r   )test_logged_out_user_cannot_reset_summaryrK      s    ?,o.B.BSXYGI&A==C~gjj\HIA==C!&&(*+++(NNNNr!   )r9   pytesttasks.modelsr   tests.conftestr   tests.utilsr   mark	django_db
pytestmarkr    r@   rC   rF   rK    r!   r   <module>rT      s@       * $[[""
' !?H 8F('VOr!   