mxiP ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z  ddl ZddlZddlZddlmZmZmZddlmZddlmZ ddlmZddlmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%ejNj6Z(dZ)d Z*d Z+d Z,d Z-d Z.dZ/GddeZ0GddeZ1GddeZ2Gdde2Z3Gdde2Z4Gdde2Z5GddeZ6GddeZ7Gdd eZ8Gd!d"eZ9Gd#d$eZ:Gd%d&eZ;Gd'd(eZ<Gd)d*eZ=Gd+d,eZ>Gd-d.eZ?Gd/d0e Z@Gd1d2eZAGd3d4e ZBGd5d6eZCd7ZDGd8d9e$ZEGd:d;e#ZFGd<d=e"ZGGd>d?e!ZHGd@dAeZIGdBdCe"ZJGdDdEe!ZKGdFdGeZLGdHdIe!ZMGdJdKe"ZNGdLdMeZOGdNdOeZPGdPdQe"ZQGdRdSe$ZRGdTdUe#ZSGdVdWe"ZTGdXdYe$ZUGdZd[e$ZVGd\d]e#ZWGd^d_e"ZXGd`dae!ZYGdbdceZZddZ[deZ\dfZ]dgZ^dhZ_ejNjejNjdiZadjZbdkZcdlZdejNjdmZfdnZgdoZhdpZidqZjdrZkGdsdteZlduZmdvZndwZoejNjejdxk(dyzd{ZrejNjd|Zsd}Ztd~ZudZvdZwdZxdZydZzdZ{dZ|dZ}e?ddiZ~eAddiZejNjdZejNjejdxk(dzdZdZdZdZeGZeCZdZdZdZdZdZdZdZejNjejNjdZejNjdZejNjdZejNjdZdZdZdZdZejNjdZejNjdZejNjejNjdZdZdZdZdZdZdZdZdZejNjdZdZdZdZdZdZdZejNjdZGddeZejNjdZdZejNjZejNjdZejNjZejNjdZGdde2ZdZejNjdZdZGddeZdZdZdZd„ZejrdÄZejNjwdddgdǬȫdɄZy#e$rdZYDwxYw#e$rdZe&e&cZZe&e&cZZ e&e&cZ!Z"e&e&cZ#Z$e&Z%Y-wxYw)N)datetime)IpcReadOptions ReadStatstobytes)find_free_port)util)flight) FlightClientFlightServerBaseServerAuthHandlerClientAuthHandlerServerMiddlewareServerMiddlewareFactoryClientMiddlewareClientMiddlewareFactoryFlightCallOptionsc ddl}y)Nr)pyarrow.flight)pyarrows P/home/htdocs/ttos/venv/lib/python3.12/site-packages/pyarrow/tests/test_flight.py test_importrAsctjjds tdt j tjddz S)z-Get the path to the test resources directory.ARROW_TEST_DATAzITest resources not found; set ARROW_TEST_DATA to /testing/datar )osenvironget RuntimeErrorpathlibPathrr resource_rootr"FsF ::>>+ ,IJ J << #45 6 AArct}|sy ||z jd5}|jcdddS#1swYyxYw#t$r't d||z dt j wxYw)z)Get the contents of a test resource file.NrbzTest resource z< not found; did you initialize the test resource submodule? )r"openreadFileNotFoundErrorr traceback format_exc)pathrootfs rread_flight_resourcer-Ns ?D  Tk   % 668     TD[M*))2)=)=)?(@ B   s%A = A AA A 0A9ctdtjtdtdtjtdtdgdS)z'Get the paths to test TLS certificates.z root-ca.pemz cert0.pemz cert0.key)certkeyz cert1.pemz cert1.key) root_cert certificates)r-r CertKeyPairr!rrexample_tls_certsr4]sY*-8   )+6(5    )+6(5     rcvtjgdg}tjj|dgS)Nir some_intsnames)paarrayTable from_arrays)datas rsimple_ints_tablerBns4 $% D 88  [M  ::rctjgdtj}tjgdtj}tjtjj gd|tjj ddg|tjj ddg|gg}tj j |d g S) N)foobazquuxtype)rDrErFnew)rNrJr some_dictsr;)r=r>utf8 chunked_arrayDictionaryArrayr@r?) dict_valuesnew_dict_valuesrAs rsimple_dicts_tablerSus((1 BKhh<2779MO     * *< E    * *Aq6; ?    * *Aq6? C   D 88  \N  ;;rctjjtjgdtjgdgddgS)N)rDbarrEquxrJrKrLabr;)r=r?r@r>r!rrmultiple_column_tabler[sB 88  *F!G!#,!7!9'*Cj  22rc4eZdZdZdZdfd ZdZdZxZS)ConstantFlightServerzA Flight server that always returns the same data. See ARROW-4796: this server implementation will segfault if Flight does not properly hold a reference to the Table object. sthe expected criteriac bt||fi|tttd|_||_y)N)intsdictsmulti)super__init__rBrSr[table_factoriesoptions)selflocationrekwargs __class__s rrczConstantFlightServer.__init__s3 ,V,'(+   rc#K||jk(rJtjtjgtj j dgyyw)N/foo)CRITERIAr FlightInfor=schemaFlightDescriptorfor_pathrfcontextcriterias r list_flightsz!ConstantFlightServer.list_flightssL t}} $## " ''008  %sAAc|j|j}tj||jS)Nre)rdticketr RecordBatchStreamre)rfrrrwtables rdo_getzConstantFlightServer.do_gets44$$V]]35''t||DDrNN) __name__ __module__ __qualname____doc__rlrcrtrz __classcell__ris@rr]r]s (HErr]c@eZdZdZdfd ZdZdZedZxZ S)MetadataFlightServerz4A Flight server that numbers incoming/outgoing data.c 2t|di|||_yNr!rbrcrerfrerhris rrczMetadataFlightServer.__init__ "6" rctjgdg}tjj|dg}t j |j |j||jS)Nr6rYr;rv) r=r>r?r@r GeneratorStreamrnnumber_batchesre)rfrrrwrArys rrzzMetadataFlightServer.do_getsb HH( ) $$T#$7%% LL    &LL" "rc d}gd}|jjdk(sJ|D]\}}|jtjj tj ||ggdgsJ|Jtjd|j\} || k(sJ|jtjd||dz }|jjdk(sJ|jjdk(sJy)Nrr6rJrYstructunpack to_pybyteswritepacknum_record_batches) rfrr descriptorreaderwritercounter expected_databatchbufclient_counters rdo_putzMetadataFlightServer.do_puts+ ||((A--- JE3<< : :=1234! ? "?$mmD#..2BCONn, ,, LLT73 4 qLG ||((A---||..!333rc#Kt|jD]!\}}tjd|}||f#yw)Nr) enumerate to_batchesrr)ryidxrrs rrz#MetadataFlightServer.number_batchessA#E$4$4$67 JC++dC(C*  s?AN) r|r}r~rrcrzr staticmethodrrrs@rrrs)>"4"rrc6eZdZdZdfd ZdZdZdZxZS)EchoFlightServerz4A Flight server that returns the last data uploaded.c Bt||fi|d|_||_yr)rbrc last_messageexpected_schema)rfrgrrhris rrczEchoFlightServer.__init__s% ,V, .rc@tj|jSr)r rxrrfrrrws rrzzEchoFlightServer.do_gets''(9(9::rc||jr|j|jk(sJ|j|_yr)rrnread_allrrfrrrrrs rrzEchoFlightServer.do_puts2   ''6==8 88"OO-rc|D]}yrr!)rfrrrrrchunks r do_exchangezEchoFlightServer.do_exchanges E  rr{) r|r}r~rrcrzrrrrs@rrrs>/ ;. rrc"eZdZdZdZdZdZy)EchoStreamFlightServerz6An echo server that streams individual record batches.ctj|jj|jj dS)N max_chunksize)r rrrnrrs rrzzEchoStreamFlightServer.do_gets;%%    $ $    ( (t ( <> >rcgSrr!rfrrs r list_actionsz#EchoStreamFlightServer.list_actions rc|jdk(r/|j|jjdgStNwho-am-iutf-8rH peer_identitypeerencodeNotImplementedErrorrfrractions r do_actionz EchoStreamFlightServer.do_action9 ;;* $))+W\\^-B-B7-KL L!!rNr|r}r~rrzrrr!rrrrs@> "rrc"eZdZdZdZdZdZy)EchoTableStreamFlightServerz,An echo server that streams the whole table.cltj|jj|jgSr)r rrrnrs rrzz"EchoTableStreamFlightServer.do_gets0%%    $ $    ! !rcgSrr!rs rrz(EchoTableStreamFlightServer.list_actionsrrc|jdk(r/|j|jjdgStrrrs rrz%EchoTableStreamFlightServer.do_actionrrNrr!rrrrs6! "rrc"eZdZdZdZdZdZy)'EchoRecordBatchReaderStreamFlightServerzCAn echo server that streams the whole table as a RecordBatchReader.ctj|jj|jj gSr)r rrrn to_readerrs rrzz.EchoRecordBatchReaderStreamFlightServer.do_get s9%%    $ $    ( ( * +- -rcgSrr!rs rrz4EchoRecordBatchReaderStreamFlightServer.list_actionsrrc|jdk(r/|j|jjdgStrrrs rrz1EchoRecordBatchReaderStreamFlightServer.do_actionrrNrr!rrrr sM- "rrceZdZdZdZdZy)GetInfoFlightServerz)A Flight server that tests GetFlightInfo.ctjtjdtjfg|tj ddgtj dtj jddgtjdjtjddgd d d d S) NrYr grpc://test localhost2023-04-05T12:34:56.789012345nszendpoint app metadatarJ*Tzinfo app metadata) r rmr=rnint32FlightEndpointLocation for_grpc_tcpscalarcast timestamprfrrrs rget_flight_infoz#GetInfoFlightServer.get_flight_infos  IIRXXZ() * %%cM?;%%__11+tDEII=>CCBLLQUDVW+        rcd|j||}tj|jSr)rr SchemaResultrn)rfrrrinfos r get_schemazGetInfoFlightServer.get_schema1s)##GZ8""4;;//rN)r|r}r~rrrr!rrrrs3 &0rrc&eZdZdZedZdZy)ListActionsFlightServer'A Flight server that tests ListActions.c4ddtjddgS)N)action-1 description)zaction-2zaction-3z more detailr ActionType)clss rexpected_actionsz(ListActionsFlightServer.expected_actions9s$ (    j- 8  rc#@K|jEd{y7wr)rrs rrz$ListActionsFlightServer.list_actionsAs((***s N)r|r}r~r classmethodrrr!rrrr6s1  +rrceZdZdZdZy)ListActionsErrorFlightServerrc#Kddyw)N)rrrDr!rs rrz)ListActionsErrorFlightServer.list_actionsHs s N)r|r}r~rrr!rrrrEs 1rrc0eZdZdZdfd ZdZdZxZS)CheckTicketFlightServerzDA Flight server that compares the given ticket to an expected value.c 4t||fi|||_yr)rbrcexpected_ticket)rfrrgrhris rrcz CheckTicketFlightServer.__init__Ps ,V,.rc|j|jk(sJtjgdtjg}tj j |dg}tj|S)Nr6rGrYr;) rrwr=r>rr?r@r rx)rfrrrwdata1rys rrzzCheckTicketFlightServer.do_getTs`##v}}444-BHHJ?@$$U3%$8''..rc.|j|_yr)rr)rfrrrrs rrzCheckTicketFlightServer.do_putZs"OO-rr)r|r}r~rrcrzrrrs@rrrMsN// .rrc^eZdZdZej dej fgZdZy)InvalidStreamFlightServerzEA Flight server that tries to return messages with differing schemas.rYctjgdtjg}tjgdtjg}|j|jk7sJtj j |dg}tj j |dg}|j|jk(sJtj|j||gS)Nr6rG)g$gg@g$@rYr;) r=r>rfloat64rHr?r@rnr r)rfrrrwrdata2table1table2s rrzz InvalidStreamFlightServer.do_getcs-BHHJ?@7bjjlKLzzUZZ'''%%eC5%9%%eC5%9}} +++%%dkkFF3CDDrNr|r}r~rr=rnrrzr!rrrr^s+O RYYhbhhj)* +FErrc^eZdZdZej dej fgZdZy)NeverSendsDataFlightServerz0A Flight server that never actually yields data.rYc|jdk(r|jj|jjtjj t dg|jg}tj|j|Stj|jtj|jjS)N yield_datar8rn) rwrn empty_tabler=rr@ranger r itertoolsrepeat)rfrrrwrAs rrzz!NeverSendsDataFlightServer.do_getss ==M ) '') '')**E!H:dkk*JD ))$++t< <%% KK))$++*A*A*CDF FrNr r!rrrrns+: RYYhbhhj)* +F Frrc,eZdZdZdZdZedZy)SlowFlightServerz;A Flight server that delays its responses to test timeouts.ctjtjdtjfg|j S)NrY)r rr=rnr slow_streamrs rrzzSlowFlightServer.do_gets:%%bii#rxxz1B0C&D&*&6&6&8: :rc0tjdgSN?)timesleeprs rrzSlowFlightServer.do_actions 3 rc#Ktjgdtjg}tjj |dgt j dtjj |dgyw)Nr6rGrYr;r9)r=r>rr?r@rr)rs rrzSlowFlightServer.slow_streamse-BHHJ?@hh""5"66 2hh""5"66sB B N)r|r}r~rrzrrrr!rrrrs#E:77rrc2eZdZdZedZdZdZdZy)ErrorFlightServerz9A Flight server that uses all the Flight-specific errors.c tjtjtjtjtj t tjtdS)N)internaltimedoutcancelunauthenticated unauthorizednotimplementedinvalidr0) r FlightInternalErrorFlightTimedOutErrorFlightCancelledErrorFlightUnauthenticatedErrorFlightUnauthorizedErrorrr= ArrowInvalidKeyErrorr!rr error_caseszErrorFlightServer.error_casessH222211%@@"::1  rctj}|j|vr||jd|jdk(rd}tjd|t )NrDprotobufthis is an error message)r"r2rHr r/r)rfrrrr2err_msgs rrzErrorFlightServer.do_actions\'335 ;;+ %*+fkk*51 1 [[J &1G00@ @!!rc#Ktjtjgtjj dgtj dw)NrkrD)r rmr=rnrorpr+rqs rrtzErrorFlightServer.list_flightssN IIbM  # # , ,V 4   ((//sAA!c|jdk(rtjd|jdk(rtjd|jdk(rtjd|jdk(rtj d|jdk(rtj d|jdk(rd}tj d|y) NsinternalrDstimedoutscancelsunauthenticateds unauthorizedsprotobufr5)commandr r+r,r-r.r/)rfrrrrrr6s rrzErrorFlightServer.do_puts    ,,,U3 3   ; .,,U3 3   9 ,--e4 4   #5 533E: :   ? 2007 7   ; .1G00@ @/rN) r|r}r~rrr2rrtrr!rrr"r"s'C    "0 Arr"cBeZdZdZdfd ZdZdZdZdZdZ xZ S) ExchangeFlightServerz A server for testing DoExchange.c 2t|di|||_yrrrs rrczExchangeFlightServer.__init__rrc|jjdk(sJ|jtjj k7rt jd|jdk(r|j|||S|jdk(r|j|||S|jdk(r|j|||S|jdk(r|j|||St jd|j)Nrz!Must provide a command descriptorechogetput transformzUnknown command: ) rrdescriptor_typer DescriptorTypeCMDr=r0r9 exchange_echoexchange_do_getexchange_do_putexchange_transformrs rrz ExchangeFlightServer.do_exchanges||((A---  % %)>)>)B)B B//"EF F   7 *%%gvv> >   6 )''@ @   6 )''@ @   < /**7FFC C//#J$6$6#78: :rctjjtjt ddgdg}|j |j |j|y)Emulate DoGet with DoExchange.r(rYr;N)r=r?r@r>rbeginrn write_table)rfrrrrrAs rrFz$ExchangeFlightServer.exchange_do_getsXxx## HHU1i( )% $  T[[!4 rc d}|D]C}|jstjd|jjdk7sJ|dz }E|jj |k(sJ|j t|jdy)Emulate DoPut with DoExchange.rzAll chunks must have data.rJrN) rAr=r0rrrwrite_metadatastrr)rfrrrr num_batchesrs rrGz$ExchangeFlightServer.exchange_do_puts  E::oo&BCC<<,,1 11 1 K   ||..+===c+.55g>?rc.|jjdk(sJd}|D]}|s?|jr3|j|jj|j d}|j r3|jr'|j|j|j |j r|j|j |jr7|jjdk7sJ|j|jJdy)zRun a simple echo server.rFrvTzShould not happenN) rrrArLrnre app_metadatawrite_with_metadatarP write_batch)rfrrrrstartedrs rrEz"ExchangeFlightServer.exchange_echos||((A--- 2Euzz UZZ.. E!!ejj**5::u7I7IJ##%%e&8&89||00A555""5::.111u 2rc|jjdk(sJ|jD]L}tjj |j r-tjdt|z|j}|jjdk7sJdg|jz}|D]0}t|D] \}} ||xx| jz cc<"2tjjtj|gdg} |j!| j|j#| y)zSum rows in an uploaded table.rzInvalid field: sumr;N)rrrnr=types is_integerrHr0reprrnum_rowsras_pyr?r@r>rLrM) rfrrrrfieldrysumscolumnrowvalueresults rrHz'ExchangeFlightServer.exchange_transforms||((A---]] GE88&&uzz2oo&7$u+&EFF G!||((A---sU^^# +F'/ + US U[[]*  + +%%rxx~&6ug%F V]]#6"rr) r|r}r~rrcrrFrGrErHrrs@rr;r;s%*: ! @2$#rr;c.eZdZdZfdZdZdZxZS)HttpBasicServerAuthHandler7An example implementation of HTTP basic authentication.c0t|||_yrrbrccredsrfrjris rrcz#HttpBasicServerAuthHandler.__init__  rcz|j}tjj|}|j|j vrtj d|j |j|jk7rtj d|jt|jy)N unknown userzwrong password) r&r BasicAuth deserializeusernamerjr.passwordrr)rfoutgoingincomingrauths r authenticatez'HttpBasicServerAuthHandler.authenticatesmmo++C0 == *33NC C ::dmm $ 5334DE Ewt}}-.rcz|stjd||jvrtjd|S)Nztoken not providedrn)r r.rjrftokens ris_validz#HttpBasicServerAuthHandler.is_valid$s9334HI I  "33NC C rr|r}r~rrcrvrzrrs@rrfrfsA/rrfc.eZdZdZfdZdZdZxZS)HttpBasicClientAuthHandlerrgcft|tj|||_d|_yr)rbrcr ro basic_authryrfrqrrris rrcz#HttpBasicClientAuthHandler.__init__/s)  **8X> rc|jj}|j||j|_yr)r serializerr&ry)rfrsrtrus rrvz'HttpBasicClientAuthHandler.authenticate4s-((*t]]_ rc|jSrryrfs r get_tokenz$HttpBasicClientAuthHandler.get_token9 zzrr|r}r~rrcrvrrrs@rr}r},sA % rr}c.eZdZdZfdZdZdZxZS)TokenServerAuthHandler:An example implementation of authentication via handshake.c0t|||_yrrirks rrczTokenServerAuthHandler.__init__@rlrc|j}|j}||jvr:|j||k(r(|jtjd|zyt j d)Nsecret:zinvalid username/password)r&rjrbase64 b64encoder r.)rfrsrtrqrrs rrvz#TokenServerAuthHandler.authenticateDsf==?==? tzz !djj&:h&F NN6++J,AB C33+- -rctj|}|jdstjd|ddS)Nrz invalid token)r b64decode startswithr r.rxs rrzzTokenServerAuthHandler.is_validMs=  ' +33OD DQRyrr{rs@rrr=sD-rrc.eZdZdZfdZdZdZxZS)TokenClientAuthHandlerrcLt|||_||_d|_y)Nr)rbrcrqrrryrs rrczTokenClientAuthHandler.__init__Ws#      rc|j|j|j|j|j|_yr)rrqrrr&ryrfrsrts rrvz#TokenClientAuthHandler.authenticate]s/t}}%t}}%]]_ rc|jSrrrs rrz TokenClientAuthHandler.get_tokenbrrrrs@rrrTsD % rrceZdZdZdZdZy)NoopAuthHandlerzA no-op auth handler.cy)z Do nothing.Nr!rs rrvzNoopAuthHandler.authenticateisrcy)zV Returning an empty string. Returning None causes Type error. rr!rxs rrzzNoopAuthHandler.is_validls rN)r|r}r~rrvrzr!rrrrfsrrcz|D]6}|j|jk(s%|j|cSy)zcLookup the value of given key in the given headers. The key lookup is case-insensitive. N)lowerr)headers lookup_keyr0s rcase_insensitive_header_lookuprts9$ 99;***, ,;;s# #$rc"eZdZdZdZdZdZy)!ClientHeaderAuthMiddlewareFactoryz@ClientMiddlewareFactory that creates ClientAuthHeaderMiddleware.cg|_yrcall_credentialrs rrcz*ClientHeaderAuthMiddlewareFactory.__init__s !rct|Sr)ClientHeaderAuthMiddlewarerfrs r start_callz,ClientHeaderAuthMiddlewareFactory.start_calls )$//rc||_yrr)rfrs rset_call_credentialz5ClientHeaderAuthMiddlewareFactory.set_call_credentials .rN)r|r}r~rrcrrr!rrrr}sJ"0/rrceZdZdZdZdZy)ra ClientMiddleware that extracts the authorization header from the server. This is an example of a ClientMiddleware that can extract the bearer token authorization header from a HTTP header authentication enabled server. Parameters ---------- factory : ClientHeaderAuthMiddlewareFactory This factory is used to set call credentials if an authorization header is found in the headers from the server. c||_yrfactoryrfrs rrcz#ClientHeaderAuthMiddleware.__init__  rct|d}|r0|jjd|djdgyy)N Authorization authorizationrr)rrrr)rfr auth_headers rreceived_headersz+ClientHeaderAuthMiddleware.received_headerssB4WoN  LL , , A%%g..0 1 rN)r|r}r~rrcrr!rrrrs 1rrceZdZdZdZy)!HeaderAuthServerMiddlewareFactoryz)Validates incoming username and password.ct|d}|djd}d}d}|ddk(rjtj|d}|j djd }|dd k(r|dd k(st j |d }t|S|dd k(r|d}|d k(s*t j |t j |t|S)Nrr rzInvalid credentialsBasicrJr:testrr token1234Bearer)rsplitrrdecoder r.HeaderAuthServerMiddleware) rfrrrvaluesry error_messagedecodedpairs rrz,HeaderAuthServerMiddlewareFactory.start_calls4   Q%%c*- !9 &&vay1G>>'*005DGv%$q'Z*?77 FFE*%00AY( "1IEK'77 FF33MB B)%00rNr|r}r~rrr!rrrrs 31rrceZdZdZdZdZy)rzBA ServerMiddleware that transports incoming username and password.c||_yrrrxs rrcz#HeaderAuthServerMiddleware.__init__s  rc$dd|jziS)N authorizationzBearer rrs rsending_headersz*HeaderAuthServerMiddleware.sending_headerssTZZ!788rNr|r}r~rrcrr!rrrrsL9rrceZdZdZdZy)HeaderAuthFlightServerzz)ActionType(type='foo', description='bar')z?z,z locations=[] expiration_time=2023-04-05 12:34:56+00:00 app_metadata=b'endpoint app metadata'>z endpoints=[] total_records=1 total_bytes=42 ordered=True app_metadata=b'test app metadata'>z6z&z'z%rDrrUuserpassfoo2023-04-05T12:34:56sendpoint app metadatarJrTtest app metadatagrpc+tcp://localhost:1234)intr5z1z endpoints=[] total_records=1 total_bytes=42 ordered=True app_metadata=b'test app metadata'>rs grpc://testz2023-04-05T01:02:03)expiration_timeirXr8rKrLrT)r\r Actionrroro for_commandrr=rrrrmrnrprResultrTicketpytestraises TypeErrorobjectr) action_repraction_type_reprbasic_auth_reprdescriptor_repr endpoint_repr info_repr location_repr result_reprschema_result_repr ticket_reprendpointrs r test_reprrWsEKBWODO>M  -MM:KB9K  eS) *k 99 9 !!%/ 04D DD D   0 1_ DD D ''33E: ; NN N$$BII3499",,s:KL H >] ** *    " v..7792 2t) D : "" "  ;< = NN N  f% &+ 55 5 ##BIIbM2 37I II I ##BII/?.@$AB C; << <  f% &+ 55 5 ;;"))B- '' '    f%%..0" 2t) D  - : "" " ;;   y !# eT"# y !,fh+, y !Pem^VX%NOP y !PeR9NOP y !XeR$1aQRTU9VWX y !@eRfh?@@##,,PPPPXX@@sHP6Q"Q Q'Q( !Q46QQ QQ%(Q14Q=cddddddddd d d d d ddddddddg}|D]G}|\}}|\}}||k(sJ||k(sJ||k(sJ||k(sJ||k(sJ||k(sJ||k7rGJy)NcZtjddtjddfS)NrDrrUr rDr!rrztest_eq..s!uc*FMM%,EFrcZtjddtjddfS)NrDrbarrZr!rrr[ztest_eq..s!uc*FMM%,HIrcZtjddtjddfS)NrDrUrErr!rrr[ztest_eq..s(""5%0""5%02rcZtjddtjddfS)Nr9r:user2r ror!rrr[ztest_eq..s(!!&&1!!'624rcZtjddtjddfS)Nr9r:pass2rar!rrr[ztest_eq..s(!!&&1!!&'24rc~tjjdtjjdfS)NrD)r rorErpr!rrr[ztest_eq..s0((44U;((11%8:rcZtjdgtjdgfSNr;r]r rr!rrr[ztest_eq..s(&&vr2&&vr24rctjdtjdgtjdtjdgfS)Nr;r@grpc+tls://localhost:1234)r rrr!rrr[ztest_eq..sK  ! !)DEF H  ! !)DEF H rc 2tjdgtjdj tj dtjdgtjdj tj dfS)Nr;r<r=z2023-04-05T12:34:56.789ms)r rr=rrrr!rrr[ztest_eq..ss  ! !BII&;<AA",,sBST V  ! ! 3499",,t:LM OPrcbtjdgdtjdgdfS)Nr;rrCmetargr!rrr[ztest_eq..s-&&vrD&&vrHJrcDtjtjgtjj gtjtjdtj fgtjj gfSNints)r rmr=rnrorpr5r!rrr[ztest_eq..st    " ''002B 8    FBHHJ/01''002B 8 9rctjtjgtjj gtjtjgtjj dgfSNr;)r rmr=rnrorprEr!rrr[ztest_eq.. si    " ''002B 8    " ''33F;R A Brc ntjtjgtjj tj dggtjtjgtjj tj dggfSrf)r rmr=rnrorprr!rrr[ztest_eq..s    " ''002&&vr23 5    " ''002&&vr23 5 6rc"tjtjgtjj gdtjtjgtjj gdfS)N) total_recordsrJr rmr=rnrorpr!rrr[ztest_eq..sl    " ''002Bb J    " ''002Ba I Jrc"tjtjgtjj gdtjtjgtjj gdfS)Nru) total_bytesrrwr!rrr[ztest_eq..#sl    " ''002BB H    " ''002BB H Irc"tjtjgtjj gdtjtjgtjj gdfS)NF)orderedTrwr!rrr[ztest_eq..*sl    " ''002B G    " ''002B F Grc"tjtjgtjj gdtjtjgtjj gdfS)NrrCrmrwr!rrr[ztest_eq..1sl    " ''002BS J    " ''002BW N OrcVtjdtjdfS)Nr@ri)r rr!rrr[ztest_eq..8s"!<=!<=?rcVtjdtjdfSrf)r rFr!rrr[ztest_eq..:sv& f(=>rctjtjgtjtjdtjfgfSro)r rr=rnr5r!rrr[ztest_eq..;sD$$RYYr]3$$RYY0D/E%FGIrcVtjdtjdfS)Nrr;)r rGr!rrr[ztest_eq..=ss#V]]6%:;rr!)itemsgenlhs1rhs1lhs2rhs2s rtest_eqrsFI 2 4 4 : 4  P  J 9 B 6 J I G O ?> I;gT El U dU dt||t||t||t||t||t||t|| rctjtjgtjj g}tjtjgtjj gdd}tjtjgtjj gdd}|j dk(sJ|j dk(sJ|j dk(sJ|jdk(sJ|jdk(sJ|jdk(sJy)Nru)rvry)r rmr=rnrorprvry)fi1fi2fi3s rtest_flight_info_defaultsrLs    BIIbM6+B+B+K+K+Mr RC    " ((*Bbb RC    " ((*BdPT VC    "" "    "" "    "" " ??b  ??b  ??b  rcdddtfg}|D])}t|5}t|tsJ ddd+y#1swY6xYw)Ngrpc://localhost:0r)rr isinstance) locationsrgservers r$test_flight_server_location_argumentr^s`  n&'I 8 h ' 86f&67 77 8 88 8 8s ?A ctjt5t5t#1swYnxYw dddy#1swYyxYwr)rHrI ValueErrorr r!rr#test_server_exit_reraises_exceptionrisF z "   ,    s A 09 AAcDdtfd}fd}t5}tj|d}|j t j }|j dt j |z }|dk\sJ dddy#1swYyxYw)Nrcntjdtatj yr)rrr rservergsrrz-test_client_wait_for_available..serveus! 3!(+ rTtargetdaemonr8timeoutr)rr threadingThreadstartrwait_for_available)rrclientthreadrWelapsedrgs @rtest_client_wait_for_availableros^-.H F h 6!!t< ))+!!!!,))+'#~~s A,BBc\t5}tjd|jf5}t |j gk(sJ|j tj }tt |dk(sJ ddddddy#1swYxYw#1swYyxYw)zTry a simple list_flights call.rrJN)r]r connectportlistrtrllen)rrflightss rtest_flight_list_flightsrs  '6 NNK5 6':@F'')*b000%%&:&C&CD4=!Q&&& ''''''s#"B"ABB"B B""B+ct5}tjd|jf5}t |j gk(sJ|j |j tjtj5t |j dddddddddy#1swYxYw#1swYxYw#1swYyxYw)Nr) r]r rrrrtcloserHrIr=r0rrs rtest_flight_client_closers  (6 NNK5 6(:@F'')*b000   ]]2?? + ( $$& ' ( (((  ( ( ((((s;"CA"CC)C1CC CC CC#c<t}t5}tjd|jf5}|j tj dj}|j|sJ ddddddtjjtjjj}t|5}tjd|jf5}|j tj dj}|j|sJ|j tj djj}|j|sJ ddddddtj tj"d5td5}tjd|jf5}|j tj dj}dddddddddy#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYLxYw#1swYPxYw#1swYyxYw) Try a simple do_get call.rr_Nmetadata_versionrvz+expected IpcWriteOptions, got matchr)rBr]r rrrzrGrrr=ipcIpcWriteOptionsMetadataVersionV4rrHrIr)ryrrrAres rtest_flight_do_get_intsrs  E  "6 NNK5 6":@}}V]]734==?{{5!!!"" ff$$//22%4G g ."& NNK5 6":@}}V]]734==?{{5!!!}}V]]734>>@IIK{{5!!!"" v//J LD !" - D V[[9: D>D==w!78AACD D DDD!"""""""" D D D DDDs"IAI>I"I.>BI"I. J"J<3I:/J7JI II"I+ 'I..I7:J ?JJ JJct}t5}tjd|jf5}|j tj dj}t|d|jdjk(sJ ddddddy#1swYxYw#1swYyxYw)rrr_r:rN) rBr]r rrrzrG read_pandasrra to_pylist)ryrrrAs rtest_do_get_ints_pandasrs  E  F6 NNK5 6F:@}}V]]734@@BD%&%,,q/*C*C*EEEEFFFFFFs#"B8A#B,B8,B5 1B88Cc t}t5}tjd|jf5}|j tj d}|jjdk(sJ|j}|j|sJ|jtdddddk(sJ ddddddy#1swYxYw#1swYyxYw) Nrr`rJrrLrKrrrnum_dictionary_batchesnum_dictionary_deltasnum_replaced_dictionaries) rSr]r rrrzrGrrrrr)ryrrrrAs rtest_flight_do_get_dictsrs  E    6 NNK5 6  :@v}}X67||((A--- {{5!!!||y #$"#&'           s#"CBC :C C CC ctjgdtjg}tjj |dg}t d5}t jd|jf5}|jt jdj}|j|sJ ddddddy#1swYxYw#1swYyxYw) z+Make sure Tickets get passed to the server.r6rGrYr;s the-ticket)rrN) r=r>rr?r@rr rrrzrGrr)rryrrrAs rtest_flight_do_get_ticketrs XX) ; r)rr rrr rorErvryr{rTrnr=rr endpointsrrBrrrrrrrrs rtest_flight_get_infors  <&{FKK89%%f&=&=&I&I#&NO!!Q&&&2%%%|||  $8888{{bii#rxxz):(;<<<<4>>"a'''4>>!$../1444~~a 00888~~a --444~~a **1-1OOOO~~a 00 II; <tBLLu-./ //~~a --1IIII~~a **1- OO ( (d ;< <<#<<  !8!8!D!DS!IJ{{bii#rxxz):(;<<<<======s#B)A(B B)B& "B))B2c8t5}td|jf5}tjt j d5t|jdddddddddt5}td|jf5}t|jtjk(sJ ddddddy#1swY~xYw#1swYxYw#1swYxYw#1swY6xYw#1swYyxYw)z6Make sure the return type of ListActions is validated.rz3Results of list_actions must be ActionType or tuplerN) rr rrHrIr rrrrrrs rtest_list_actionsrs & '(6 +v{{3 4(8> ]]((-  ( $$& '  ((( ! "7f +v{{3 478>F'')* # 4 4 67 77777 ( (((((7777s^C8&C, C #C,+C8D1DD C) %C,,C5 1C88DD DDc&eZdZdZedZdZy)ConvenienceServerzT Server for testing various implementation conveniences (auto-boxing, etc.) c gdS)N)r;r]sbazr!rs rsimple_action_resultsz'ConvenienceServer.simple_action_resultss''rc|jdk(r |jS|jdk(r |jgS|jdk(rdgS|jdk(rtj|jdk(r fd}|Sy)N simple-actionecho bad-actionrDarrow-exceptionforeverc3ZKjsdjsyywrr) is_cancelled)rrsrrz(ConvenienceServer.do_action..gens&!..0 L"..0s%++)rHrbodyr=ArrowMemoryError)rfrrrrs ` rrzConvenienceServer.do_actions ;;/ )-- - [[F "KK= [[L (7N [[- -%%' ' [[I % !5L &rN)r|r}r~rpropertyrrr!rrrr s (( rrct5}td|jf5}|jdDcgc]}|j}}||j k(sJd}|jd|fDcgc]}|j}}||gk(sJ ddddddycc}wcc}w#1swYxYw#1swYyxYw)Nrrsthe-bodyr)rr rrrr)rrxresultsrs r!test_do_action_result_conveniencer$s   ! +v{{3 4 !8>$*#3#3O#DEa166EE&66666#)#3#3VTN#CDa166DD4&    ! ! !F E ! ! ! !s?B9B-B# *B-4B( B-B9# B--B6 2B99Cct5}td|jf5}tjt j d5t|jddddtjt j d5t|jddddddddddy#1swYbxYw#1swY&xYw#1swY*xYw#1swYyxYw)Nrza bytes-like object is requiredrrrr) rr rrHrIr rrrrs rtest_nicer_server_exceptionsr2s   6 +v{{3 4 68> ]]633!BD 1 !!,/ 0 1]]633!35 6 !!"34 5 6 6 6 6 1 1 6 6 6 6 6 6sRC)&C C$-CC,C4C)C CC CC& "C))C2ctd} |jdkDsJ |jy#|jwxYw)zMake sure port() works.rrN)rrshutdown)rs r test_get_portrAs7 !5 6F{{Qs 0Antz'Unix sockets can't be tested on Windows)reasonc,tj5}|jtjj |j }t|5t|5}|jtjd}t}|jj|jsJ|j}|j|sJ|jtjd}t}|jj|jsJ|j}|j|sJ dddddddddy#1swYxYw#1swYxYw#1swYyxYw)z3Try a simple do_get call over a Unix domain socket.rr_r`N)tempfileNamedTemporaryFilerr r for_grpc_unixnamer]r rzrGrBrnrrrS)sockrgrrryrAs rtest_flight_domain_socketrJs<  $ $ &&$ ??00; !8 4 &X& &*0]]6==#9:F%'E=='' 5 55??$D;;u% %%]]6==#:;F&(E=='' 5 55??$D;;u% %% & &&& & & & &&&s=AF  E>'C1E2E>!F 2E; 7E>>F F  Fcztjjtjt ddgdg}t |j 5}td|jf5}|jtjjd|j \}}|j|d|j|jtj dj#}|j%|sJ d d d d d d y #1swYxYw#1swYy xYw) zTry sending/receiving a large message via Flight. See ARROW-4421: by default, gRPC won't allow us to send messages > 4MiB in size. rirYr;)rrrrN)r=r?r@r>rrrnr rrr rorprMrrzrGrrrArrr_rds rtest_flight_large_messageras 88   q*+,!U  D $++ 6#& +v{{3 4#8>MM&"9"9"B"B6"J"&++/  4!12 v}}S12;;=}}T"""######s%D1/B$D%D1%D. *D11D:c`tjjtjt ddgdg}t 5}t d|jf5}|jtjjd|j\}}|j||j|jtj dj#}|j%|sJ ddddddy#1swYxYw#1swYyxYw) z?Try downloading a flight of RecordBatches in a GeneratorStream.rrKrYr;rrrN)r=r?r@r>rrr rrr rorprnrMrrzrGrrrs r'test_flight_generator_stream_of_batchesrws 88   q)$%!U  D !#V +v{{3 4#8>MM&"9"9"B"B6"J"&++/ 4  v}}S12;;=}}T"""######% D$#B#DD$D! D$$D-c tjjtjgdtjtj tj tjgdgddg}t5}td|jf5}|jtjjd|j\}}|j||j!|j#tj$dj'}|j)|sJ d d d d d d y #1swYxYw#1swYy xYw) z_ Try downloading a flight of RecordBatches with dictionaries in a GeneratorStream. rDrUrErDrD{iYii7rYrZr;rrrNr=r?r@r> dictionaryr5rNrr rrr rorprnrMrrzrGrrrs r1test_flight_generator_stream_of_batches_with_dictr 88   4rxxz27795 7 *+!3Z  D 1 2#f +v{{3 4#8>MM&"9"9"B"B6"J"&++/ 4  v}}S12;;=}}T"""######%E,+B#E E, E) %E,,E5c`tjjtjt ddgdg}t 5}t d|jf5}|jtjjd|j\}}|j||j|jtj dj#}|j%|sJ ddddddy#1swYxYw#1swYyxYw) z7Try downloading a flight of Table in a GeneratorStream.rrKrYr;rrrN)r=r?r@r>rrr rrr rorprnrMrrzrGrrrs r%test_flight_generator_stream_of_tablers 88   q)$%!U  D % &#& +v{{3 4#8>MM&"9"9"B"B6"J"&++/ 4  v}}S12;;=}}T"""######rc tjjtjgdtjtj tj tjgdgddg}t5}td|jf5}|jtjjd|j\}}|j||j!|j#tj$dj'}|j)|sJ d d d d d d y #1swYxYw#1swYy xYw) zW Try downloading a flight of Table with dictionaries in a GeneratorStream. rrrYrZr;rrrNrrs r/test_flight_generator_stream_of_table_with_dictrrrc`tjjtjt ddgdg}t 5}t d|jf5}|jtjjd|j\}}|j||j|jtj dj#}|j%|sJ ddddddy#1swYxYw#1swYyxYw) zCTry downloading a flight of RecordBatchReader in a GeneratorStream.rrKrYr;rrrN)r=r?r@r>rrr rrr rorprnrMrrzrGrrrs r3test_flight_generator_stream_of_record_batch_readerr s 88   q)$%!U  D 1 2#f +v{{3 4#8>MM&"9"9"B"B6"J"&++/ 4  v}}S12;;=}}T"""######rc tjjtjgdtjtj tj tjgdgddg}t5}td|jf5}|jtjjd|j\}}|j||j!|j#tj$dj'}|j)|sJ d d d d d d y #1swYxYw#1swYy xYw) zc Try downloading a flight of RecordBatchReader with dictionaries in a GeneratorStream. rrrYrZr;rrrNrrs r=test_flight_generator_stream_of_record_batch_reader_with_dictr rrcnt5}td|jf5}tjt j 5|jtjdjdddddddddy#1swYxYw#1swYxYw#1swYyxYw)z+Try streaming data with mismatched schemas.rrN) rr rrHrIr=ArrowExceptionrzr rGrrs r$test_flight_invalid_generator_streamrs " $9 +v{{3 498> ]]2,, - 9 MM&--, - 6 6 8 9999 9 99999s:B+$B3B:BB+B BB( $B++B4c t5}td|jf5}tjdd}tj d}t jtj5t|j||dddddddddy#1swYxYw#1swYxYw#1swYyxYw)z)Make sure timeouts fire on slow requests.rrrg?rrvN) rr rr rDrrHrIr,rr)rrrres rtest_timeout_firesrs  r3'**37]]655 6 < !!&'!: ; < <<<  < < <<< :CC c.t5}td|jf5}tjd}|j tj d|jddddddy#1swYxYw#1swYyxYw)z0Make sure timeouts do not fire on fast requests.rrrr_rvN)r]r rr rrzrGr)rrres rtest_timeout_passesrs  J6 +v{{3 4J8>**37 fmmG,g >GGIJJJJJJs#B A A?.B ?B B  Bctjjtjgdgdg}t 5}t d|j f5}tjtdg}|jtjd| j}|jtjdj}|jd k(sJ|jdk(sJ||k(sJ|tk(sJ d d d d d d y #1swYxYw#1swYy xYw) z"Make sure ReadOptions can be used.rWrZr;rrJ)included_fields) read_optionsrarvrKN)r=r?r@r>r]r rr rrrzrGr num_columnsr[)expectedrrre response1 response2s rtest_read_optionsrsxx##RXXl%;$**'<>MM&-- #&"((0  MM&--"9:CCE $$)))$$)))H$$$13333 4 4 4 4 4 4s%D9CD-D9-D6 2D99Etestsp4ssw0rd)rjcztt5}td|jf5}t j dd}t jtjd5t|j|dddddddddy#1swYxYw#1swYxYw#1swYyxYw)z,Test that auth fails when not authenticated. auth_handlerrrrz.*unauthenticated.*rN) rbasic_auth_handlerr rr rDrHrIr.rrrrrs rtest_http_basic_unauthr!s -? @+F +v{{3 4+8>z3/ ]]6<z3/6vzJK""6*=}}'')W444G}   ++-444555555s#CBCCC CC'c tt5}td|jf5}t j dd}t jtjd5|jtddt|j|d d d d d d d d d y #1swYxYw#1swYxYw#1swYy xYw) z-Test that auth fails with the wrong password.rrrrz.*wrong password.*rrwrongN) rrr rr rDrHrIr.rvr}r$rr s r%test_http_basic_auth_invalid_passwordr*8s -? @+F +v{{3 4+8>z3/ ]]6<z3/26:FG((01}}'')W444 555555s#B2A+B&B2&B/ +B22B;c Ltt5}td|jf5}t j t j5|jtdddddddddddy#1swYxYw#1swYxYw#1swYyxYw)r,rrrr)N) rr-r rrHrIr r.rvrrs rtest_token_auth_invalidr0Ms -? @IF +v{{3 4I8> ]]6<< = I    6vw G H IIII I IIIIIs:B$B B)B1BB BB BB#c ttdti5}td|jf5}|j dd}|ddk(sJ|dd k(sJ d d d d d d y #1swYxYw#1swYy xYw) zATest authenticate_basic_token with bearer token and auth headers.rurrrrpasswordrrrJBearer token1234N)rno_op_auth_handlerrr rauthenticate_basic_token)rr token_pairs rtest_authenticate_basic_tokenr8Ys -?13M 4  +v{{3 449?44WkJ !} 0000!} 3333 444444s"A9'A-A9-A6 2A99BcPttdti5}td|jf5}t j tj5|jdddddddddddy#1swYxYw#1swYxYw#1swYyxYw)z7Test authenticate_basic_token with an invalid password.rur2rrs badpasswordN) rr5rr rrHrIr r.r6rs r.test_authenticate_basic_token_invalid_passwordr:ds -?13M E  +v{{3 4E9? ]]6<< = E  + +G^ D E EEE E E EEEEs:B$BB+B3BB BB BB%c ttdti5}td|jf5}|j dd}|ddk(sJ|dd k(sJt j|g }t|jt jd d | }|djjdk(sJ ddddddy#1swYxYw#1swYyxYw)z@Test authenticate_basic_token and doAction after authentication.rur2rrr3rrrJr4r test-actionrrre token1234N) rr5rr rr6r rrrrDrrrrr7rerds r(test_authenticate_basic_token_and_actionrAns -?13M  ;  +v{{3 4 ;9?44WkJ !} 0000!} 3333**J<@f&&==4g'GHay~~((*l::: ; ; ; ; ; ;s#C"BCC"C C""C+c ttdti5}t}t d|j f|g}t jd}tjdd|zfg}t|jtjd d | }|d jjd k(sJ|jd dk(sJ|jddk(sJt|jtjd d | }|d jjd k(sJ|jd dk(sJ|jddk(sJ|j!dddy#1swYyxYw)zTest authenticate_basic_token with client middleware to intercept authorization header returned by the HTTP header auth enabled server. rur2rrs test:passwordrsBasic r<r=rr>rr?rJr4N)rr5rrr rrrr rrrrDrrrr)rclient_auth_middlewarerencoded_credentialsrerdresult2s r4test_authenticate_basic_token_with_client_middlewarerG}s -?13M  !B!D &++ &./ %../?@** y+>> ?4  f&&==4g'GHay~~((*l:::%55a8) B >B Bctjdjdk(sJtjdjdk(sJy)z,Test creating locations for unknown schemes.zs3://fooss3://foozhttps://example.com/bar.parquetshttps://example.com/bar.parquetN)r rurir!rrtest_location_unknown_schemerQsD ??: & * *k 99 9 ??< = A A* ++ +rct}t|d5}tdt|jz5}t j tj5|jtjdjdddddddddy#1swYxYw#1swYxYw#1swYyxYw)z>Make sure clients cannot connect when cert verification fails.r2tls_certificateszgrpc+tls://localhost:r_N) r4r]r rQrrHrIr FlightUnavailableErrorrzrGrcertsr=rs rtest_tls_failsrXs  E u^/D E= 03qvv;> ?=CI]]688 9 = MM&--0 1 : : < = === = = ====s:"C$B8 3B,B8C,B5 1B88C =CC cbt}t}t|d5}td|jf|d5}|j t jdj}|j|sJ ddddddy#1swYxYw#1swYyxYw)z"Try a simple do_get call over TLS.r2rSrr1)tls_root_certsr_N) rBr4r]r rrzr rGrrryrWr=rrAs rtest_tls_do_getr\s  E  E u^/D E"k166*$)+$6 8";A}}V]]734==?{{5!!! """"""s$B%ABB%B" B%%B.ct}t}t|d5} td|jfd}jtjdj}|j|sJ|jdddy#t $rt jdY~wxYw#1swYyxYw) zDTry a simple do_get call over TLS with server verification disabled.r2rSrTdisable_server_verificationz4disable_server_verification feature is not availabler_N)rBr4r]r rrrHskiprzr rGrrrr[s r$test_tls_disable_server_verificationras  E  E u^/D E P!;"7>BDF}}V]]734==?{{5!!! # P KKN O P s.CBACB>;C=B>>CC ct}t|d5}tjd|jf|dd5}t j tj5|jtjddddddddddy#1swYxYw#1swYxYw#1swYyxYw) z5Check that incorrectly overriding the hostname fails.r2rSrr1 fakehostname)rZoverride_hostnamer_N) r4r]r rrrHrIrUrzrGrVs rtest_tls_override_hostnameres  E u^/D E2 QVV,&+K&8)7 92=C ]]688 9 2 MM&--0 1 2 222 2 2 2222s;(B<$B0&%B$ B0B<$B- )B00B9 5B<<Cc^tjgdg}tjj|dg}g}t 5}t d|j f5}|jtjd}d}|D]H\}}|j|tjd|j\} || k(sJ|dz }Jtjj|}|j|sJ d d d d d d y #1swYxYw#1swYy xYw) z'Try a simple do_get call with metadata.r6rYr;rrrrrJN)r=r>r?r@rr rrzr rGrrrr from_batchesr) rArybatchesrrrrrmetadata server_idxs rtest_flight_do_get_metadatarks $% D HH cU 3EG   "6 +v{{3 4 "8>v}}S12% OE8 NN5 ! --h.A.A.CDKJ*$ $$ 1HC   xx$$W-{{5!!! " " " " " "s%D#B&DD#D D##D,cTg}g}t5}td|jf5}|jt j d}d} |j \}}|j|tjd|j\}||k(sJ|dz }X#t$rYnwxYw dddn #1swYnxYwdddn #1swYnxYwt5}td|jf5}|jt j d}d}|D]H\}}|j|tjd|j\}||k(sJ|dz }J dddn #1swYnxYwdddn #1swYnxYw||k(sJy)z0Verify the iterator interface works as expected.rrrrrJN) rr rrzr rG read_chunkrrrr StopIteration) batches1batches2rrrrrrirjs r1test_flight_metadata_record_batch_reader_iteratorrqsHH   6 +v{{3 4 8>v}}S12 "("3"3"5x&$mmD(2E2E2GH j(((q !          6 +v{{3 48>v}}S12% OE8 OOE " --h.A.A.CDKJ*$ $$ 1HC    x  slC(CAB'&C' B30C2B33C7 CC CC+FA4F8 FF FFc"tjjtjgdgdg}tjj tjj j}t|5}td|jf5}|jtjd}|j}|j|sJ ddddddy#1swYxYw#1swYyxYw) z2Try a simple do_get call with V4 metadata version.r6rYr;rrvrrN)r=r?r@r>rrrrrr rrzr rGrr)ryrerrrrAs rtest_flight_do_get_metadata_v4rs4s HH % &'u ! 6Eff$$//22%4G g ."& +v{{3 4"8>v}}S12 {{5!!! """"""s%DAC9(D9D >DDc tjgdg}tjj|dg}t 5}t d|j f5}|jtjjd|j\}}|5t|jdD]n\}}tjd|}|j!|||j#} | Jtj$d| j'\} || k(rnJ d d d d d d d d d y #1swYxYw#1swYxYw#1swYy xYw) z'Try a simple do_put call with metadata.r6rYr;rrrJrrN)r=r>r?r@rr rrr rorprnrrrrrUr&rr) rAryrrrmetadata_readerrrrirrjs rtest_flight_do_put_metadatarvAsM $% D HH cU 3E   )6 +v{{3 4 )8>"(--  # # , ,R 0 LL# )'(8(8q(8(IJ ) U!;;tS1**5(;%**,&$mmD#..2BC j(((  ) ) ) ) )  ) ) ) ) ) )sCE?EB E&E*E2EE EE EE$c tjjtjt j dt j gdg}t5}td|jfd5}|jtjjd|j\}}|5tj tj"d 5}|j%|d d d j&j(dk(sJ|j+d d |j+d g}|D]}|j%| d d d tj,j/|g}|j1tj2dj5} || k(sJ d d d d d d y #1swYxYw#1swYxYw#1swY*xYw#1swYy xYw)z+Try a simple do_put call with a size limit.i)dtyperYr;ri)write_size_limit_bytesrzexceeded soft limitrNrir)r=rr@r>nponesr5rr rrr rorprnrHrIFlightWriteSizeExceededErrorrVrclimitslicer?rgrzrGr) large_batchrrrruexcinfosmaller_batchesrractuals rtest_flight_do_put_limitrWs..,, BHHJ/0.U-K  "vk6;;/,0 2"5;"(--  # # , ,R 0   #  *vBB%:< 0?F"";/ 0==&&$. ..!!!S)!!#&O) *""5) * *88((+7v}}S12;;=6!!!'""" 0 0 * * """"sV(G/?G#&G'G 9AGA!G#:G/ G GG G##G, (G//G8ct5}td|jf5}|jt j d}|j tjtjd5|jdddddddddy#1swYxYw#1swYxYw#1swYyxYw)z4Test canceling a DoGet operation on the client side.rr_z(?i).*cancel.*rN) r]r rrzr rGr&rHrIr-rm)rrrs rtest_cancel_do_getrts   6 +v{{3 4 8>v}}W56  ]]666!13              s;B?AB3=B'B3B?'B0 ,B33B< 8B??Cct5}td|jf5}|jt j dt jt jt jt jfd}t j|d}|jjdjj|jd5jsJ d d d d d d d d d y #1swYxYw#1swYxYw#1swYy xYw) z5Test canceling a DoGet operation from another thread.rr_cjjjd jy#tj$r+5jdddYy#1swYYyxYwwxYw)Nr8r)rmsetwaitr r-)raised_proper_exceptionread_first_messager result_lockstream_canceledsr block_readz/test_cancel_do_get_threaded..block_readsv      " " $   + 2!!#.. 2 2+//1222 2s)ABA6,B6B ;BBTrr8rrJN)rr rrzr rGrEventLockrrrr&rjoinis_set) rrrrrrrrrs @@@@@rtest_cancel_do_get_threadedrs  4v +v{{3 448>v}}W56&__.#//+nn& "+//"3 2 2!!DA *  A   4*113 33 454444 4 454444s;EC+E E'E /EE E  E EE!ct5}td|jf5}|jt j dd}t |jdk(sJ~ddddddy#1swYxYw#1swYyxYw)Nrrrr;)rr rrr rDr$r)rrrs rtest_streaming_do_actionrs~   +v{{3 48>""6==C#@AG}!!V+++  s#BAA5$B5A> :BB c tjdd}|tjj|jk(sJtjd}|tjj|jk(sJtj d}|tj j|jk(sJtj dd}|tj j|jk(sJtjtjdtjfg}|tjj|jk(sJtjjd}|tjj|jk(sJtjjdd d }|tjj|jk(sJtjtjdtjfg|tjd d gtjd tj j#d dgtj$dj'tj(ddgdddd}tjj|j}|j|jk(sJ|j*|j*k(sJ|j,|j,k(sJ|j.|j.k(sJ|j0|j0k(sJ|j2|j2k(sJ|j4|j4k(sJtj|d tj j#d dgtj$dj'tj(dd}|tjj|jk(sJy)z(Make sure serializable types round-trip.action1s action1-bodyrDsresult1 username1 password1rYrrZz test.arrowrrrrrrr>rJrTr?r<r=N)r rDrprrGrFrorr=rnrrorErprmrrrrrrrryrvr{rTr) rrwrdr schema_resultdescrinfo2rVs rtest_roundtrip_typesrs ]]9o 6F V]]..v/?/?/AB BB B ]]5 !F V]]..v/?/?/AB BB B ]]: &F V]]..v/?/?/AB BB B!!+{;J ))55j6J6J6LM MM M'' C3D2E(FGM F//;;!# ## #  " " . .v 6D 6**66t~~7GH HH H  " " + +Cl CD 6**66t~~7GH HH H    C$%&  ! !#  7  ! !--k4@A 9:?? T@RS(       D"    ) )$..*: ;E ;;%,, && & ??e.. .. .   u00 00 0   !4!4 44 4 <<5== (( (    2 2 22 2 >>U__ ,, ,$$ 44[$GH '(--bll3.?@ H v,,889K9K9MN NN Nrc t5}td|jf5}tjj D]S\}}t j |d5t|jtj|ddddUt j tjd5t|jdddtjgdg}tjj!|dg}tjtj"tj$tj&tj(d }|j D]\}}t j |d5|j+tj,j/||j0\} } | j3|| j5dddt j |d5|j+tj,j/||j0\} } | j5ddd ddddddy#1swYBxYw#1swYxYw#1swYxYw#1swY3xYw#1swYExYw#1swYyxYw) z:Ensure that Flight errors propagate from server to client.rz.*foo.*rrNr6rYr;)r$r%r&r'r()r"r rr2rrHrIrrr rDr+rtr=r>r?r@r,r-r.r/rrorErnrMr) rrargexc_typerAry exceptionsr9 exceptionrrs rtest_roundtrip_errorsrsJ  ! +v{{3 4!8>.::<BBD @MCxy9 @V%%fmmC&=>? @ @ @]]655Y G ( $$& ' (,-.$$T#$7222211%@@"::  #-"2"2"4  GYy : !'++77@LL""""5)   y : !'++77@LL""     )!!! @ @ ( (     ;!!!!sJ=AJ1#/I> /J1J B=J1AJ 6J1A J$ " J1-J=>J J1 J J1J! J1$J. )J11J: 6J==Kc tjgdg}tjj|dg}t 5}t d|j f5}|jtjjd|j\} dg fd}tj|}|j|jd }|5t!|D]-\}} t#j$d |} |j'| | /|j)|j+d d d  dt-|k(sJ d d d d d d y #1swY.xYw#1swYxYw#1swYy xYw) z7Ensure that separate threads can read/write on a DoPut.r6rYr;rrrcdjdxxdz cc<jyy)NrrJ)r&)countrusr_reader_threadz:test_do_put_independent_read_write.._reader_threads0!&&(4aA "&&(4rrrJrrN)r=r>r?r@rr rrr rorprnrrrrrrrrU done_writingrr) rAryrrrrrrhrrrirrus @@r"test_do_put_independent_read_writer sj $% D HH cU 3E  (6 +v{{3 4(8>"(--  # # , ,R 0 LL# !!8 """3  '0 < U!;;tS1**5(; <    ! KKM Qx3w<'''5(((   !((((s>FBE4AE(;E4F(E1 -E44E= 9FF c xtdti5}td|jf5}t |j t jdd}t|dk(sJ|djj}d|k(sJ d d d d d d y #1swYxYw#1swYy xYw) z@Ensure that server middleware run on the same thread as the RPC.rrCrrrrJrs right valueN) rrr rrrr rDrrr)rrrrcs r"test_server_middleware_same_threadr3s -/( '  +v{{3 4'9?v'' gs(CDE7|q    **,&&&''''''s#B0A#B$B0$B- )B00B9c tdti5}td|jf5}t j t j5t|jdddt j tj5t|jtjdddddtd|jftg}t|jtjdd}d|j j#k(sJ ddddddy#1swYxYw#1swYxYw#1swY*xYw#1swYyxYw)z-Test rejecting an RPC with server middleware.rrCrNrr3)rrr rrHrIr=ArrowNotImplementedErrorrrr r.rrDrr$rr)rrresponses rtest_middleware_rejectr?s= 46( 9  +v{{3 499? ]]266 7 ( $$& ' (]]6<< = < !!&--S"9: ; < &++ &<>? ((sC)@ABhmm668888#999  ( ( < <9999sSE4$E(E-+E(/EA7E(?E4E E(E% !E((E1 -E44E=c t}t}td|i5}td|jf|g5}t j jd}tjt5t|jdddtjt5|j|dddtjt5|j|dddtjt5|jt j ddddtjt5|j#|t%j&g\}}|j)dddtjt5t|j+t j,dddddtjt5t|j/dddtjt5|j1|\}}|j)dddt j2j4t j2j6t j2j8t j2j:t j2j<t j2j>t j2j@t j2jBg}|jD|k(sJ|jD|k(sJ ddddddy#1swYxYw#1swYxYw#1swYxYw#1swYZxYw#1swY xYw#1swYxYw#1swYxYw#1swYcxYw#1swYzxYw#1swYyxYw)z/Test that middleware records methods correctly.rrCrrN)#rrr r rr rorErHrIrrrtrrrzrGrr=rnrrrDrrr LIST_FLIGHTSGET_FLIGHT_INFO GET_SCHEMADO_GETDO_PUT DO_ACTIONr DO_EXCHANGEr )server_middlewareclient_middlewarerrrrrrs rtest_middleware_mappingrUs8:8: f.?%@ A%5V &++ &)* %5 ,,88= ]]. / ( $$& ' ( ]]. / /  " ": . / ]]. / *   j ) * ]]. / . MM&--, - . ]]. /  j"))B-@IFA LLN ]]. / < !!&--S"9: ; < ]]. / ( $$& ' ( ]]. / **:6IFA LLN     , ,    / /    * *    & &    & &    ) )    , ,    + +  !((H444 ((H444K%5%5%5 ( ( / / * * . .   < < ( (  -%5%5%5%5sO"9O7M.!O2M;!O%N7!O%N=!O9N"!O8/N/'!ON<"!O%O (C4OO".M8 3O;N ON ON O"N, 'O/N9 4O<O O O OO O""O+c t5}td|jf5} t|j t j ddJ#t j$r+}|jJ|j}|dk(sJYd}~nd}~wwxYw dddn #1swYnxYwdddy#1swYyxYw)Nrr4rr5) r"r rrrr rDr/ extra_info)rreeis rtest_extra_infors  5 +v{{3 458> 5 !!&-- C"@A B 5-- 5<<+ ++B44 44 55555555sEB5B 0AB(!B B BB  B5 B) %B55B>ct}t}t|ddgd|d5}td|jf|d|ddj |ddj 5}|jtjdj}|j|sJ d d d d d d y #1swYxYw#1swYy xYw) z!Test mutual TLS (mTLS) with gRPC.r2rTr1)rT verify_clientroot_certificatesr)rZ cert_chain private_keyr_N) r4rBr]r rr/r0rzr rGrr)rWryr=rrAs r test_mtlsrs  E  E #N3A67#K0 2 "67 !&& ! -^,Q/44n-a044  6 ":@}}V]]734==?{{5!!! " " " " " "s$?C,AC3CC CCctjjtjt ddgdg}t 5}t d|jf5}tjjd}|j|\}}|5|j}ddd|k(sJ ddddddy#1swY"xYw#1swYxYw#1swYyxYw)rJrrKrYr;rr?N) r=r?r@r>rr;r rr rorErr)rrrrrrrys rtest_doexchange_getrsxx## q)$%%U$H  !6 +v{{3 4!8>,,88@ ++J7  &OO%E &5   !!! & & !!!!s< C##6CC *C:C# C CC C##C,c4tjjtjt ddgdg}|j d}t 5}td|jf5}tjjd}|j|\}}|5|j|j|D]}|j||j!|j#}|j$Jt't)|j+d } |j,| k(sJ|j.j0dk(sJ d d d d d d d d d y #1swYxYw#1swYxYw#1swYy xYw) rOrrKrYr;rrr@Nr)r=r?r@r>rrr;r rr rorErrLrnrVrrmrArQrrrTrr) rArhrrrrrrr expected_bufs rtest_doexchange_putrss 88   q)$%!U  DooCo0G  26 +v{{3 428>,,88@ ++J7  2 LL %  *""5) *    !%%'E::% %%s7|,33Gr9rNrJ)r=r?r@r>rrr;r rr rorErrQrrPrmrArTrLrnrVrrrrU) rArhrrrrrirrrRrs rtest_doexchange_echors} 88   q)$%!U  DooCo0G  !F6 +v{{3 4!F8>,,88A ++J7  F2Y 1!fmmG,%%c*))+zz)))))S000  1 LL %K  F""5)}} 333))+zzU***))111q ||66+EEE F&g. F5!fmmG,**5#6))+zzU***))S000q ||66+EEE F- F !F!F!F F F !F!F!F!FsQI856I,+A'I BI 2BI I I,I8 I) %I,,I5 1I88JcZtjjtjt ddgdg}|j d}tj jtj jj}t|5}td |jf5}tjjd }|j!|\}}|5|j#|j$||D]]}|j'||j$|j$k(sJ|j)} | j*|k(sJ| j,]J d d d d d d d d d y #1swYxYw#1swYxYw#1swYy xYw) z;Try a DoExchange echo server using the V4 metadata version.rrKrYr;rrrrvrr>N)r=r?r@r>rrrrrrr;r rr rorErrLrnrVrmrArT) rArhrerrrrrrrs rtest_doexchange_echo_v4rs~ 88   q)$%!U  DooCo0Gff$$//22%4G g . 2& +v{{3 4 28>,,88A ++J7  2 LLgL 6  2""5)}} 333))+zzU***))111  2 2 2 2 2 2 2 2 2 2 2sC!F!96F/Arr;r rr rorErrLrnrMrr)rArrrrrrrys rtest_doexchange_transformr sj 88   q$  q$  q$ !  D xx## q,*+%W$H   !6 +v{{3 4 !8>,,88F ++J7  & LL %   t $    !OO%E  & 5    ! ! ! & & ! ! ! !s= F!6FA F$F4FF FF FF&c tdti5}t}td|jf|g5}t |j tjdd}|jjjd}tj|}tjj!D]l\}}|j#}t%|t&r|jd}|j)||k(sJ|j*j)||k(rlJ ddddddy#1swYxYw#1swYyxYw)rrrCrrrasciiN)rr'rr rr$rr rDrrrast literal_evalrr#rrrbytesrr)rrrr raw_headersrheaderrs rtest_middleware_multi_headerr! sC 24- B 46 fkk*#9& B)/F,,V]]3-DEFH"--224;;GDK --k:N"="F"F"L"L"N Bfe,#]]73F%))&1V;;;++//76AAA  B B BB B B BBs)%EC6E3E7EE EEct}t|d5}dg}tjd|jf|d|}t j tj5|jtjdddd|jd g}tjd|jf|d|}t j tjtjf5|jtjdddd|jdddy#1swYxYw#1swY.xYw#1swYyxYw) z$Test setting generic client options.r2rS)zgrpc.ssl_target_name_overridercrr1)rZgeneric_optionsr_N)zgrpc.max_receive_message_length )r4r]r rrrHrIrUrzrGrr=r0r-)rWr=rers rtest_generic_optionsr7 s$  E u^/D EDEaff 5/4[/A079]]688 9 2 MM&--0 1 2 :;aff 5/4[/A079]]BOOV-H-HI J 2 MM&--0 1 2 !  2 2 2 2s=AE (%E A5E %E'E E E E E  E)ceZdZdZdZdZy)CancelFlightServerzA server for testing StopToken.ctjg}tjjg|}t j |t j|S)Nr)r=rnrr@r rrr)rfrrrwrnr$s rrzzCancelFlightServer.do_getR sF2 ^^ ' '6 ' :%%fi.>.>r.BCCrc"tjg}tjjg|}|j ||j s8|j |tjd|j s7yy)Nrr) r=rnrr@rLrrVrr)rfrrrrrrnr$s rrzCancelFlightServer.do_exchangeW sg2 ^^ ' '6 ' : V&&(   r " JJsO&&(rN)r|r}r~rrzrr!rrrrO s)D rrctjjtjjk7rt j ddt tjffd}t5}td|jf5}|jtjd}||jtj j#d}|j%|\}}||j |j'ddddddy#t tj(f$rY,wxYw#1swY1xYw#1swYyxYw)Nz'test only works from main Python threadcttjdtjtjyr)rrsignal raise_signalSIGINTr!rrsignal_from_threadz*test_interrupt..signal_from_threade s 3FMM*rc tj}tj5}|j |ddd|j  j}t|tjt fs-t|jtjt fsJyy#1swYtxYw#j wxYw#t $rtjdYwxYw)Nrz2KeyboardInterrupt didn't interrupt Flight read_all) rrrHrIrrKeyboardInterruptfailrcrr=ArrowCancelled __context__)rtexc_infor exc_typesrs rrztest_interrupt..testk s N $$,>?]]9-GGIJ NN!b//1BCD q}}r'8'8:K&L M NN ME  N KKL M Ns:+CB:CC:C?CCCC<;C<rrr>)rcurrent_threadident main_threadrHr`rr=rrr rrzr rGrrorErrr-)rrrrrrrrs @@rtest_interruptr` s&!''9+@+@+B+H+HH =>+#B$5$56IN$    +v{{3 4 8>v}}S12 V__,,88A ++J7 V__  LLN   "6#>#>?       sC>EA/&J25J -J27J&J J&'J2/J>I I  I I4I$ I''I1 ,I44I> 9JJ J J2J# J&&J/ +J22J; 7J>>KceZdZdZgZdZy)ActionNoneFlightServerz@A server that implements a side effect to a non iterable action.c|jdk(r/tj|jj dgS|jdk(r|jj dyt )N get_valuerrT)rHjsondumpsVALUESrrrrs rrz ActionNoneFlightServer.do_action sV ;;+ %JJt{{+227;< < [[H $ KK  t $!!rN)r|r}r~rrrr!rrrr sJ F"rrct5}td|jf5}|jt j dd|jt j dd}t jt|jjdgk(sJ ddddddy#1swYxYw#1swYyxYw)zEnsure that actions are executed even when we don't consume iterator. See https://issues.apache.org/jira/browse/ARROW-14255 rrrrTN) rr rrr rDrloadsr$rr)rrrs rtest_none_action_side_effectr  s !?V +v{{3 4?8>x56   V]];< =zz$q',,1134>>> ??????s#CBB<+C<C CCc  d}d}tj|| tjjd}t j dt j fg}G fddtj}|5}td|jf5}|j||\} fd }tj|d } | jtjtj5} |j!t j"d gg| )#1swYnxYw j$j&|k(sJtjtj5} |j)dddn #1swYnxYw| j$j&|k(sJ| j+|j-|\} fd}tj|d } | jtjtj5} |j/d#1swYnxYw| j$j&|k(sJtjtj5} |j)dddn #1swYnxYw| j$j&|k(sJ| j+dddn #1swYnxYwdddy#1swYyxYw)z Ensure that exceptions during writing preserve error context. See https://issues.apache.org/jira/browse/ARROW-16592. rDr])rrr5c"eZdZfdZfdZy)0test_write_error_propagation..FailServercrr!rfrrrrrexcs rrz7test_write_error_propagation..FailServer.do_put Ircrr!rs rrz.FailServer.do_exchange rrN)r|r}r~rr)rsr FailServerr  s   rrrc\ j#tj$rYywxYwr)r&r FlightErrorrsr_readerz-test_write_error_propagation.._reader s1 KKM%%   ++TrrJrNc\ j#tj$rYywxYwr)rmr rrsrrz-test_write_error_propagation.._reader s4 %%'%%  rr0)r r-rorEr=rnr5r r rrrrrrHrIrVr6rcrrrrrP) expected_message expected_inforrnrrrrrrrrrs @@rtest_write_error_propagationr sM  % %] 4C((44S9J YY"((*-. /FV,, / +v{{3 4/8>z6: !!>  ]]666 7 J8""2??QC5#HI J J~~((M999 ]]666 7 8 LLN   ~~((M999  ++J7 !!>  ]]666 7 ,8%%d+ , ,~~((M999 ]]666 7 8 LLN   ~~((M999 _//////s| K<!A%K'+D11D: 6AK';F K'F BK'/II AK' J' K''J0 ,2K' K<'K0 ,K<<Lc.tjdy)z Ensure that the gRPC server is stopped at interpreter shutdown. See https://issues.apache.org/jira/browse/ARROW-16597. zarrow_16597.pyN)r invoke_scriptr!rrtest_interpreter_shutdownr0 s  '(rceZdZdZdZy)TracingFlightServerz/A server that echoes back trace context values.cf|jdj}d|jDS)Ntracingc3NK|]\}}|d|jdyw)z: rN)r).0r0rcs r z0TracingFlightServer.do_action..A s2; S%55'"**73;s#%)r trace_contextr)rfrrrr's rrzTracingFlightServer.do_action< s3..y9GG ;$1$7$7$9; ;rNrr!rrr!r!9 s 9;rr!c(tdtji5}td|jf5}tj ddg}|j d|D]} ddddddy#1swYxYw#1swYyxYw) Nr#rCr)s traceparents700-000ff00f00f0ff000f0f00ff0f00fff0-000f0000f0f00000-00)s tracestaterr<)rrrv)r!r TracingServerMiddlewareFactoryr rrr)rrrercs r test_tracingr*E s  v<<>)   +v{{3 49?** 5 4   %%j'%B E  s"B1A<+B<B BBctdd}d}tjt|5|j t j jdddddy#1swYyxYw)Nzgrpc+tls://localhost:9643Tr^zRArgument 'schema' has incorrect type \(expected pyarrow.lib.Schema, got NoneType\)rrDr)r rHrIrJrr rorE)rmsgs r.test_do_put_does_not_crash_when_schema_is_noner-W sc 56:.HeadersTrailersFlightServerc|jdd|jdd|jdd|jddtjt j g|gS) Nx-header header-value x-header-binz headervalue x-trailer trailer-value x-trailer-binz trailervalue) add_header add_trailerr rmr=rnrs rrzJtest_headers_trailers..HeadersTrailersFlightServer.get_flight_infoe sh   z> :   ~/@ A    _ =   1C D$$ "  rN)r|r}r~rr!rrHeadersTrailersFlightServerr0d s rr:ceZdZdZfdZy)?test_headers_trailers..HeadersTrailersMiddlewareFactorycg|_yrr<rs rrczHtest_headers_trailers..HeadersTrailersMiddlewareFactory.__init__q s DLrc|Srr!)rfrHeadersTrailersMiddlewares rrzJtest_headers_trailers..HeadersTrailersMiddlewareFactory.start_callt s,T2 2rN)r|r}r~rcr)r?sr HeadersTrailersMiddlewareFactoryr<p s   3rr@ceZdZdZdZy)8test_headers_trailers..HeadersTrailersMiddlewarec||_yrrrs rrczAtest_headers_trailers..HeadersTrailersMiddleware.__init__x s "DLrc|jD]3\}}|D])}|jjj||f+5yr)rrrr)rfrr0rrcs rrzItest_headers_trailers..HeadersTrailersMiddleware.received_headers{ sF&}} > V#>ELL((//e => >rN)r|r}r~rcrr!rrr?rBw s  # >rr?rrCr)r2r3)r4s headervalue)r5r6)r7s trailervalueN) r rrr rrr rorpr)r:r@rrrr?s @rtest_headers_trailersrEa s &6 3+B3>$4>/0G $ &I& +v{{3 JINTv66??CD+w>>>3wFFF-@@@5HHH IIIIIIs$C'A/C C'C$ C''C0c J tdddddtdddddd G fdd t}|5}td |jf5}t }dD] }t j j|}|j|t jtjjd  \}}|5|j|jtjj|dk( |j||j!|j#}dddj%|sJ|j& |k(r J ddddddy#1swYExYw#1swYxYw#1swYyxYw)NrrLrKrJrr) dict_deltasdict_replacementceZdZfdZy)Dtest_flight_dictionary_deltas_do_exchange..DeltaFlightServerct}|j}|j|sJ|j|jj k(sJ|jdk(rNt jjd}|j|j||j||jdk(r-|j|j|j|yy)Ns dict_deltasTemit_dictionary_deltasrvsdict_replacement) rSrrrr9rr=rrrLrnrM) rfrrrrrexpected_tablereceived_tablereexpected_statss rrzPtest_flight_dictionary_deltas_do_exchange..DeltaFlightServer.do_exchange s/1N#__.N!((8 88<<>*2D2D2K2K2M#NN NN!!^3&&000M ^22G D"">2!!%88 ^223"">29rN)r|r}r~r)rPsrDeltaFlightServerrJ s 3rrQrTrL) write_optionsrvrG)rr]r rrSr rorErrr=rrrLrnrMrrrr) rQrrrNr9rrrrOrPs @r)test_flight_dictionary_deltas_do_exchangerS s  #$"#&'  & #$"#&'  N" 30 3  ; +v{{3 4;8>+-: ;G00<2##%!'!2  3"((8 88<<>'#:: ::% ;;;; 3 3;;;;sCFA:F  A0F =-F ,F 0FF F  F FF"c|jdk(rdddddS|jdk(rgddgtjjdtjj d tjj j dSiS) NdefaultrL)rrrRrallr)sabcsdefzstd) compressionF) use_threadsensure_alignment)paramr=rrr AlignmentDataTypeSpecific)requests rcall_options_argsr_ s}} !!   % ()VV333GFF11!!#!1!1!B!B2    rr_rUrVT)indirectctdi|}|j}|jD]\}}|||vsJ|d||vrJy)N=r!)r__repr__r)r_ call_optionsr\rvals rtest_call_options_reprrf sh%9'89L  "D%++-&S ;$; ; au~%%% &r)rrrrrrrrrrr(rrnumpyrz ImportErrorrHrr= pyarrow.librrr pyarrow.utilr pyarrow.testsrr rr r r r rrrrrrKmark pytestmarkrr"r-r4rBrSr[r]rrrrrrrrrrrrr"r;rfr}rrrrrrrrrrrrrrrrrrrrrrrr'r)r.rWrrrrslowrrrrpandasrrrrrrrrrrskipifrrrrrrrr r rrrrrr-r!r'r*r.r0%header_auth_server_middleware_factoryr5r8r:rArGrKrNrQrequires_testing_datarXr\rarerkrqrsrvrrrrrrrrrrrrrrrrrrrrrr large_memoryrrrr rrr!r*r-rErSfixturer_ parametrizerfr!rrrvs$     ::'"[[    B  "; <2 E+ED&+&R'*"-""""2""".>""0*06 +. +#3..." E 0 E F!1F&7'7*.A(.AbJ#+J#Z!20!2"...$ ' $ /(? /1!1161(?189!19 /- /9.E9'7F#3F*+-+5$;5).-+B-&/+B/  $4  '>  '>  1)@ 12"22.4)@44"24. 0."J@Z`F!$8 ('(D6FF $"<.=7$(2 ! 6BGGtODF&F&*##* # #* # #* # #*9 <J4"0 [7, [3 ++BGGtOIK 5K 5+5I)J(K%$&4E ;@8."+"" =# ="" "# """ #  "" 2# 2"* @ "),""8  44@8Ov#L$(N '9,)5X 5"""#"$ !20(FV20!.B,""#.)"))X # 1 1 ""@ "- " ?CCL) ;* ;$#&IR5;p,)U+d< &< &sU B$ F%+V"L"+16((06--06-- s#Y6Z6ZZZ%$Z%