
    g9                        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 dd      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      T/var/www/openai/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      sE    ##'(++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*   .   sI      ]]./1OO 0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_clientr1   6   sF     	F^^/}=>&L ? 
>> 
s1   A#AAA	AA	AAAc              #   F   K   t        |        | j                          y wr    )r*   r$   r1   s    r   async_clientr4   =   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   r3   s    r   test_async_support_propertyr8   C   sR    m22D999## ]]./""$ 0//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   infor4   s     r   _testz#test_get_flight_info.<locals>._testM   sI     ,,88C
!11*==}00000 >s   4AAAasynciorunr4   r<   s   ` r   test_get_flight_inforA   L   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   excinfor4   s     r   r<   z)test_get_flight_info_error.<locals>._testV   sl     ,,88D
]]./7..z::: 0 !D$7777 ; 0/s.   9BA:A8A:"B8A::B?Br=   r@   s   ` r   test_get_flight_info_errorrF   U   s    8 KKr   )r>   r"   r   importorskipr   mark
pytestmarkFlightServerBaser   r*   fixturer1   r4   r8   rA   rF   r   r   r   <module>rL      s   $   			-	.[[
8F++ 8$" h   h#  #
%r   