
    z-Ph                          d dl Z d dlZd dlZ ej        d          Zej        j        Z G d dej                  Zd Z	 ej
        d          d             Z ej
        d          d	             Zd
 Zd Zd ZdS )    Nzpyarrow.flightc                       e Zd Ze                     ej        dg          ej                            d          g           Z	d Z
dS )ExampleServer)aint32   simplec                 x    |j         dk    r| j        S |j         dk    rt          d          t          d          )Nr      unknownUnknown commandzUnknown descriptor)commandsimple_infoNotImplementedError)selfcontext
descriptors      _/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pyarrow/tests/test_flight_async.pyget_flight_infozExampleServer.get_flight_info#   sE    **##:--%&7888!"6777    N)__name__
__module____qualname__flight
FlightInfopyarrowschemaFlightDescriptorfor_commandr   r    r   r   r   r      sa        ##'((++I66
 K8 8 8 8 8r   r   c                     | j         smt          j        t                    5 }|                                  d d d            n# 1 swxY w Y   t          j        t          |j                             d S d S N)supports_asyncpytestraisesr   as_asyncskipstrvalue)clientes     r   async_or_skipr)   ,   s      "].// 	1OO	 	 	 	 	 	 	 	 	 	 	 	 	 	 	CLL!!!!!	" "s   AA	Amodule)scopec               #      K   t                      5 } t                              d| j                   5 }|V  d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )Nzgrpc://localhost:)r   r   connectport)serverr'   s     r   flight_clientr0   4   s      	 F^^===>> 	&LLL	 	 	 	 	 	 	 	 	 	 	 	 	 	 	                 s3   #AAAA			AA		AA!$A!c              #   T   K   t          |            |                                 V  d S r   )r)   r#   r0   s    r   async_clientr3   ;   s2      -   

 
 
"
""""""r   c                    t          | j        t                    sJ | j        r|                                  d S t	          j        t                    5  |                                  d d d            d S # 1 swxY w Y   d S r   )
isinstancer    boolr#   r!   r"   r   r2   s    r   test_async_support_propertyr7   A   s    m2D99999# %     ].// 	% 	%""$$$	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	%s   A55A9<A9c                 J      fd}t          j         |                       d S )Nc                     K   t           j                            d          }                     |            d {V }|t          j        k    sJ d S )Nr   )r   r   r   r   r   r   )r   infor3   s     r   _testz#test_get_flight_info.<locals>._testK   s\      ,88CC
!11*========}0000000r   asynciorunr3   r;   s   ` r   test_get_flight_infor@   J   s:    1 1 1 1 1
 Kr   c                 J      fd}t          j         |                       d S )Nc                    K   t           j                            d          } t          j        t
                    5 }                    |            d {V  d d d            n# 1 swxY w Y   dt          |j                  v sJ d S )Nr	   r
   )	r   r   r   r!   r"   r   r   reprr&   )r   excinfor3   s     r   r;   z)test_get_flight_info_error.<locals>._testT   s      ,88DD
].// 	;7..z:::::::::	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; !D$7$7777777s   A$$A(+A(r<   r?   s   ` r   test_get_flight_info_errorrE   S   s:    8 8 8 8 8 Kr   )r=   r!   r   importorskipr   mark
pytestmarkFlightServerBaser   r)   fixturer0   r3   r7   r@   rE   r   r   r   <module>rK      s  $   		-	.	.[
8 8 8 8 8F+ 8 8 8 " " " h    h# #  #
% % %      r   