o
    h                      @   s   d dl Z d dlZd dlZedZejjZG dd dejZdd Z	ej
ddd	d
 Zej
dddd Zdd Zdd Zdd ZdS )    Nzpyarrow.flightc                   @   s2   e Zd Zeedgejdg Z	dd Z
dS )ExampleServer)aint32   simplec                 C   s*   |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
descriptor r   S/var/www/vscode/kcb/lib/python3.10/site-packages/pyarrow/tests/test_flight_async.pyget_flight_info#   s
   

zExampleServer.get_flight_infoN)__name__
__module____qualname__flight
FlightInfopyarrowschemaFlightDescriptorfor_commandr	   r   r   r   r   r   r      s    

r   c                 C   sP   | j s&tt}|   W d    n1 sw   Y  tt|j d S d S N)supports_asyncpytestraisesr
   as_asyncskipstrvalue)clienter   r   r   async_or_skip,   s   
r$   module)scopec               	   c   sv    t  -} td| j }|V  W d    n1 sw   Y  W d    d S W d    d S 1 s4w   Y  d S )Nzgrpc://localhost:)r   r   connectport)serverr"   r   r   r   flight_client4   s   "r*   c                 c   s    t |  |  V  d S r   )r$   r   r*   r   r   r   async_client;   s   r,   c                 C   sZ   t | jtsJ | jr|   d S tt |   W d    d S 1 s&w   Y  d S r   )
isinstancer   boolr   r   r   r
   r+   r   r   r   test_async_support_propertyA   s   
"r/   c                        fdd}t |  d S )Nc                     s0   t jd}  | I d H }|tjksJ d S )Nr   )r   r   r   r   r   r	   )r   infor,   r   r   _testK   s   z#test_get_flight_info.<locals>._testasynciorunr,   r3   r   r2   r   test_get_flight_infoJ   s   r8   c                    r0   )Nc                     s^   t jd} tt} | I d H  W d    n1 sw   Y  dt|jv s-J d S )Nr   r   )	r   r   r   r   r   r
   r   reprr!   )r   excinfor2   r   r   r3   T   s   z)test_get_flight_info_error.<locals>._testr4   r7   r   r2   r   test_get_flight_info_errorS   s   r;   )r5   r   r   importorskipr   mark
pytestmarkFlightServerBaser   r$   fixturer*   r,   r/   r8   r;   r   r   r   r   <module>   s   




		