
    3i                         S SK r S SKJr  S SKJr  S SKJrJr  S SKJ	r	  \ R                  S\4S j5       rS\SS4S	 jrS\SS4S
 jrS\SS4S jrS\SS4S jrg)    Nmaybe_set_tenant_and_databaseUserIdentity)DEFAULT_DATABASEDEFAULT_TENANT)ChromaAuthErrorreturnc                      [        SSS/S9$ )Ntest_user_idtest_tenanttest_database)user_idtenant	databasesr        \/home/james-whalen/.local/lib/python3.13/site-packages/chromadb/test/auth/test_auth_utils.pyuser_identityr   	   s    "# r   r   c                 @    [        U SSSS9u  pUS:X  d   eUS:X  d   eg )NFuser_provided_tenantuser_provided_databaser   4overwrite_singleton_tenant_database_access_from_authr   r   r   r   resolved_tenantresolved_databases      r   test_doesnt_overrite_from_authr      s;    )F#=B37	*&O 4444 8888r   c                     [        U S[        [        S9u  pUS:X  d   eUS:X  d   e[        U SS S S9u  pUS:X  d   eUS:X  d   eg )NTr   r   r   )r   r   r   r   s      r   ;test_sets_tenant_and_database_when_none_or_default_providedr       ss     *G#=A+/	*&O m+++///)F#=A!#	*&O m+++///r   c                     [         R                  " [        5         [        U SSSS9  S S S 5        g ! , (       d  f       g = f)NTr   r   r   )pytestraisesr	   r   )r   s    r   Btest_errors_when_provided_tenant_and_database_dont_match_from_authr$   6   s0     
	'%'AE!7#;		
 
(	'	's   1
?c                     SU l         S/U l        [        U SS S S9u  pUb   eUb   e[        U SSSS9u  pUS:X  d   eUS:X  d   eg )N*Tr   r   r   )r   r   r   r   s      r   -test_doesnt_overrite_from_auth_when_ambiguousr'   B   s    M"eM)F#=A!#	*&O """$$$)F#=A37	*&O 4444 8888r   )r"   chromadb.auth.utilsr   chromadb.authr   chromadb.configr   r   chromadb.errorsr	   fixturer   r   r    r$   r'   r   r   r   <module>r-      s     = & < + |  	9, 	94 	900	00	
	
		
9 9RV 9r   