
    j6                        d dl Z d dlZd dlZ ej                  de      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.flight)exc_typec                       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      F/tmp/pip-target-jj7clzyk/lib/python/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_clientr1   4   sZ     	 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   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   infor4   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r4   r<   s   ` r   test_get_flight_inforA   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   excinfor4   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_errorrF   S   s    8 KKr   )r>   r"   r   importorskipImportErrorr   mark
pytestmarkFlightServerBaser   r*   fixturer1   r4   r8   rA   rF   r   r   r   <module>rM      s   $   			-	D[[
8F++ 8 " h   h#  #
%r   