Team Ai
Datasetpublic

codekingpro/portable-devtools

sourceHugging Faceupdated 5mo agoView on Hugging Face
1likes14kdownloads
test_client.cpython-313.pyc84 linesDownload Raw Back to __pycache__
1�

2��j�%��n�SSKrSSKJrJrJrJrJrJr SSKJ	r	J3r4 SSKrSSKJ
r
Jr SSKJr SSKrSSKJr SSKrSSKrSSKr\R.S\\SS44Sj5r\R.S\\SS44S	j5r\S5\4r\R."SS/S
9S\R6S\\SS44Sj5r\R."5S\S\\SS44Sj5rS\SS4SjrS\SS4SjrS\SS4Sjr S\SS4Sjr!S\SS4Sjr"S\\S6\4\\#\444Sjr$S!Sjr%S!Sjr&S!Sjr'S!Sjr(S!Sjr)S!S jr*g)"�N)�Any�Callable�	Generator�cast�Dict�Tuple)�	MagicMock�patch)�Settings�System)�	ClientAPI)�FastAPI�returnc#��# �[RRS5(a[R"S5 [7R"5nUv� UR5 g7f)N�CHROMA_INTEGRATION_TEST_ONLY�Integration test only)�os�environ�get�pytest�skip�chromadb�EphemeralClient�clear_system_cache��clients �_D:\code\apps\devtools\python\user_packages\Python313\site-packages\chromadb/test/test_client.py�
ephemeral_apirsE���	�z�z�~�~�4�5�5����+�,�
�
%�
%�
'�F�8�L�9����s�A%A'c#��# �[RRS5(a[R"S5 [10R"[R"5S-S9nUv� UR5 g7f)Nrrz/test_server��path)11rrrrrr�PersistentClient�tempfile�12gettempdirrrs r�persistent_apir%sZ���	�z�z�~�~�4�5�5����+�,�
�
&�
&�
�
 �
 �
