o
    žº¥gÑ
  ã                   @   sœ   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	m
Z
 d dlmZm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)ÚAdminClientÚClient)ÚSystem)ÚSqliteDB)ÚInvalidCollectionExceptionÚNotFoundError)ÚNOT_CLUSTER_ONLYÚClientFactoriesÚclient_factoriesÚreturnc                 C   sì   |   ¡ }| ¡  |  ¡ }| d¡ | j dd}| d¡}| d¡ t t¡ | 	d¡ W d   ƒ n1 s6w   Y  t t
¡ | d¡ W d   ƒ n1 sPw   Y  t t
¡ | dgg d¢¡ W d   ƒ d S 1 sow   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_databaser   Úget_collectionÚupsert)r
   ÚclientÚadmin_clientÚ
collection© r   ú]/var/www/visachat/venv/lib/python3.10/site-packages/chromadb/test/api/test_delete_database.pyÚtest_deletes_database	   s    


ÿÿ"ÿr!   c                 C   s²   |   ¡ }| ¡  |  ¡ }| d¡ | d¡ | j dd}| d¡ | j dd}| d¡}| d¡ | d¡j|jks<J ‚t 	t
¡ | d¡ W d   ƒ d S 1 sRw   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_collectionr   r   r    Ú$test_does_not_affect_other_databases    s   




"ÿr)   Úsqlite_persistentc                 C   sÀ   |   t¡}t | ¡}| d¡ tj| dd}| d¡ | d¡ t 	t
¡ | d¡ W d   ƒ n1 s5w   Y  | ¡ }| d¡ ¡ }|d dksNJ ‚W d   ƒ d S 1 sYw   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Úrowr   r   r    Útest_collection_was_removed7   s   




ÿ
"þr3   c                 C   sR   |   ¡ }| ¡  |  ¡ }t t¡ | d¡ W d   ƒ d S 1 s"w   Y  d S )Nr   )r   r   r   r   r   r   r   )r
   r   r   r   r   r    Ú(test_errors_when_database_does_not_existK   s   "ÿr4   )r   Úchromadb.api.clientr   r   Úchromadb.configr   Úchromadb.db.impl.sqliter   Úchromadb.errorsr   r   Úchromadb.test.conftestr   r	   r!   r)   r3   r4   r   r   r   r    Ú<module>   s    