
    _jo
                         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	 d dl
mZ deddfd	Zdeddfd
ZdeddfdZdeddfdZdS )    N)AdminClientClient)System)SqliteDB)NotFoundError)ClientFactoriesclient_factoriesreturnc                    |                                  }|                                 |                                 }|                    d           |                      d          }|                    d          }|                    d           t          j        t                    5  |	                    d           d d d            n# 1 swxY w Y   t          j        t                    5  |
                    d           d d d            n# 1 swxY w Y   t          j        t                    5  |                    dgg d           d d d            d S # 1 swxY w Y   d S )Ntest_delete_databasedatabasefoo)        r   r   )create_clientresetcreate_admin_client_from_systemcreate_databasecreate_collectiondelete_databasepytestraisesr   get_databaseget_collectionupsert)r	   clientadmin_client
collections       c/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/chromadb/test/api/test_delete_database.pytest_deletes_databaser    	   s   ++--F
LLNNN#CCEEL  !7888++5K+LLF))%00J  !7888	}	%	% : :!!"8999: : : : : : : : : : : : : : : 
}	%	% % %e$$$% % % % % % % % % % % % % % % 
}	%	% 4 45'???3334 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4s6   +CCC1DDD7EE"%E"c                    |                                  }|                                 |                                 }|                    d           |                    d           |                      d          }|                    d           |                      d          }|                    d          }|                    d           |                    d          j        |j        k    sJ t          j	        t                    5  |                    d           d d d            d S # 1 swxY w Y   d S )Nfirstsecondr   test)r   r   r   r   r   r   r   idr   r   r   )r	   r   r   first_clientsecond_clientsecond_collections         r   $test_does_not_affect_other_databasesr)       so   ++--F
LLNNN#CCEEL  )))  ***#1171CCL""6***$22H2EEM%77??  )))''//26G6JJJJJ	}	%	% , ,##F+++, , , , , , , , , , , , , , , , , ,s   D33D7:D7sqlite_persistentc                 ^   |                      t                    }t          j        |           }|                    d           t          j        | d          }|                    d           |                    d           t          j	        t                    5  |                    d           d d d            n# 1 swxY w Y   |                                5 }|                    d                                          }|d         dk    sJ 	 d d d            d S # 1 swxY w Y   d S )Nr   r   r   z SELECT COUNT(*) from collectionsr   )instancer   r   from_systemr   r   r   r   r   r   r   r   txexecutefetchone)r*   sqliter   r   currows         r   test_collection_was_removedr4   7   s   ''11F*+<==L  !7888 1<RSSSF
U###  !7888	}	%	% % %e$$$% % % % % % % % % % % % % % % 
 kk<==FFHH1v{{{{{                 s$   B??CC6D""D&)D&c                    |                                  }|                                 |                                 }t          j        t
                    5  |                    d           d d d            d S # 1 swxY w Y   d S )Nr   )r   r   r   r   r   r   r   )r	   r   r   s      r   (test_errors_when_database_does_not_existr6   K   s    ++--F
LLNNN#CCEEL	}	%	% , ,$$U+++, , , , , , , , , , , , , , , , , ,s   A99A= A=)r   chromadb.api.clientr   r   chromadb.configr   chromadb.db.impl.sqliter   chromadb.errorsr   chromadb.test.conftestr   r    r)   r4   r6        r   <module>r>      s    3 3 3 3 3 3 3 3 " " " " " " , , , , , , ) ) ) ) ) ) 2 2 2 2 2 24O 4 4 4 4 4.,? ,t , , , ,.6 d    (, ,SW , , , , , ,r=   