"�^�
3��F��L�13����s�A:A<.�sync_client�async_client)�params�requestc#��# �URS:Xa;[S5 [S5 [Rv� SSS5 SSS5 g[S5 [S5 S[S[S[4S	jn[[U5v� SSS5 SSS5 g!,(df   Ni=f!,(df   g=f!,(df   N9=f!,(df   g=f7f)14Nr&z4chromadb.api.client.Client._validate_tenant_databasez,chromadb.api.client.Client.get_user_identityz?chromadb.api.async_client.AsyncClient._validate_tenant_databasez7chromadb.api.async_client.AsyncClient.get_user_identity�args�kwargsrc�v�[R"5R[R"U0UD65nU$�N)�asyncio�get_event_loop�run_until_completer�AsyncHttpClient)r+r,�clss   r�factory�!http_api_factory.<locals>.factory1s5��!�0�0�2�E�E� �0�0�$�A�&�A��C��J�)�paramr15r�16HttpClientrr�HttpAPIFactory)r)r4s  r�http_api_factoryr:%s�����}�}�
�%�
�I�
J��E�F��)�)�)�G�K�
J��T�
U��P�Q��3��#��#���>�7�3�3�R�V�
U�G�F��K�
J��17R�Q��V�
U�sb�C$�B1�B �B1�C$�C�%*C�C�	C$� 18B.	�*B1�119B?�;C$�20C	�C�21C!�C$r:c#��# �[RRS5b/[[RRS55nU"US9nOU"5nUv� UR	5 g7f)N�CHROMA_SERVER_HTTP_PORT)�port)rrr�intr)r:r=rs   r�http_apir?:sT���	�z�z�~�~�/�0�<��2�:�:�>�>�";�<�=��!�t�,��!�#��22�L�23����s�A,A.rc�F�UR5nURSLdeg)NF��get_settings�
is_persistent)r�settingss  r�test_ephemeral_clientrEEs$���)�)�+�H��!�!�U�*�*�*r6r%c�F�UR5nURSLdeg)NTrA)r%rDs  r�test_persistent_clientrGJs$���*�*�,�H��!�!�T�)�)�)r6r?c�j�UR5nURS:XdURS:Xdegg)N�chromadb.api.fastapi.FastAPIz'chromadb.api.async_fastapi.AsyncFastAPI)rB�chroma_api_impl)r?rDs  r�test_http_clientrKOsB���$�$�&�H�� � �$B�B��#�#�'P�P��	Q�P�	Cr6c�p�U"[SS9S9 g![an[U5S:XdeSnAgSnAff=f)Nz	127.0.0.1)�chroma_server_host)rDznChroma server host provided in settings[127.0.0.1] is different to the one provided in HttpClient: [localhost]�r�24ValueError�str�r:�es  r�0test_http_client_with_inconsistent_host_settingsrSWs?��25��(�k�"J�K���26���F��
@�	27�
@��28�s�
�295�0�5c�r�U"S[SS9S9 g![an[U5S:XdeSnAgSnAff=f)NiBiA)�chroma_server_http_port)r=rDziChroma server http port provided in settings[8001] is different to the one provided in HttpClient: [8002]rNrQs  r�0test_http_client_with_inconsistent_port_settingsrVcsJ��30����(,��	31���32���F�z�
{�	33�
{��34�s��356�1�6c�D^�0mS[S[S[4U4SjjnUT4$)N�_r,rc�L>�TRU5 [5n0UlU$r.)�updater	�headers)rXr,�session�captureds   �rr4�)make_sync_client_factory.<locals>.factoryxs"���������+������r6)r)r4r]s @r�make_sync_client_factoryr_ts/���!�H��C��3��3���H��r6c36���[SSSSSSSS9n[U5n[5up#[R"[37S	[
5[
5/S389 [SUS399 [U5nSSS5 SSS5 WR5 USnURS:XdeURS:XdeURS:XdeUS
beUSSLdeg!,(df   Nt=f!,(df   N}=f)NrI�	localhosti(#Tg)@�@�)rJrMrU�chroma_server_ssl_verify�chroma_http_keepalive_secs�chroma_http_max_connections�%chroma_http_max_keepalive_connections�require)�side_effectz!chromadb.api.fastapi.httpx.Client�limits�timeout�verify)rrr_r40�objectrr	�stop�keepalive_expiry�max_connections�max_keepalive_connections)rD�systemr4r]�apirjs      r�+test_fastapi_uses_http_limits_from_settingsrt�s����6�&� $�!%�#'�$&�.0��H��H�
�F�0�2��G�	���g�y�y�{�I�K�6P�	Q�
�6�G�
L��&�/�C�M�41R��H�H�J�
�h�
�F��"�"�d�*�*�*��!�!�R�'�'�'��+�+�r�1�1�1��I��&�&�&��H���%�%�%�M�
L��42R�	Q�s$�C)�C�*C)�43C&	�"C)�)44C7c���[RRS5(a[R"S5 [45R"5n[R"US9nURS5nURSS/SS/S	S460S	S0/S9 URnUR5 URS
Lde[R"US9nURS5nUR5n[US5S:XdeSUS;deSUS;deUR5 UR!5 UR!5 SSS5 g!,(df   g=f)zBTest that close() properly releases resources in PersistentClient.rrr �test_collection�id1�id2�doc1�doc2�key�value1�value2��ids�	documents�	metadatasFr�N)rrrrrr#�TemporaryDirectoryrr"�create_collection�add�_system�close�_running�get_collection�lenr)�tmpdirr�47collectionrr�client2�collection2�resultss       r�test_persistent_client_closer��sE��	�z�z�~�~�4�5�5����+�,�	�	$�	$�	&�&��*�*��7���-�-�.?�@�48��������v�&��x�(�5�(�*;�<�	�	49�����	�������%�'�'�'��+�+��8���,�,�->�?���/�/�#���7�5�>�"�a�'�'�'�����&�&�&�����&�&�&��
�
���!�!�#��"�"�$�;50'�	&�	&�s
�C=E�51E$c	���[RRS5(a[R"S5 [52R"5n[R"US9nURnURS5nURSS/SS/S	S530S	S0/S9 S
S
S
5 WRSLde[R"US9nURS5nUR5n[US5S:XdeS
S
S
5 WR5 WR5 S
S
S
5 g
!,(df   N�=f!,(df   NH=f!,(df   g
=f)z6Test that PersistentClient works as a context manager.rrr rvrwrxryrzr{r|r}r~NFrr�)rrrrrr#r�rr"r�r�r�r�r�r�r)r�rrrr�r�r�r�s       r�&test_persistent_client_context_managerr��s,��	�z�z�~�~�4�5�5����+�,�	�	$�	$�	&�&�
�
&�
&�F�
3�v��^�^�F��1�1�2C�D�J��N�N��E�N�!�6�*�!�8�,�u�h�.?�@�
�
�	4����%�'�'�'��
&�
&�F�
3�w�!�0�0�1B�C�K�!�o�o�'�G��w�u�~�&�!�+�+�+�4�54	�!�!�#��"�"�$�-55'�	&�
3�
3��4�
3��!56'�	&�s<�E�%9D3�-E�6E�(E�357E	�=E�58E	�E�59E#c�b�[RRS5(a[R"S5 [60R"5nURnURS5nURS/S/S9 UR5 URSLdeUR5 g)	z-Test that close() works with EphemeralClient.rrrvrwry�rr�FN)
rrrrrrrr�r�r�r�r�r�rrrr�s   r�test_ephemeral_client_closer��s���	�z�z�~�~�4�5�5����+�,�
�
%�
%�
'�F�
�^�^�F��)�)�*;�<�J��N�N��w�6�(�N�3��L�L�N��?�?�e�#�#�#�61���r6c��[RRS5(a[R"S5 [62R"5nURnURS5nURS/S/S9 URSLdeSSS5 WRS	LdeWR5 g!,(df   N0=f)63z5Test that EphemeralClient works as a context manager.rrrvrwryr�TNF)rrrrrrrr�r�r�r�rr�s   r�%test_ephemeral_client_context_managerr��s���	�z�z�~�~�4�5�5����+�,�	�	!�	!�	#�v������-�-�.?�@�64����E�7�v�h��7����$�&�&�&�65$��?�?�e�#�#�#�66����67$�	#�s
�AB<�<68C69c���[RRS5(a[R"S5 [70R"5n[R"US9nURS5nURS/S/S9 UR5 UR5 UR5 UR5 SSS5 g!,(df   g=f)	z9Test that calling close() multiple times is a safe no-op.rrr rvrwryr�N)
rrrrrr#r�rr"r�r�r�r)r�rr�s   r�test_client_close_idempotentr�s���	�z�z�~�~�4�5�5����+�,�	�	$�	$�	&�&��*�*��7���-�-�.?�@�71����E�7�v�h��7�	����	����	�����!�!�#�72'�	&�	&�s
�A8C�73C)rN)+r/�typingrrrrrr�
unittest.mockr	r74r�chromadb.configrr�chromadb.apir
�chromadb.server.fastapi�chromadb.api.fastapirrr#r�fixturerr%r9�FixtureRequestr:r?rErGrKrSrVrPr_rtr�r�r�r�r��r6r�<module>r�s����>�>�*��,�"��(�
��	���� �y��D�$�!6�7� �� ���� �	�)�T�4�"7�8� �� ��#�y�.�)�����
�~�6�7�4�
�
"�
"�4��~�t�T�)�*�4�8�4�(���� �~� �)�I�t�T�<Q�2R� �� �+��+�t�+�75*�9�*��*�76�y��T��	77�$�	78�	�	79�80�$�81�	�82�"83�%���c��(:�D��c��N�(J�"K�84�&�6"%�J%�< �( �$$r6
codekingpro/portable-devtools · Team Ai