o
    Zh9                     @   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                   @   s6   e Zd Zeedgejdg ddZ	dd Z
dS )ExampleServer)aZ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   V/var/www/html/lang_env/lib/python3.10/site-packages/pyarrow/tests/test_flight_async.pyget_flight_info%   s
   

zExampleServer.get_flight_infoN)__name__
__module____qualname__flightZ
FlightInfopyarrowZ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_client6   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_propertyC   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   _testM   s   z#test_get_flight_info.<locals>._testasynciorunr*   r1   r   r0   r   test_get_flight_infoL   s   r6   c                    r.   )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   excinfor0   r   r   r1   V   s   z)test_get_flight_info_error.<locals>._testr2   r5   r   r0   r   test_get_flight_info_errorU   s   r9   )r3   r   r   Zimportorskipr   markZ
pytestmarkZFlightServerBaser   r"   Zfixturer(   r*   r-   r6   r9   r   r   r   r   <module>   s   




		