o
    g                     @   s   d dl Z d dlmZ d dlmZ d dlmZmZ d dlm	Z	 e j
defddZdeddfd	d
ZdeddfddZdeddfddZdeddfddZdS )    Nmaybe_set_tenant_and_databaseUserIdentity)DEFAULT_DATABASEDEFAULT_TENANT)ChromaAuthErrorreturnc                   C   s   t dddgdS )Ntest_user_idtest_tenanttest_database)user_idtenant	databasesr    r   r   Y/var/www/visachat/venv/lib/python3.10/site-packages/chromadb/test/auth/test_auth_utils.pyuser_identity	   s
   r   c                 C   s0   t | dddd\}}|dksJ |dksJ d S )NFuser_provided_tenantuser_provided_databaser   4overwrite_singleton_tenant_database_access_from_authr   r   r   r   resolved_tenantresolved_databaser   r   r   test_doesnt_overrite_from_auth   s   
r   c                 C   s\   t | dttd\}}|dksJ |dksJ t | dd d d\}}|dks&J |dks,J d S )NTr   r   r   )r   r   r   r   r   r   r   ;test_sets_tenant_and_database_when_none_or_default_provided   s    

r   c                 C   s@   t t t| dddd W d    d S 1 sw   Y  d S )NTr   r   r   )pytestraisesr   r   )r   r   r   r   Btest_errors_when_provided_tenant_and_database_dont_match_from_auth6   s   "r   c                 C   sj   d| _ dg| _t| dd d d\}}|d u sJ |d u sJ t| dddd\}}|dks-J |dks3J d S )N*Tr   r   r   )r   r   r   r   r   r   r   -test_doesnt_overrite_from_auth_when_ambiguousB   s$   

r    )r   chromadb.auth.utilsr   chromadb.authr   chromadb.configr   r   chromadb.errorsr   fixturer   r   r   r   r    r   r   r   r   <module>   s&    

