
    	]j
                     \   d Z ddlZddlmZmZ ddlmZ ddlmZ ddl	m
Z
 ej                  j                  d        Zej                  j                  d        Zej                  j                  d	        Zej                  j                  d
        Zej                  j                  d        Zy)zThis file and its contents are licensed under the Apache License 2.0. Please see the included NOTICE for copyright information and LICENSE for a copy of the license.
    N)OrganizationOrganizationMember)Task)make_annotation)Userc                     | j                  d      }|j                         }|d   d   | j                  j                  k(  sJ y )Nz/api/users/r   active_organization)getjsonorganizationidbusiness_clientresponseresponse_datas      O/root/env/lib/python3.12/site-packages/label_studio/tests/test_organizations.pytest_active_organization_filledr   
   sB    ""=1HMMOM12o6R6R6U6UUUU    c                     | j                  d      }|j                         }t        |      dk(  sJ |d   d   | j                  j                  k(  sJ y )N/api/organizations/   r   r   )r
   r   lenr   r   r   s      r   test_api_list_organizationsr      sU    ""#89HMMOM}"""D!_%A%A%D%DDDDr   c                    | j                   }| j                  }t        j                  j	                  |      j                         }t        d|i|j                         | j                  d|j                   d|j                   d      }|j                         }|d   |j                  k(  sJ |d   |j                  k(  sJ |d	   d
k(  sJ |d   d
k(  sJ y )N)projectcompleted_by)task_idr   /memberships//userr   annotations_countr   contributed_projects_count)
r    r   r   objectsfilterfirstr   r   r
   r   )r   configured_projectr    r   taskr   r   s          r   +test_organization_member_retrieve_same_userr(      s    D"//L<<'9:@@BD^T*DGG<""%88IW[W^W^V__`#abHMMOM DGG+++(LOO;;;,-22256!;;;r   c                    | j                   }t        j                  j                  d      }t        j                  j                  ||       | j                  d|j                   d|j                   d      }|j                         }t        |       |d   |j                  k(  sJ |d   |j                  k(  sJ |d	   d
k(  sJ |d   d
k(  sJ y )Nother_user@pytest.netemailr    r   r   r   r   r    r   r!   r   r"   )	r   r   r#   creater   r
   r   r   print)r   r   
other_userr   r   s        r   3test_organization_member_retrieve_other_user_in_orgr1   '   s    "//L$$+B$CJ%%:L%Q""%88IWaWdWdVeef#ghHMMOM	- JMM111(LOO;;;,-22256!;;;r   c                 D   | j                   }t        j                  j                  d      }t	        j
                  |      }t        j                  j                  ||       | j                  d|j                   d|j                   d      }|j                  dk(  sJ y )	Nr*   r+   )
created_byr-   r   r   r   i  )
r    r   r#   r.   r   create_organizationr   r
   r   status_code)r   r    r0   other_organizationr   s        r   0test_organization_member_retrieve_not_active_orgr7   5   s    D$$+B$CJ%99ZP%%4>P%Q""%89K9N9N8O}]a]d]d\eef#ghH3&&&r   )__doc__pytestorganizations.modelsr   r   tasks.modelsr   tests.utilsr   users.modelsr   mark	django_dbr   r   r(   r1   r7    r   r   <module>rA      s     A  '  V V E E 
< 
< 
< 
< ' 'r   