
    ei                         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y)    Nzpyarrow.flightc                       e Zd Zej	                   ej                  dg      ej                  j                  d      g       Z	d Z
y)ExampleServer)aint32   simplec                     |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      a/var/www/html/glpi_dashboard/venv/lib/python3.12/site-packages/pyarrow/tests/test_flight_async.pyget_flight_infozExampleServer.get_flight_info#   sB    *###:-%&788!"677    N)__name__
__module____qualname__flight
FlightInfopyarrowschemaFlightDescriptorfor_commandr   r    r   r   r   r      s?    ##'(++I6
K8r   r   c                     | j                   s[t        j                  t              5 }| j	                          d d d        t        j
                  t        j                               y y # 1 sw Y   3xY wN)supports_asyncpytestraisesr   as_asyncskipstrvalue)clientes     r   async_or_skipr)   ,   sR      ]]./ 	1OO	CL!	 !	 	s   A))A2module)scopec               #      K   t               5 } t        j                  d| j                         5 }| d d d        d d d        y # 1 sw Y   xY w# 1 sw Y   y xY ww)Nzgrpc://localhost:)r   r   connectport)serverr'   s     r   flight_clientr0   4   sZ     	 F^^/}=> 	&L	 	 	 s1   A#AAA	AA	AAAc              #   F   K   t        |        | j                          y wr   )r)   r#   r0   s    r   async_clientr3   ;   s     - 

 
 
""s   !c                     t        | j                  t              sJ | j                  r| j                          y t	        j
                  t              5  | j                          d d d        y # 1 sw Y   y xY wr   )
isinstancer    boolr#   r!   r"   r   r2   s    r   test_async_support_propertyr7   A   s[    m22D999## ]]./ 	%""$	% 	% 	%s   A--A6c                 D      fd}t        j                   |              y )Nc                     K   t         j                  j                  d      } j                  |        d {   }|t        j
                  k(  sJ y 7 w)Nr   )r   r   r   r   r   r   )r   infor3   s     r   _testz#test_get_flight_info.<locals>._testK   sI     ,,88C
!11*==}00000 >s   4AAAasynciorunr3   r;   s   ` r   test_get_flight_infor@   J   s    1
 KKr   c                 D      fd}t        j                   |              y )Nc                    K   t         j                  j                  d      } t        j                  t
              5 }j                  |        d {    d d d        dt        j                        v sJ y 7 &# 1 sw Y   %xY ww)Nr	   r
   )	r   r   r   r!   r"   r   r   reprr&   )r   excinfor3   s     r   r;   z)test_get_flight_info_error.<locals>._testT   ss     ,,88D
]]./ 	;7..z:::	; !D$7777 ;	; 	;s.   9BA:A8A:"B8A::B?Br<   r?   s   ` r   test_get_flight_info_errorrE   S   s    8 KKr   )r=   r!   r   importorskipr   mark
pytestmarkFlightServerBaser   r)   fixturer0   r3   r7   r@   rE   r   r   r   <module>rK      s   $   			-	.[[
8F++ 8 " h   h#  #
%r   