<>i6 SSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSK r SSK r SSK r SSK J r SSK rSSKrSSKrSSKJrJrJr SSKJr SSKJr SSKJr SSKJrJrJrJ r J!r!J"r"J#r#J$r$J%r% \RNR6r(Sr)S r*S r+S r,S r-S r.Sr/"SS\5r0"SS\5r1"SS\5r2"SS\25r3"SS\25r4"SS\25r5"SS\5r6"SS\5r7"SS \5r8"S!S"\5r9"S#S$\5r:"S%S&\5r;"S'S(\5r<"S)S*\5r="S+S,\5r>"S-S.\5r?"S/S0\ 5r@"S1S2\5rA"S3S4\ 5rB"S5S6\5rCS7rD"S8S9\$5rE"S:S;\#5rF"S<S=\"5rG"S>S?\!5rH"S@SA\5rI"SBSC\"5rJ"SDSE\!5rK"SFSG\5rL"SHSI\!5rM"SJSK\"5rN"SLSM\5rO"SNSO\5rP"SPSQ\"5rQ"SRSS\$5rR"STSU\#5rS"SVSW\"5rT"SXSY\$5rU"SZS[\$5rV"S\S]\#5rW"S^S_\"5rX"S`Sa\!5rY"SbSc\5rZSdr[Ser\Sfr]Sgr^Shr_\RNR\RNRSi55raSjrbSkrcSlrd\RNRSm5rfSnrgSorhSpriSqrjSrrk"SsSt\5rlSurmSvrnSwro\RNR\RSx:HSySz9S{5rr\RNRS|5rsS}rtS~ruSrvSrwSrxSrySrzSr{Sr|Sr}\?"SS0S9r~\A"SS0S9r\RNRS5r\RNR\RSx:HSSz9S5rSrSrSr\G"5r\C"5rSrSrSrSrSrSrSr\RNR\RNGRS55r\RNGRS5r\RNGRS5r\RNGRS5rSrSrSrSr\RNRS5r\RNRS5r\RNR\RNRS55rSrSrSrSrSrSrSrSr\RNGRS5rSrSrSrSrSrSr\RNGRS5r"SS\5r\RNRS5rSr\RNGRZ\RNRS55r\RNGRZ\RNRS55r"SS\25rSr\RNRS5rSr"SS\5rSrSrSrSr\GRrS5r\RNGRwSSS/SS9S5rg!\a SrGN6f=f!\a Sr\&\&srr\&\&srr \&\&sr!r"\&\&sr#r$\&r%GN f=f)N)datetime)IpcReadOptions ReadStatstobytes)find_free_port)util)flight) FlightClientFlightServerBaseServerAuthHandlerClientAuthHandlerServerMiddlewareServerMiddlewareFactoryClientMiddlewareClientMiddlewareFactoryFlightCallOptionsc SSKng)Nr)pyarrow.flight)pyarrows X/var/www/html/land-tabula/venv/lib/python3.13/site-packages/pyarrow/tests/test_flight.py test_importrAsc[RRS5(d [S5e[R "[RS5S- $)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"FsI ::>>+ , ,IJ J << #45 6 AArc [5nU(dgX- RS5nUR5sSSS5 $!,(df  g=f![a' [ SX- S[ R "535ef=f)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-Nsy ?D  k   %668& % %  T[M*))2)=)=)?(@ B   s'AA A A AA1Bc[S5[R"[S5[S5S9[R"[S5[S5S9/S.$)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     rcr[R"/SQ5/n[RRUS/S9$)Nir some_intsnames)paarrayTable from_arrays)datas rsimple_ints_tablerBns4 $% D 88  [M  ::rc[R"/SQ[R"5S9n[R"/SQ[R"5S9n[R"[RR /SQU5[RR SS/U5[RR SS/U5/5/n[R R US /S 9$) 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  ;;rc[RR[R"/SQ5[R"/SQ5/SS/S9$)N)rDbarrEquxrJrKrLabr;)r=r?r@r>r!rrmultiple_column_tabler[sB 88  *F!G!#,!7!9'*Cj  22rc@^\rSrSrSrSrSU4SjjrSrSrSr U=r $) 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 b>[TU]"U40UD6 [[[S.UlX lg)N)intsdictsmulti)super__init__rBrSr[table_factoriesoptions)selflocationrfkwargs __class__s rrdConstantFlightServer.__init__s1 ,V,'(+   rc## X R:XaL[R"[R"/5[R R S5/5v gg7f)N/foo)CRITERIAr FlightInfor=schemaFlightDescriptorfor_pathrgcontextcriterias r list_flights!ConstantFlightServer.list_flightssI }} $## " ''008  %sAAczURUR"5n[R"X0RS9$)Nrf)reticketr RecordBatchStreamrf)rgrtrztables rdo_getConstantFlightServer.do_gets0$$V]]35''||DDr)rfreNN) __name__ __module__ __qualname____firstlineno____doc__rnrdrvr}__static_attributes__ __classcell__rjs@rr]r]s% (HEErr]cL^\rSrSrSrSU4SjjrSrSr\S5r Sr U=r $) MetadataFlightServerz4A Flight server that numbers incoming/outgoing data.c 2>[TU]"S0UD6 XlgNr!rcrdrfrgrfrirjs rrdMetadataFlightServer.__init__ "6" rc[R"/SQ5/n[RRUS/S9n[R "UR URU5URS9$)Nr6rYr;ry) r=r>r?r@r GeneratorStreamrpnumber_batchesrf)rgrtrzrAr|s rr}MetadataFlightServer.do_getsb HH( ) $$T#$7%% LL    &LL" "rc (Sn/SQnURRS:XdeUHupxUR[RR [R "Xe/5/S/55(deUce[R"SUR55un XY:XdeUR[R"SU55 US- nM URRS:XdeURRS:Xdeg)Nrr6rJrYstructunpack to_pybyteswritepacknum_record_batches) rgrt descriptorreaderwritercounter expected_databatchbufclient_counters rdo_putMetadataFlightServer.do_puts+ ||((A--- JE<< : :=1234! ? "?$mmD#..2BCON, ,, LLT73 4 qLG!||((A---||..!333rc## [UR55H!up[R"SU5nX#4v M# g7f)Nr) enumerate to_batchesrr)r|idxrrs rr#MetadataFlightServer.number_batchess7#E$4$4$67JC++dC(C* 8sAAryN) rrrrrrdr}r staticmethodrrrrs@rrrs)>"4"rrcB^\rSrSrSrSU4SjjrSrSrSrSr U=r $) EchoFlightServerz4A Flight server that returns the last data uploaded.c B>[TU]"U40UD6 SUlX lgr)rcrd last_messageexpected_schema)rgrhrrirjs rrdEchoFlightServer.__init__s# ,V, .rcB[R"UR5$r)r r{rrgrtrzs rr}EchoFlightServer.do_gets''(9(9::rcUR(aURUR:XdeUR5Ulgr)rrpread_allrrgrtrrrs rrEchoFlightServer.do_puts2   ''6==8 88"OO-rcUHnM grr!)rgrtrrrchunks r do_exchangeEchoFlightServer.do_exchangesE r)rrr) rrrrrrdr}rrrrrs@rrrs>/ ;. rrc*\rSrSrSrSrSrSrSrg)EchoStreamFlightServerz6An echo server that streams individual record batches.c[R"URRURR SS95$)N max_chunksize)r rrrprrs rr}EchoStreamFlightServer.do_gets;%%    $ $    ( (t ( <> >rc/$rr!rgrts r list_actions#EchoStreamFlightServer.list_actions rcURS:Xa/UR5UR5RS5/$[eNwho-am-iutf-8rH peer_identitypeerencodeNotImplementedErrorrgrtactions r do_action EchoStreamFlightServer.do_action9 ;;* $))+W\\^-B-B7-KL L!!rr!N rrrrrr}rrrr!rrrrs@> "rrc*\rSrSrSrSrSrSrSrg)EchoTableStreamFlightServerz,An echo server that streams the whole table.cn[R"URRUR/5$r)r rrrprs rr}"EchoTableStreamFlightServer.do_gets0%%    $ $    ! !rc/$rr!rs rr(EchoTableStreamFlightServer.list_actionsrrcURS:Xa/UR5UR5RS5/$[errrs rr%EchoTableStreamFlightServer.do_actionrrr!Nrr!rrrrs6! "rrc*\rSrSrSrSrSrSrSrg)'EchoRecordBatchReaderStreamFlightServeri zCAn echo server that streams the whole table as a RecordBatchReader.c[R"URRURR 5/5$r)r rrrp to_readerrs rr}.EchoRecordBatchReaderStreamFlightServer.do_get s9%%    $ $    ( ( * +- -rc/$rr!rs rr4EchoRecordBatchReaderStreamFlightServer.list_actionsrrcURS:Xa/UR5UR5RS5/$[errrs rr1EchoRecordBatchReaderStreamFlightServer.do_actionrrr!Nrr!rrrr sM- "rrc$\rSrSrSrSrSrSrg)GetInfoFlightServeriz)A Flight server that tests GetFlightInfo.c[R"[R"S[R"54/5U[R "SS/5[R "S[R RSS5/[R"S5R[R"S55S5/S S S S 5$) NrYr grpc://test localhost2023-04-05T12:34:56.789012345nszendpoint app metadatarJ*Tzinfo app metadata) r ror=rpint32FlightEndpointLocation for_grpc_tcpscalarcast timestamprgrtrs rget_flight_info#GetInfoFlightServer.get_flight_infos  IIRXXZ() * %%cM?;%%__11+tDEII=>CCBLLQUDVW+        rcdURX5n[R"UR5$r)rr SchemaResultrp)rgrtrinfos r get_schemaGetInfoFlightServer.get_schema1s'##G8""4;;//rr!N)rrrrrrr rr!rrrrs3 &0rrc.\rSrSrSr\S5rSrSrg)ListActionsFlightServeri6'A Flight server that tests ListActions.c6SS[R"SS5/$)N)action-1 description)zaction-2zaction-3z more detailr ActionType)clss rexpected_actions(ListActionsFlightServer.expected_actions9s$ (    j- 8  rc#@# UR5ShvN gN7fr)rrs rr$ListActionsFlightServer.list_actionsAs((***s r!N) rrrrr classmethodrrrr!rrr r 6s1  +rr c\rSrSrSrSrSrg)ListActionsErrorFlightServeriEr c## Sv Sv g7f)N)rrrDr!rs rr)ListActionsErrorFlightServer.list_actionsHs s r!N)rrrrrrrr!rrrrEs 1rrc<^\rSrSrSrSU4SjjrSrSrSrU=r $)CheckTicketFlightServeriMzDA Flight server that compares the given ticket to an expected value.c 4>[TU]"U40UD6 Xlgr)rcrdexpected_ticket)rgr!rhrirjs rrd CheckTicketFlightServer.__init__Ps ,V,.rcURUR:Xde[R"/SQ[R"5S9/n[R R US/S9n[R"U5$)Nr6rGrYr;) r!rzr=r>rr?r@r r{)rgrtrzdata1r|s rr}CheckTicketFlightServer.do_getTs`##v}}444-BHHJ?@$$U3%$8''..rc.UR5Ulgr)rr)rgrtrrs rrCheckTicketFlightServer.do_putZs"OO-r)r!rr) rrrrrrdr}rrrrs@rrrMsN// ..rrcf\rSrSrSr\R "S\R"54/5rSrSr g)InvalidStreamFlightServeri^zEA Flight server that tries to return messages with differing schemas.rYc[R"/SQ[R"5S9/n[R"/SQ[R"5S9/nURUR:wde[R R US/S9n[R R US/S9nURUR:Xde[R"URXV/5$)Nr6rG)g$gg@g$@rYr;) r=r>rfloat64rHr?r@rpr r)rgrtrzr$data2table1table2s rr} InvalidStreamFlightServer.do_getcs-BHHJ?@7bjjlKLzzUZZ'''%%eC5%9%%eC5%9}} +++%%dkkF3CDDrr!N rrrrrr=rprr}rr!rrr)r)^s'O YYbhhj)* +FErr)cf\rSrSrSr\R "S\R"54/5rSrSr g)NeverSendsDataFlightServerinz0A Flight server that never actually yields data.rYcURS:XaURR5URR5[RR [ S5/URS9/n[R"URU5$[R"UR[R"URR555$)N yield_datar8rp) rzrp empty_tabler=rr@ranger r itertoolsrepeat)rgrtrzrAs rr}!NeverSendsDataFlightServer.do_getss ==M ) '') '')**E!H:dkk*JD ))$++t< <%% KK))$++*A*A*CDF Frr!Nr1r!rrr3r3ns': YYbhhj)* +F Frr3c4\rSrSrSrSrSr\S5rSr g)SlowFlightServeriz;A Flight server that delays its responses to test timeouts.c[R"[R"S[R"54/5UR 55$)NrY)r rr=rpr slow_streamrs rr}SlowFlightServer.do_gets:%%bii#rxxz1B0C&D&*&6&6&8: :rc2[R"S5 /$N?)timesleeprs rrSlowFlightServer.do_actions 3 rc## [R"/SQ[R"5S9/n[RR US/S9v [ R "S5 [RR US/S9v g7f)Nr6rGrYr;r9)r=r>rr?r@rDrE)r$s rr?SlowFlightServer.slow_streamsd-BHHJ?@hh""5"66 2hh""5"66sBBr!N) rrrrrr}rrr?rr!rrr=r=s#E:77rr=c:\rSrSrSr\S5rSrSrSr Sr g) ErrorFlightServeriz9A Flight server that uses all the Flight-specific errors.c [R[R[R[R[R [ [R[S.$)N)internaltimedoutcancelunauthenticated unauthorizednotimplementedinvalidr0) r FlightInternalErrorFlightTimedOutErrorFlightCancelledErrorFlightUnauthenticatedErrorFlightUnauthorizedErrorrr= ArrowInvalidKeyErrorr!rr error_casesErrorFlightServer.error_casessH222211%@@"::1  rc[R5nURU;aX2R"S5eURS:XaSn[R"SU5e[ e)NrDprotobufthis is an error message)rJrZrHr rWr)rgrtrrZerr_msgs rrErrorFlightServer.do_actionsX'335 ;;+ %kk*51 1 [[J &1G00@ @!!rc## [R"[R"/5[RR S5/5v [R "S5e7f)NrmrD)r ror=rprqrrrSrss rrvErrorFlightServer.list_flightssM IIbM  # # , ,V 4   ((//sA"A$cURS:Xa[R"S5eURS:Xa[R"S5eURS:Xa[R"S5eURS:Xa[R "S5eURS:Xa[R "S5eURS:XaSn[R "SU5eg) NsinternalrDstimedoutscancelsunauthenticateds unauthorizedsprotobufr^)commandr rSrTrUrVrW)rgrtrrrr_s rrErrorFlightServer.do_puts    ,,,U3 3   ; .,,U3 3   9 ,--e4 4   #5 533E: :   ? 2007 7   ; .1G00@ @/rr!N) rrrrrrrZrrvrrr!rrrJrJs'C    "0 ArrJcN^\rSrSrSrS U4SjjrSrSrSrSr Sr S r U=r $) ExchangeFlightServeriz A server for testing DoExchange.c 2>[TU]"S0UD6 Xlgrrrs rrdExchangeFlightServer.__init__rrc URRS:XdeUR[RR :wa[ R"S5eURS:XaURXU5$URS:XaURXU5$URS:XaURXU5$URS:XaURXU5$[ R"SUR35e)Nrz!Must provide a command descriptorechogetput transformzUnknown command: ) rrdescriptor_typer DescriptorTypeCMDr=rXrd exchange_echoexchange_do_getexchange_do_putexchange_transformrs rr ExchangeFlightServer.do_exchanges||((A---  % %)>)>)B)B B//"EF F   7 *%%gv> >   6 )''@ @   6 )''@ @   < /**7FC C//#J$6$6#78: :rc[RR[R"[ SS55/S/S9nUR UR 5 URU5 g)Emulate DoGet with DoExchange.r(rYr;N)r=r?r@r>r8beginrp write_table)rgrtrrrAs rrs$ExchangeFlightServer.exchange_do_getsXxx## HHU1i( )% $  T[[!4 rc4SnUHKnUR(d[R"S5eURRS:wdeUS- nMM URR U:XdeUR [U5RS55 g)Emulate DoPut with DoExchange.rzAll chunks must have data.rJrN) rAr=rXrrrwrite_metadatastrr)rgrtrr num_batchesrs rrt$ExchangeFlightServer.exchange_do_puts E::oo&BCC<<,,1 11 1 K  ||..+===c+.55g>?rctURRS:XdeSnUGHnU(dAUR(a0URURRUR S9 SnUR (a9UR(a(URURUR 5 MUR (aURUR 5 MUR(a:URRS:wdeURUR5 GMS5e g)zRun a simple echo server.rFryTzShould not happenN) rrrArzrprf app_metadatawrite_with_metadatar write_batch)rgrtrrstartedrs rrr"ExchangeFlightServer.exchange_echos||((A---Euzz UZZ.. E!!ejj**5::u7I7IJ##%%e&8&89||00A555""5::.111urcURRS:XdeURHSn[RR UR 5(aM3[R"S[U5-5e UR5nURRS:wdeS/UR-nUH1n[U5HupXh==U R5- ss'M! M3 [RR[R"U5/S/S9n UR!U R5 UR#U 5 g)zSum rows in an uploaded table.rzInvalid field: sumr;N)rrrpr=types is_integerrHrXreprrnum_rowsras_pyr?r@r>rzr{) rgrtrrfieldr|sumscolumnrowvalueresults rru'ExchangeFlightServer.exchange_transforms||((A---]]E88&&uzz22oo&7$u+&EFF#!||((A---sU^^#F'/  U[[]* 0%%rxx~&6ug%F V]]#6"rryr) rrrrrrdrrsrtrrrurrrs@rrgrgs**: ! @2$##rrgc8^\rSrSrSrU4SjrSrSrSrU=r $)HttpBasicServerAuthHandleri7An example implementation of HTTP basic authentication.c.>[TU]5 Xlgrrcrdcredsrgrrjs rrd#HttpBasicServerAuthHandler.__init__  rcUR5n[RRU5nURUR ;a[R "S5eUR URUR:wa[R "S5eUR[UR55 g)N unknown userzwrong password) r&r BasicAuth deserializeusernamerrVpasswordrr)rgoutgoingincomingrauths r authenticate'HttpBasicServerAuthHandler.authenticatesmmo++C0 == *33NC C ::dmm $ 5334DE Ewt}}-.rcU(d[R"S5eXR;a[R"S5eU$)Nztoken not providedr)r rVrrgtokens ris_valid#HttpBasicServerAuthHandler.is_valid$s7334HI I  "33NC C rr rrrrrrdrrrrrs@rrrsA/rrc8^\rSrSrSrU4SjrSrSrSrU=r $)HttpBasicClientAuthHandleri,rcf>[TU]5 [R"X5UlSUlgr)rcrdr r basic_authrrgrrrjs rrd#HttpBasicClientAuthHandler.__init__/s'  **8> rcURR5nURU5 UR5Ulgr)r serializerr&r)rgrrrs rr'HttpBasicClientAuthHandler.authenticate4s-((*t]]_ rcUR$rrrgs r get_token$HttpBasicClientAuthHandler.get_token9 zzr)rr rrrrrrdrrrrrs@rrr,sA % rrc8^\rSrSrSrU4SjrSrSrSrU=r $)TokenServerAuthHandleri=:An example implementation of authentication via handshake.c.>[TU]5 Xlgrrrs rrdTokenServerAuthHandler.__init__@rrcUR5nUR5nX0R;a<URUU:Xa)UR[R"SU-55 g[ R "S5e)Nsecret:zinvalid username/password)r&rrbase64 b64encoder rV)rgrrrrs rr#TokenServerAuthHandler.authenticateDsd==?==? zz !djj&:h&F NN6++J,AB C33+- -rc[R"U5nURS5(d[R"S5eUSS$)Nrz invalid token)r b64decode startswithr rVrs rrTokenServerAuthHandler.is_validMs@  ' ++33OD DQRyrrrrs@rrr=sD-rrc8^\rSrSrSrU4SjrSrSrSrU=r $)TokenClientAuthHandleriTrcH>[TU]5 XlX lSUlg)Nr)rcrdrrrrs rrdTokenClientAuthHandler.__init__Ws    rcURUR5 URUR5 UR5Ulgr)rrrr&rrgrrs rr#TokenClientAuthHandler.authenticate]s/t}}%t}}%]]_ rcUR$rrrs rr TokenClientAuthHandler.get_tokenbrr)rrrrrs@rrrTsD % rrc$\rSrSrSrSrSrSrg)NoopAuthHandlerifzA no-op auth handler.cg)z Do nothing.Nr!rs rrNoopAuthHandler.authenticateisrcg)z> Returning an empty string. Returning None causes Type error. rr!rs rrNoopAuthHandler.is_validls rr!N)rrrrrrrrr!rrrrfsrrcUH8nUR5UR5:XdM'URU5s $ g)zXLookup the value of given key in the given headers. The key lookup is case-insensitive. N)lowerr)headers lookup_keyr0s rcase_insensitive_header_lookuprts4 99;***, ,;;s# #rc*\rSrSrSrSrSrSrSrg)!ClientHeaderAuthMiddlewareFactoryi}z@ClientMiddlewareFactory that creates ClientAuthHeaderMiddleware.c/Ulgrcall_credentialrs rrd*ClientHeaderAuthMiddlewareFactory.__init__s !rc[U5$r)ClientHeaderAuthMiddlewarergrs r start_call,ClientHeaderAuthMiddlewareFactory.start_calls )$//rcXlgrr)rgrs rset_call_credential5ClientHeaderAuthMiddlewareFactory.set_call_credentials.rrN) rrrrrrdrrrr!rrrr}sJ"0/rrc$\rSrSrSrSrSrSrg)ria 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. cXlgrfactoryrgrs rrd#ClientHeaderAuthMiddleware.__init__ rc[US5nU(a0URRSUSRS5/5 gg)N Authorization authorizationrr)rrrr)rgr auth_headers rreceived_headers+ClientHeaderAuthMiddleware.received_headerssB4WoN  LL , , A%%g..0 1 rrN)rrrrrrdrrr!rrrrs 1rrc\rSrSrSrSrSrg)!HeaderAuthServerMiddlewareFactoryiz)Validates incoming username and password.c[US5nUSRS5nSnSnUSS:Xad[R"US5nUR S5RS 5nUSS :Xa USS :Xd[ R "U5eS nOAUSS :Xa"USnUS :Xd[ R "U5eO[ R "U5e[U5$)Nrr rzInvalid credentialsBasicrJr:testr token1234Bearer)rsplitrrdecoder rVHeaderAuthServerMiddleware) rgrrrvaluesr error_messagedecodedpairs rr,HeaderAuthServerMiddlewareFactory.start_calls4   Q%%c*- !9 &&vay1G>>'*005DGv%$q'Z*?77 FFE AY( "1IEK'77 FF(33MB B)%00rr!Nrrrrrrrr!rrrrs 31rrc$\rSrSrSrSrSrSrg)r izBA ServerMiddleware that transports incoming username and password.cXlgrrrs rrd#HeaderAuthServerMiddleware.__init__s rc$SSUR-0$)N authorizationzBearer rrs rsending_headers*HeaderAuthServerMiddleware.sending_headerssTZZ!788rrNrrrrrrdrrr!rrr r sL9rr c\rSrSrSrSrSrg)HeaderAuthFlightServeriz[TU]5 /Ulgrrcrdmethodsrgrjs rrd)RecordingServerMiddlewareFactory.__init__B  rcNURRUR5 grraappendrPr%s rr+RecordingServerMiddlewareFactory.start_callF DKK(rra rrrrrrdrrrrs@rr]r]?*rr]c2^\rSrSrSrU4SjrSrSrU=r$) RecordingClientMiddlewareFactoryiKr^c0>[TU]5 /Ulgrr`rbs rrd)RecordingClientMiddlewareFactory.__init__NrdrcNURRUR5 grrfrs rr+RecordingClientMiddlewareFactory.start_callRrirrjrkrs@rrnrnKrlrrnc$\rSrSrSrSrSrSrg)"MultiHeaderClientMiddlewareFactoryiWrHc0Ulgr last_headersrs rrd+MultiHeaderClientMiddlewareFactory.__init__Zs rc[U5$r)MultiHeaderClientMiddlewarers rr-MultiHeaderClientMiddlewareFactory.start_call_s *400rrvN)rrrrrrdrrr!rrrtrtWsB 1rrtcD\rSrSrSrSS/SS/S/S/S.rS rS rS rS r g )rzicrHrDrUrE)zx-textz x-binary-binz x-MIXED-casesx-other-MIXED-casecXlgrrrs rrd$MultiHeaderClientMiddleware.__init__nrrcUR$r)EXPECTEDrs rr+MultiHeaderClientMiddleware.sending_headersqr-rcNURRRU5 gr)rrwupdate)rgrs rr,MultiHeaderClientMiddleware.received_headersts !!((1rrN) rrrrrrrdrrrr!rrrzrzcs6B%. '* %w H2rrzc\rSrSrSrSrSrg)"MultiHeaderServerMiddlewareFactoryizrHc[U5$r)MultiHeaderServerMiddlewarer%s rr-MultiHeaderServerMiddlewareFactory.start_call}s *733rr!Nrr!rrrrzs B4rrc$\rSrSrSrSrSrSrg)rirHcXlgrrJ)rgrJs rrd$MultiHeaderServerMiddleware.__init__s,rc"[R$r)rzrrs rr+MultiHeaderServerMiddleware.sending_headerss*333rrNrr!rrrrsB-4rrc8^\rSrSrSrU4SjrSrSrSrU=r $)LargeMetadataFlightServeriz Regression test for ARROW-13253.c:>[TU]"U0UD6 SS-Ulg)N )rcrd _metadata)rgargsrirjs rrd"LargeMetadataFlightServer.__init__s! $)&)-rc[R"S[R"54/5n[R"U[R "S//US9UR 4/5$)NrYrJr6)r=rpint64r r record_batchr)rgrtrzrps rr} LargeMetadataFlightServer.do_getsSS"((*-./%%f __qcU6 2DNN C/   rc:URUR5 gr)rrrs rr%LargeMetadataFlightServer.do_exchangesdnn-r)r) rrrrrrdr}rrrrs@rrrs*. ..rrc  SnSnSnSnSnSnSnSnS nS n [[R"S S 55U:Xde[[R"S S 55U:Xde[[R"SS55U:Xde[[R R S 55U:Xde[R"S/[R"S5R[R"S55S5n [U 5U:Xde[R"[R"/5[R R5/SSSS5n [U 5U:Xde[[R"S55U:Xde[[R "S55U:Xde[[R""[R"/555U:Xde[[R""[R"S/555S:Xde[[R$"S55U :XdeU R[R"/5:Xde[R"S[R R5/SSSS5n Sn[U 5U:XdeU Rbe[&R("[*5 [R"S S5 SSS5 [&R("[*5 [R"[-5/5 SSS5 [&R("[*5 [R"S SS[-5/5 SSS5 [&R("[*5 [R"S /SS9 SSS5 [&R("[*5 [R"S /[/S S!S"SS#S$5S9 SSS5 [&R("[*5 [R"S /[-5S%9 SSS5 g!,(df  GNX=f!,(df  GN(=f!,(df  N=f!,(df  N=f!,(df  N=f!,(df  g=f)&Nz1z)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)intrz1z endpoints=[] total_records=1 total_bytes=42 ordered=True app_metadata=b'test app metadata'>rs grpc://testz2023-04-05T01:02:03)expiration_timeirXr8rKrLr)rr Actionrrrq for_commandrr=rrrrorprrrResultrTicketpytestraises TypeErrorobjectr) action_repraction_type_reprbasic_auth_reprdescriptor_repr endpoint_repr info_repr location_repr result_reprschema_result_repr ticket_reprendpointrs r test_reprrsEKBWODO>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 !em^VX%NO " y !eR9NO " y !eR$1aQRTU9VW " y !eRfh? " ! " ! " ! ! ! ! ! ! ! ! !sH/Q) Q*+#Q<0R )%R0R/ Q'* Q9< R  R R,/ R=cSSSSSSSSS S S S S SSSSSSSS/nUHFnU"5up#U"5upEX":XdeX$:XdeXB:XdeX3:XdeX5:XdeXS:XdeX#:waMFe g)Nc^[R"SS5[R"SS54$)NrDrrUr rr!rrtest_eq..suc*FMM%,EFrc^[R"SS5[R"SS54$)NrDrbarrr!rrrrsuc*FMM%,HIrc^[R"SS5[R"SS54$)NrDrUrErr!rrrrs&""5%0""5%02rc^[R"SS5[R"SS54$)Nrruser2r rr!rrrrs&!!&&1!!'624rc^[R"SS5[R"SS54$)Nrrpass2rr!rrrrs&!!&&1!!&'24rc~[RRS5[RRS54$)NrD)r rqrrrr!rrrrs.((44U;((11%8:rc^[R"S/5[R"S/54$Nrrr rr!rrrrs&&&vr2&&vr24rc[R"S[R"S5/5[R"S[R"S5/54$)Nrrgrpc+tls://localhost:1234)r rrr!rrrrsI  ! !)DEF H  ! !)DEF H rc >[R"S/[R"S5R [R "S555[R"S/[R"S5R [R "S5554$)Nrrrz2023-04-05T12:34:56.789ms)r rr=rrrr!rrrrsq  ! !BII&;<AA",,sBST V  ! ! 3499",,t:LM OPrcZ[R"S/SS9[R"S/SS94$)Nrrrmetarr!rrrrs+&&vrD&&vrHJrcN[R"[R"/5[RR 5/5[R"[R"S[R "54/5[RR 5/54$Nints)r ror=rprqrrrr!rrrrsr    " ''002B 8    FBHHJ/01''002B 8 9rc$[R"[R"/5[RR 5/5[R"[R"/5[RR S5/54$Nr)r ror=rprqrrrr!rrrr sg    " ''002B 8    " ''33F;R A Brc z[R"[R"/5[RR 5[R "S/5/5[R"[R"/5[RR 5[R "S/5/54$r)r ror=rprqrrrr!rrrrs    " ''002&&vr23 5    " ''002&&vr23 5 6rc[R"[R"/5[RR 5/SS9[R"[R"/5[RR 5/SS94$)N) total_recordsrJr ror=rprqrrr!rrrrsj    " ''002Bb J    " ''002Ba I Jrc[R"[R"/5[RR 5/SS9[R"[R"/5[RR 5/SS94$)Nr) total_bytesrrr!rrrr#sj    " ''002BB H    " ''002BB H Irc[R"[R"/5[RR 5/SS9[R"[R"/5[RR 5/SS94$)NF)orderedTrr!rrrr*sj    " ''002B G    " ''002B F Grc[R"[R"/5[RR 5/SS9[R"[R"/5[RR 5/SS94$)Nrrrrr!rrrr1sj    " ''002BS J    " ''002BW N OrcZ[R"S5[R"S54$)Nrr)r rr!rrrr8s !<=!<=?rcZ[R"S5[R"S54$r)r rr!rrrr:sv& f(=>rc[R"[R"/55[R"[R"S[R"54/554$r)r rr=rprr!rrrr;sB$$RYYr]3$$RYY0D/E%FGIrcZ[R"S5[R"S54$)Nrr)r rr!rrrr=ss#V]]6%:;rr!)itemsgenlhs1rhs1lhs2rhs2s rtest_eqrsFI 2 4 4 : 4  P  J 9 B 6 J I G O ?> I;gT ElU U ||||||||||||||rc[R"[R"/5[RR 5/5n[R"[R"/5[RR 5/SSS9n[R"[R"/5[RR 5/SSS9nUR S:XdeUR S:XdeUR S:XdeURS:XdeURS:XdeURS:Xdeg)Nr)rr)r ror=rprqrrrr)fi1fi2fi3s rtest_flight_info_defaultsrLs    BIIbM6+B+B+K+K+Mr RC    " ((*Bbb RC    " ((*BdPT VC    "" "    "" "    "" " ??b  ??b  ??b  rcSSS[54/nUH/n[U5n[U[5(deSSS5 M1 g!,(df  MC=f)Ngrpc://localhost:0r)rr isinstance) locationsrhservers r$test_flight_server_location_argumentr^sS  n&'I  h '6f&677 77( ' ' 's A A c[R"[5 [5 [5e!,(df  O=fSSS5 g!,(df  g=fr)rr ValueErrorr r!rr#test_server_exit_reraises_exceptionris4 z "  ,    # " "s A  1 ? A  AcL^S[54mSnU4Sjn[T5n[R"USS9nUR 5 [ R "5nUR SS9 [ R "5U- nUS:deSSS5 g!,(df  g=f)Nrcp>[R"S5 [T5q[R 5 grB)rDrEr rserverhsrr-test_client_wait_for_available..serveus! 3!(+ rTtargetdaemonr8timeoutrC)rr threadingThreadstartrDwait_for_available)rrclientthreadrelapsedrhs @rtest_client_wait_for_availableros^-.H F h 6!!t< ))+!!!!,))+'#~~  s A*B B#cv[5n[R"SUR45n[ UR 55/:XdeUR [R 5n[[ U55S:XdeSSS5 SSS5 g!,(df  N=f!,(df  g=f)zTry a simple list_flights call.rrJN)r]r connectportlistrvrnlen)rr flightss rtest_flight_list_flightsrs  6 NNK5 6&F'')*b000%%&:&C&CD4=!Q&&& 7  6 6 s##B*ABB* B' #B** B8c[5n[R"SUR45n[ UR 55/:XdeUR 5 UR 5 [R"[R5 [ UR 55 SSS5 SSS5 SSS5 g!,(df  N=f!,(df  N(=f!,(df  g=f)Nr) r]r rrrrvcloserrr=rXrr s rtest_flight_client_closers  6 NNK5 6&F'')*b000   ]]2?? + $$& ', 7  , + 7 6 s;#C'A$CC,C4C' C C C$ C'' C5c[5n[5n[R"SUR45nUR [R "S55R5nURU5(deSSS5 SSS5 [RR[RRRS9n[US9n[R"SUR45nUR [R "S55R5nURU5(deUR [R "S55R5R5nURU5(deSSS5 SSS5 [R "[R"SS9 [SS9n[R"SUR45nUR [R "S55R5nSSS5 SSS5 SSS5 g!,(df  GN=f!,(df  GN=f!,(df  N=f!,(df  N=f!,(df  Ne=f!,(df  Nn=f!,(df  g=f) Try a simple do_get call.rr`Nmetadata_versionryz+expected IpcWriteOptions, got matchr)rBr]r rrr}rrrr=ipcIpcWriteOptionsMetadataVersionV4rrrr2)r|rr rArfs rtest_flight_do_get_intsr&s  E  6 NNK5 6&}}V]]734==?{{5!!!! 7 ff$$//22%4G g .& NNK5 6&}}V]]734==?{{5!!!!}}V]]734>>@IIK{{5!!!! 7 / v//J L !" - V[[9:f==w!78AACD;. L L 7 6  7 6 / .;:. - L Ls#I%A II%#JB%I7&J J;##J*4J:J*J; I" I%% I47 J J J J' #J** J8 4J;; K c[5n[5n[R"SUR45nUR [R "S55R5n[US5URS5R5:XdeSSS5 SSS5 g!,(df  N=f!,(df  g=f)rrr`r:rN) rBr]r rrr}r read_pandasrr to_pylist)r|rr rAs rtest_do_get_ints_pandasr*s  E  6 NNK5 6&}}V]]734@@BD%&%,,q/*C*C*EEEE 7  6 6 s##CA%B/C/ B= 9C Cc [5n[5n[R"SUR45nUR [R "S55nURRS:XdeUR5nURU5(deUR[SSSSSS9:XdeSSS5 SSS5 g!,(df  N=f!,(df  g=f) NrrarJrrLrKrrrnum_dictionary_batchesnum_dictionary_deltasnum_replaced_dictionaries) rSr]r rrr}rrrrrr)r|rr rrAs rtest_flight_do_get_dictsr0s  E  6 NNK5 6&v}}X67||((A--- {{5!!!!||y #$"#&'     7  6 6 s##C"BCC" C C"" C0c[R"/SQ[R"5S9/n[RR US/S9n[ SS9n[ R"SUR45nUR[ R"S55R5nURU5(deSSS5 SSS5 g!,(df  N=f!,(df  g=f) z+Make sure Tickets get passed to the server.r6rGrYr;s the-ticket)r!rN) r=r>rr?r@rr rrr}rrr)r$r|rr rAs rtest_flight_do_get_ticketr2s XX) ; >"a'''4>>!$../1444~~a 00888~~a --444~~a **1-1OOOO~~a 00 II; <tBLLu-./ //~~a --1IIII~~a **1- OO ( (d ;< <<#   s H5I  Ic[5n[SUR45nUR[R R S55nUR[R"S[R"54/5:XdeSSS5 SSS5 g!,(df  N=f!,(df  g=f)z+Make sure GetSchema returns correct schema.rrrYN) rr rr r rqrrpr=rr6s rtest_flight_get_schemar9s  & +v{{3 4  !8!8!D!DS!IJ{{bii#rxxz):(;<<<< 5   4 4  s#B1A+B B1 B. *B11 B?ch[5n[SUR45n[R"[ R SS9 [UR55 SSS5 SSS5 SSS5 [5n[SUR45n[UR55[R5:XdeSSS5 SSS5 g!,(df  N=f!,(df  N=f!,(df  N=f!,(df  NJ=f!,(df  g=f)z6Make sure the return type of ListActions is validated.rz3Results of list_actions must be ActionType or tupler N) rr rrrr r2rrr rrs rtest_list_actionsr;s & '6 +v{{3 4 ]]((- $$& '   5 ( ! "f +v{{3 4F'')* # 4 4 67 77 5 # "   5 4 ( ' 5 4 # "s^D$C0C!C0)DD#2DD# C- )C00 C> :D D D D## D1c.\rSrSrSr\S5rSrSrg)ConvenienceServeri zL Server for testing various implementation conveniences (auto-boxing, etc.) c /SQ$)N)rrsbazr!rs rsimple_action_results'ConvenienceServer.simple_action_resultss''rc"^URS:Xa UR$URS:Xa UR/$URS:XaS/$URS:Xa[R"5eURS:Xa U4SjnU"5$g)N simple-actionecho bad-actionrDarrow-exceptionforeverc3p># TR5(dSv TR5(dMgg7fr) is_cancelled)rtsrr(ConvenienceServer.do_action..gens+!..00 L"..00s/66)rHr?bodyr=ArrowMemoryError)rgrtrrs ` rrConvenienceServer.do_actions ;;/ )-- - [[F "KK= [[L (7N [[- -%%' ' [[I % !5L &rr!N) rrrrrpropertyr?rrr!rrr=r= s (( rr=c[5n[SUR45nURS5Vs/sHo"RPM nnX0R :XdeSnURSU45Vs/sHo"RPM nnX4/:XdeSSS5 SSS5 gs snfs snf!,(df  N!=f!,(df  g=f)NrrBsthe-bodyrC)r=r rrrJr?)rr xresultsrJs r!test_do_action_result_conveniencerQ$s   +v{{3 4$*#3#3O#DE#Da66#DE66666#)#3#3VTN#CD#Ca66#CD&    5  F E 5 4  s?CB1B' *B16B, B1C' B11 B? ;C Cc [5n[SUR45n[R"[ R SS9 [URS55 SSS5 [R"[ R SS9 [URS55 SSS5 SSS5 SSS5 g!,(df  Ne=f!,(df  N0=f!,(df  N9=f!,(df  g=f)Nrza bytes-like object is requiredr rDrKrE) r=r rrrr r2rrrs rtest_nicer_server_exceptionsrS2s   +v{{3 4 ]]633!BD !!,/ 0D]]633!35 !!"34 55 5  DD55 5 4  sRC4$C#C"+C# C(C#0C4 C C# C C## C1 -C44 Dc[S5nURS:deUR5 g!UR5 f=f)zMake sure port() works.rrN)rrshutdown)rs r test_get_portrVAs7 !5 6F{{Qs 1Antz'Unix sockets can't be tested on Windows)reasoncr[R"5nUR5 [RR UR 5n[US9 [U5nUR[R"S55n[5nURRUR5(deUR5nURU5(deUR[R"S55n[5nURRUR5(deUR5nURU5(deSSS5 SSS5 SSS5 g!,(df  N=f!,(df  N(=f!,(df  g=f)z3Try a simple do_get call over a Unix domain socket.rr`raN)tempfileNamedTemporaryFilerr r for_grpc_unixnamer]r r}rrBrprrrS)sockrhr rr|rAs rtest_flight_domain_socketr_Js0  $ $ &$ ??00; !8 4X&&]]6==#9:F%'E=='' 55 55??$D;;u%% %%]]6==#:;F&(E=='' 55 55??$D;;u%% %%'5 ' &'&5 4 ' &s=AF( F%DF-F5F( F F F% !F(( F6c[RR[R"[ SS55/S/S9n[ UR S9n[SUR45nUR[RRS5UR 5up4URUS5 UR5 UR[R "S55R#5nUR%U5(deS S S 5 S S S 5 g !,(df  N=f!,(df  g =f) 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>r8rrpr rrr rqrrr{rr}rrrrArr r_rs rtest_flight_large_messagercas 88   q*+,!U  D $++ 6& +v{{3 4MM&"9"9"B"B6"J"&++/  4!12 v}}S12;;=}}T"""" 5 7 6 4 4 7 6s%D6*B)D%D6% D3 /D66 Ecz[RR[R"[ SS55/S/S9n[ 5n[ SUR45nUR[RRS5UR5up4URU5 UR5 UR[R "S55R#5nUR%U5(deSSS5 SSS5 g!,(df  N=f!,(df  g=f) z?Try downloading a flight of RecordBatches in a GeneratorStream.rryrYr;rrrN)r=r?r@r>r8rr rrr rqrrrpr{rr}rrrras r'test_flight_generator_stream_of_batchesrews 88   q)$%!U  D !V +v{{3 4MM&"9"9"B"B6"J"&++/ 4  v}}S12;;=}}T"""" 5 " ! 4 4 " !% D,!B(D D, D) %D,, D:c [RR[R"/SQ[R"[R "5[R "555[R"/SQ5/SS/S9n[5n[SUR45nUR[RRS5UR5up4URU5 UR!5 UR#[R$"S55R'5nUR)U5(deS S S 5 S S S 5 g !,(df  N=f!,(df  g =f) zS Try downloading a flight of RecordBatches with dictionaries in a GeneratorStream. rDrUrErDrD{iYii7rYrZr;rrrNr=r?r@r> dictionaryrrNrr rrr rqrrrpr{rr}rrrras r1test_flight_generator_stream_of_batches_with_dictrn 88   4rxxz27795 7 *+!3Z  D 1 2f +v{{3 4MM&"9"9"B"B6"J"&++/ 4  v}}S12;;=}}T"""" 5 3 2 4 4 3 2%E8-B(E'E8' E5 1E88 Fcz[RR[R"[ SS55/S/S9n[ 5n[ SUR45nUR[RRS5UR5up4URU5 UR5 UR[R "S55R#5nUR%U5(deSSS5 SSS5 g!,(df  N=f!,(df  g=f) z7Try downloading a flight of Table in a GeneratorStream.rryrYr;rrrN)r=r?r@r>r8rr rrr rqrrrpr{rr}rrrras r%test_flight_generator_stream_of_tablerrs 88   q)$%!U  D % && +v{{3 4MM&"9"9"B"B6"J"&++/ 4  v}}S12;;=}}T"""" 5 ' & 4 4 ' &rfc [RR[R"/SQ[R"[R "5[R "555[R"/SQ5/SS/S9n[5n[SUR45nUR[RRS5UR5up4URU5 UR!5 UR#[R$"S55R'5nUR)U5(deS S S 5 S S S 5 g !,(df  N=f!,(df  g =f) zK Try downloading a flight of Table with dictionaries in a GeneratorStream. rhrirYrZr;rrrNrlras r/test_flight_generator_stream_of_table_with_dictrtrorpcz[RR[R"[ SS55/S/S9n[ 5n[ SUR45nUR[RRS5UR5up4URU5 UR5 UR[R "S55R#5nUR%U5(deSSS5 SSS5 g!,(df  N=f!,(df  g=f) zCTry downloading a flight of RecordBatchReader in a GeneratorStream.rryrYr;rrrN)r=r?r@r>r8rr rrr rqrrrpr{rr}rrrras r3test_flight_generator_stream_of_record_batch_readerrvs 88   q)$%!U  D 1 2f +v{{3 4MM&"9"9"B"B6"J"&++/ 4  v}}S12;;=}}T"""" 5 3 2 4 4 3 2rfc [RR[R"/SQ[R"[R "5[R "555[R"/SQ5/SS/S9n[5n[SUR45nUR[RRS5UR5up4URU5 UR!5 UR#[R$"S55R'5nUR)U5(deS S S 5 S S S 5 g !,(df  N=f!,(df  g =f) zW Try downloading a flight of RecordBatchReader with dictionaries in a GeneratorStream. rhrirYrZr;rrrNrlras r=test_flight_generator_stream_of_record_batch_reader_with_dictrxrorpc[5n[SUR45n[R"[ R 5 UR[R"S55R5 SSS5 SSS5 SSS5 g!,(df  N=f!,(df  N(=f!,(df  g=f)z+Try streaming data with mismatched schemas.rrN) r)r rrrr=ArrowExceptionr}r rrrs r$test_flight_invalid_generator_streamr{sx " $ +v{{3 4 ]]2,, - MM&--, - 6 6 8. 5 % $ - - 5 4 % $s:B7%B&4B<B&B7 B# B&& B4 0B77 Cc [5n[SUR45n[R"SS5n[R "SS9n[ R"[R5 [URX#S95 SSS5 SSS5 SSS5 g!,(df  N=f!,(df  N(=f!,(df  g=f)z)Make sure timeouts fire on slow requests.rrrg?rryN) r=r rr rrrrrTrr)rr rrfs rtest_timeout_firesr}s  v +v{{3 4r3'**37]]655 6 !!&!: ;7 5   7 6 5 4  s;CAB63B% B6C% B3 /B66 C C Cc:[5n[SUR45n[R"SS9nUR [R "S5US9R5 SSS5 SSS5 g!,(df  N=f!,(df  g=f)z0Make sure timeouts do not fire on fast requests.rr+rr`ryN)r]r rr rr}rr)rr rfs rtest_timeout_passesrsl  6 +v{{3 4**37 fmmG,g >GGI 5  4 4 s#B AA;*B ; B B  Bc[RR[R"/SQ5/S/S9n[ 5n[ SUR 45n[R"[S/S9S9nUR[R"S5US 9R5nUR[R"S55R5nURS :XdeURS:XdeX@:XdeU[5:XdeS S S 5 S S S 5 g !,(df  N=f!,(df  g =f) z"Make sure ReadOptions can be used.rWrZr;rrJ)included_fields) read_optionsrbryrKN)r=r?r@r>r]r rr rrr}rr num_columnsr[)expectedrr rf response1 response2s rtest_read_optionsrsxx##RXXl%;$MM&-- #&"((0  MM&--"9:CCE $$)))$$)))$$$13333 5  4 4 s%D9B=D(D9( D6 2D99 Etestsp4ssw0rdrc[[S9n[SUR45n[R "SS5n[ R"[RSS9 [URU55 SSS5 SSS5 SSS5 g!,(df  N=f!,(df  N(=f!,(df  g=f)z,Test that auth fails when not authenticated. auth_handlerrrrz.*unauthenticated.*r N) rbasic_auth_handlerr rr rrrrVrrrr rs rtest_http_basic_unauthrs -? @F +v{{3 4z3/ ]]6< :C C C C c[[S9n[SUR45n[R "SS5nUR [SS55 [URU55nURR5S:XdeSSS5 SSS5 g!,(df  N=f!,(df  g=f) -Test an auth mechanism that uses a handshake.rrrrrrrN) rtoken_auth_handlerr rr rrrrrrJr)rr rrs rtest_token_authrCs -? @F +v{{3 4z3/26:FG((01}}'')W444 5 A @ 4 4 A @s#B6A-B%B6% B3 /B66 Cc f[[S9n[SUR45n[R "[ R5 UR[SS55 SSS5 SSS5 SSS5 g!,(df  N=f!,(df  N(=f!,(df  g=f)rrrrrN) rrr rrrr rVrrrs rtest_token_auth_invalidrMsu -? @F +v{{3 4 ]]6<< =    6vw G H> 5 A @ = = 5 4 A @s:B"%B B'B/B" B B B B"" B0c[[S[50S9n[SUR45nUR SS5nUSS:XdeUSS :XdeS S S 5 S S S 5 g !,(df  N=f!,(df  g =f) zATest authenticate_basic_token with bearer token and auth headers.rrrrrpasswordrrrJBearer token1234N)rno_op_auth_handlerrr rauthenticate_basic_token)rr  token_pairs rtest_authenticate_basic_tokenrYs -?13M   +v{{3 444WkJ !} 0000!} 3333 5   5 4  s"A=)A,A=, A: 6A== B cj[[S[50S9n[SUR45n[ R "[R5 URSS5 SSS5 SSS5 SSS5 g!,(df  N=f!,(df  N(=f!,(df  g=f)z7Test authenticate_basic_token with an invalid password.rrrrs badpasswordN) rrrr rrrr rVrrs r.test_authenticate_basic_token_invalid_passwordrds -?13M   +v{{3 4 ]]6<< =  + +G^ D> 5  > = 5 4  s:B$%BB)B1B$ B B B! B$$ B2c [[S[50S9n[SUR45nUR SS5nUSS:XdeUSS :Xde[ R"U/S 9n[UR[ R"S S 5US 95nUSRR5S:XdeSSS5 SSS5 g!,(df  N=f!,(df  g=f)z@Test authenticate_basic_token and doAction after authentication.rrrrrrrrJrr test-actionrrrf token1234N) rrrr rrr rrrrrJrrr rrfrs r(test_authenticate_basic_token_and_actionrns -?13M   +v{{3 444WkJ !} 0000!} 3333**J<@f&&==4g'GHay~~((*l::: 5   5 4  s#C#BCC# C C## C1c [[S[50S9n[5n[ SUR 4U/S9n[ R"S5n[R"SSU-4/S9n[UR[R"S S 5US 95nUS RR5S :XdeURS S:XdeURSS:Xde[UR[R"S S 5US 95nUS RR5S :XdeURS S:XdeURSS:XdeUR!5 SSS5 g!,(df  g=f)zTest authenticate_basic_token with client middleware to intercept authorization header returned by the HTTP header auth enabled server. rrrrs test:passwordrsBasic rrrrrrrJrN)rrrrr rrrr rrrrrJrrr)rclient_auth_middlewarer encoded_credentialsrfrresult2s r4test_authenticate_basic_token_with_client_middlewarer}s -?13M  !B!D &++ &./ %../?@** y+>> ?4  f&&==4g'GHay~~((*l:::%55a8Make sure clients cannot connect when cert verification fails.r2tls_certificateszgrpc+tls://localhost:r`N) r4r]r rrrrr FlightUnavailableErrorr}rrcertsrr s rtest_tls_failsrs  E u^/D E 03qvv;> ?6]]688 9 MM&--0 1 : : <: @ F E: 9 @ ? F Es:"C %B<4B+B<C + B9 5B<< C C  Ccv[5n[5n[USS9n[SUR4USS9nUR [ R"S55R5nURU5(deSSS5 SSS5 g!,(df  N=f!,(df  g=f)z"Try a simple do_get call over TLS.r2rrr1)tls_root_certsr`N) rBr4r]r rr}r rrrr|rrr rAs rtest_tls_do_getrs  E  E u^/D Ek166*$)+$6 8;A}}V]]734==?{{5!!!! 8 F E 8 8 F Es#B*A BB* B' #B** B8c[5n[5n[USS9n[SUR4SS9nWR[R"S55R5nURU5(deUR5 SSS5 g![ a [ R"S5 Nf=f!,(df  g=f) zDTry a simple do_get call over TLS with server verification disabled.r2rrTdisable_server_verificationz4disable_server_verification feature is not availabler`N)rBr4r]r rrrskipr}r rrrrrs r$test_tls_disable_server_verificationrs  E  E u^/D E P!;"7>BDF}}V]]734==?{{5!!!!  F E# P KKN O P F Es.CBAC C=C?CC Cc[5n[USS9n[R"SUR4USSS9n[ R "[R5 UR[R"S55 SSS5 SSS5 SSS5 g!,(df  N=f!,(df  N(=f!,(df  g=f) z5Check that incorrectly overriding the hostname fails.r2rrr1 fakehostname)roverride_hostnamer`N) r4r]r rrrrrr}rrs rtest_tls_override_hostnamers  E u^/D E QVV,&+K&8)7 9r?r@rr rr}r rrgrrr from_batchesr) rAr|batchesrr rrrmetadata server_idxs rtest_flight_do_get_metadatars $% D HH cU 3EG  6 +v{{3 4v}}S12%OE NN5 ! --h.A.A.CDKJ$ $$ 1HC & xx$$W-{{5!!!! 5  4 4 s%D.B.D D. D+ 'D.. D<c/n/n[5n[SUR45nUR[R "S55nSnUR 5upgURU5 [R"SUR55unXX:XdeUS- nMY![a Of=fSSS5 O!,(df  O=fSSS5 O!,(df  O=f[5n[SUR45nUR[R "S55nSnUHIupgURU5 [R"SUR55unXX:XdeUS- nMK SSS5 O!,(df  O=fSSS5 O!,(df  O=fX:Xdeg)z0Verify the iterator interface works as expected.rrrrrJN) rr rr}r r read_chunkrgrrr StopIteration) batches1batches2rr rrrrrs r1test_flight_metadata_record_batch_reader_iteratorrsoHH  6 +v{{3 4v}}S12 "("3"3"5&$mmD(2E2E2GH (((q !   5 4 4    6 +v{{3 4v}}S12%OE OOE " --h.A.A.CDKJ$ $$ 1HC & 5 4 4     slC)CAB)'C) B63C5B66C9 C C C C*7F)A7F F) F F)) F7c2[RR[R"/SQ5/S/S9n[RR [RR RS9n[US9n[SUR45nUR[R"S55nUR5nURU5(deSSS5 SSS5 g!,(df  N=f!,(df  g=f) z2Try a simple do_get call with V4 metadata version.r6rYr;rryrrN)r=r?r@r>r"r#r$r%rr rr}r rrr)r|rfrr rrAs rtest_flight_do_get_metadata_v4r4s HH % &'u ! 6Eff$$//22%4G g .& +v{{3 4v}}S12 {{5!!!! 5 / . 4 4 / .s%?DAC7&D7 D D Dc [R"/SQ5/n[RRUS/S9n[ 5n[ SUR 45nUR[RRS5UR5upEU [URSS95Hpupg[R"SU5nUR!Xx5 UR#5n U ce[R$"SU R'55un Xj:XaMpe S S S 5 S S S 5 S S S 5 g !,(df  N=f!,(df  N(=f!,(df  g =f) z'Try a simple do_put call with metadata.r6rYr;rrrJrrN)r=r>r?r@rr rrr rqrrrprrrrrr&rr) rAr|rr rmetadata_readerrrrrrs rtest_flight_do_put_metadatarAs( $% D HH cU 3E  6 +v{{3 4"(--  # # , ,R 0 LL#'(8(8q(8(IJ !;;tS1**5;%**,&$mmD#..2BC ((( K 5  V 5 4 sCE!>EBD?"D?&E.E!? E E E E!! E/c [RR[R"[R "S[R "5S95/S/S9n[5n[SUR4SS9nUR[RRS5UR5up4U [R "[R"S S 9nUR%U5 S S S 5 WR&R(S:XdeUR+S S 5UR+S 5/nUHnUR%U5 M S S S 5 [R,R/U/5nUR1[R2"S55R55n X:XdeS S S 5 S S S 5 g !,(df  N=f!,(df  N=f!,(df  N9=f!,(df  g =f)z+Try a simple do_put call with a size limit.i)dtyperYr;ri)write_size_limit_bytesrzexceeded soft limitr Nrir)r=rr@r>nponesrrr rrr rqrrrprrFlightWriteSizeExceededErrorrrlimitslicer?rr}rr) large_batchrr rrexcinfosmaller_batchesrractuals rtest_flight_do_put_limitrWs..,, BHHJ/0.U-K  vk6;;/,0 25;"(--  # # , ,R 0   # vBB%:G':$GG 0A!GA"G'4G8 G G G$ G'' G5 1G88 Hc[5n[SUR45nUR[R "S55nUR 5 [R"[RSS9 UR5 SSS5 SSS5 SSS5 g!,(df  N=f!,(df  N(=f!,(df  g=f)z4Test canceling a DoGet operation on the client side.rr`z(?i).*cancel.*r N) r]r rr}r rrNrrrUr)rr rs rtest_cancel_do_getrts  6 +v{{3 4v}}W56  ]]666!13    3 5 33 5 4 s;CAB7<B& B7C& B4 0B77 C C Cc^^^^^[5n[SUR45nUR[R "S55m[ R"5m[ R"5m[ R"5m[ R"5mUUUUU4Sjn[ R"USS9nUR5 TRSS9 TR5 TR5 URSS9 T TR5(deS S S 5 S S S 5 S S S 5 g !,(df  N=f!,(df  N(=f!,(df  g =f) z5Test canceling a DoGet operation from another thread.rr`c>TR5 TR5 TRSS9 TR5 g![Ra0 T TR5 SSS5 g!,(df  g=ff=f)Nr8r)rsetwaitr rU)raised_proper_exceptionread_first_messager result_lockstream_canceledsr block_read/test_cancel_do_get_threaded..block_readsm      " " $   + 2!!#.. 2 +//1![[[ 2s)ABA4*B4 B >BBTrr8rrJN)r=r rr}r rr EventLockr r rrNrjoinis_set) rr rrrrrrrs @@@@@rtest_cancel_do_get_threadedrs  v +v{{3 4v}}W56&__.#//+nn& "+//"3 2 2!!DA *  A  *1133 333 5  4[3 5 4  s;E%C)EE*E2E% E E E" E%% E3c2[5n[SUR45nUR[R "SS55n[ U5RS:XdeASSS5 SSS5 g!,(df  N=f!,(df  g=f)NrrFrr)r=r rrr rrrJ)rr rPs rtest_streaming_do_actionrsp   +v{{3 4""6==C#@AG}!!V+++  5   4 4  s#BAA7&B7 B B Bc@ [R"SS5nU[RRUR55:Xde[R"S5nU[RRUR55:Xde[R "S5nU[R RUR55:Xde[R "SS5nU[R RUR55:Xde[R"[R"S[R"54/55nU[RRUR55:Xde[RRS5nU[RRUR55:Xde[RRSS S 5nU[RRUR55:Xde[R"[R"S[R"54/5U[R"S S /5[R"S [R R#S S5/[R$"S5R'[R("S55S5/SSSS5n[RRUR55nURUR:XdeUR*UR*:XdeUR,UR,:XdeUR.UR.:XdeUR0UR0:XdeUR2UR2:XdeUR4UR4:Xde[R"US [R R#S S5/[R$"S5R'[R("S55S5nU[RRUR55:Xdeg)z(Make sure serializable types round-trip.action1s action1-bodyrDsresult1 username1 password1rYrrZz test.arrowrrrrrrrrJrTrrrN)r rrrrrrrr=rprrqrrrrorrrrrrrrrrrr5) rrzrr schema_resultdescrinfo2rs rtest_roundtrip_typesr s ]]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 [5n[SUR45n[R5R 5HRup#[ R "USS9 [UR[R"US555 SSS5 MT [ R "[RSS9 [UR55 SSS5 [R"/SQ5/n[RR!US/S9n[R[R"[R$[R&[R(S .nUR 5Hupx[ R "USS9 UR+[R,R/U5UR05upU R3U5 U R55 SSS5 [ R "USS9 UR+[R,R/U5UR05upU R55 SSS5 M SSS5 SSS5 g!,(df  GM>=f!,(df  GN=f!,(df  N=f!,(df  GMC=f!,(df  N`=f!,(df  g=f) z:Ensure that Flight errors propagate from server to client.rz.*foo.*r rNr6rYr;)rLrMrNrOrP)rJr rrZrrrrrr rrSrvr=r>r?r@rTrUrVrWrrqrrpr{r) rr argexc_typerAr| exceptionsrd exceptionrrs rtest_roundtrip_errorsrs   +v{{3 4.::<BBDMCxy9V%%fmmC&=>?:9E]]655Y G $$& 'H,-.$$T#$7222211%@@"::  #-"2"2"4 Gy :!'++77@LL""""5)  ;y :!'++77@LL""  ;:#5' 5  :9 G G ;:;:9 5 4  sK=J= 0I4 /J=?JB8J=AJ .J= A J*  J=#K4 J > J= J J= J' #J=* J: 4 J== K K Kc.^ ^ [R"/SQ5/n[RRUS/S9n[ 5n[ SUR 45nUR[RRS5UR5unm S/m U U 4Sjn[R"US9nUR5 URS S 9nU [!U5H-up["R$"S U5n UR'X5 M/ UR)5 UR+5 S S S 5 T S[-U5:XdeS S S 5 S S S 5 g !,(df  N4=f!,(df  N(=f!,(df  g =f) z7Ensure that separate threads can read/write on a DoPut.r6rYr;rrrcj>TR5b!TS==S- ss'TR5bM gg)NrrJ)r&)countrsr_reader_thread:test_do_put_independent_read_write.._reader_threads0!&&(4aA "&&(4rrrJrrN)r=r>r?r@rr rrr rqrrrpr r r rrrrr done_writingrr) rAr|rr rrrrrrrrrs @@r"test_do_put_independent_read_writer sH $% D HH cU 3E  6 +v{{3 4"(--  # # , ,R 0 LL# !!8 """3 '0 !;;tS1**5;1    ! KKMQx3w<'''3 5  V 5 4 s>FA &++ &<>? ((sC)@ABhmm668888 5   8 7> = 5 4  sSF%E1E+,E10E A6E1>F E E1 E. *E11 E? ;F Fc J[5n[5n[SU0S9n[SUR4U/S9n[ R RS5n[R"[5 [UR55 SSS5 [R"[5 URU5 SSS5 [R"[5 URU5 SSS5 [R"[5 UR[ R "S55 SSS5 [R"[5 UR#U[$R&"/55upVUR)5 SSS5 [R"[5 [UR+[ R,"SS555 SSS5 [R"[5 [UR/55 SSS5 [R"[5 UR1U5upVUR)5 SSS5 [ R2R4[ R2R6[ R2R8[ R2R:[ R2R<[ R2R>[ R2R@[ R2RB/nURDU:XdeURDU:XdeSSS5 SSS5 g!,(df  GN=f!,(df  GN=f!,(df  GN=f!,(df  GNt=f!,(df  GN+=f!,(df  GN=f!,(df  GN=f!,(df  GN=f!,(df  N=f!,(df  g=f)z/Test that middleware records methods correctly.rrrrN)#r]rnr r rr rqrrrrrrvrr r}rrr=rprrrrrrQ LIST_FLIGHTSGET_FLIGHT_INFO GET_SCHEMADO_GETDO_PUT DO_ACTIONrR DO_EXCHANGEra)server_middlewareclient_middlewarerr rrrbrs rtest_middleware_mappingr*Us8:8: f.?%@ AV &++ &)*  ,,88= ]]. / $$& '0 ]]. /  " ": .0 ]]. /   j )0 ]]. / MM&--, -0 ]]. / j"))B-@IF LLN0]]. / !!&--S"9: ;0 ]]. / $$& '0 ]]. /**:6IF LLN0    , ,    / /    * *    & &    & &    ) )    , ,    + +  !((H444 ((H444I  B A0 / / / / / / / / /0 / / / / /+   B AsP:P2M3 "P.N"P"N4"P&N)<"P9N;"P90O )"P O%"P$O1+C6P"P3 N =P N P N& !P) N8 3P; O P O P O. )P1 P ;P P P P"c [5n[SUR45n[UR [ R "SS555 e![ Ra-nURceURnUS:XdeSnAOSnAff=fSSS5 O!,(df  O=fSSS5 g!,(df  g=f)Nrr]rr^) rJr rrrr rrW extra_info)rr eeis rtest_extra_infor/s   +v{{3 4 5 !!&-- C"@A B 5-- 5<<+ ++B44 44 5 5 4 4   sEB=B#1AB*#B B#BB# B=# B1 -B== C c[5n[5n[USS/SUSS9n[SUR4USUSSR USSR S9nUR[R"S55R5nURU5(deS S S 5 S S S 5 g !,(df  N=f!,(df  g =f) z!Test mutual TLS (mTLS) with gRPC.r2rTr1)r verify_clientroot_certificatesr)r cert_chain private_keyr`N) r4rBr]r rr/r0r}r rrr)rr|rr rAs r test_mtlsr5s  E  E #N3A67#K0 267 !&& ! -^,Q/44n-a044  6:@}}V]]734==?{{5!!!!  6  2 2 6 6  2 2s$r8rgr rr rqrrr)rrr rrrr|s rtest_doexchange_getr7sxx## q)$%%U$H  6 +v{{3 4,,88@ ++J7 OO%E5   5 V 5 4 s< C+!5CC 'C8C+ C C C( $C++ C9cP[RR[R"[ SS55/S/S9nUR SS9n[ 5n[SUR45n[RRS5nURU5upVU URUR5 UHnURU5 M UR!5 UR#5nUR$be['[)U55R+S 5n UR,U :XdeUR.R0S:XdeS S S 5 S S S 5 S S S 5 g !,(df  N=f!,(df  N(=f!,(df  g =f) r~rryrYr;rrrmNr)r=r?r@r>r8rrgr rr rqrrrzrprrrrArrrrrr) rArrr rrrrr expected_bufs rtest_doexchange_putr;sS 88   q)$%!U  DooCo0G  6 +v{{3 4,,88@ ++J7  LL % ""5)!    !%%'E::% %%s7|,33Gr8rrgr rr rqrrrrrrrArrzrprrrrr) rArrr rrrirrrrs rtest_doexchange_echor>s> 88   q)$%!U  DooCo0G  6 +v{{3 4,,88A ++J7 2Y!fmmG,%%c*))+zz)))))S000  LL %K ""5)}} 333))+zzU***))111q ||66+EEE!&g.!fmmG,**56))+zzU***))S000q ||66+EEE/- 5 V 5 4 sQJ05I7%A'I&B!I&5BI& I& I7J& I4 0I77 J J Jcb[RR[R"[ SS55/S/S9nUR SS9n[R R[R RRS9n[US9n[S UR45n[RRS 5nUR!U5upgU UR#UR$US9 UHanUR'U5 UR$UR$:XdeUR)5n U R*U:XdeU R,cMae S S S 5 S S S 5 S S S 5 g !,(df  N=f!,(df  N(=f!,(df  g =f) z;Try a DoExchange echo server using the V4 metadata version.rryrYr;r9rrryrrkN)r=r?r@r>r8rr"r#r$r%rgr rr rqrrrzrprrrAr) rArrfrr rrrrrs rtest_doexchange_echo_v4r@s^ 88   q)$%!U  DooCo0Gff$$//22%4G g .& +v{{3 4,,88A ++J7  LLgL 6 ""5)}} 333))+zzU***))111 ! 5 / .V 5 4 / .sCF .5F#A:E>!E>%F-F > F F F F  F.c l[RR[R"[ SS55[R"[ SS55[R"[ SS55//SQS9n[RR[R"[ S S S 55/S /S9n[ 5n[ S UR45n[RRS 5nURU5upVU URUR5 URU5 UR5 UR!5nSSS5 UW:XdeSSS5 SSS5 g!,(df  N(=f!,(df  N(=f!,(df  g=f)z!Transform a table with a service.rrrJirKi)rYrZcr;rLi rrrnN)r=r?r@r>r8rgr rr rqrrrzrpr{rr)rArrr rrrr|s rtest_doexchange_transformrC sO 88   q$  q$  q$ !  D xx## q,*+%W$H  6 +v{{3 4,,88F ++J7  LL %   t $    !OO%E  5    5 V 5 4 s=F%5FA F!F2F% F F F" F%% F3c [S[50S9n[5n[SUR4U/S9n[ UR [R"SS555nURR5RS5n[R"U5n[RR!5HsupgUR#5n[%U[&5(aURS5nUR)U5U:XdeUR*R)U5U:XaMse SSS5 SSS5 g!,(df  N=f!,(df  g=f)rHrrrrrasciiN)rGrrtr rrrr rrJrr ast literal_evalrzrrrrbytesrrw)rrr r raw_headersrJheaderr s rtest_middleware_multi_headerrK! s+ 24-  46 fkk*#9&)/F,,V]]3-DEFH"--224;;GDK --k:N"="F"F"L"L"Nfe,,#]]73F%))&1V;;;++//76AAA #O&   &&   s)"EC&E$E' E E' E$ E'' E5c$\rSrSrSrSrSrSrg)CancelFlightServeriO zA server for testing StopToken.c[R"/5n[RR/US9n[R "U[ R"U55$)Nr6)r=rprr@r rr9r:)rgrtrzrpr$s rr}CancelFlightServer.do_getR sF2 ^^ ' '6 ' :%%fi.>.>r.BCCrc6[R"/5n[RR/US9nUR U5 UR 5(d?UR U5 [R"S5 UR 5(dM>gg)Nr6rC) r=rprr@rzrHrrDrE)rgrtrrrrpr$s rrCancelFlightServer.do_exchangeW sm2 ^^ ' '6 ' : V&&((   r " JJsO&&((rr!N)rrrrrr}rrr!rrrQrQO s)D rrQc^^[R"5R[R"5R:wa[R "S5 Sm[ [R4mUU4Sjn[5n[SUR45nUR[R"S55nU"UR5 [R R#S5nUR%U5upSU"UR5 UR'5 SSS5 SSS5 g![ [R(4a N-f=f!,(df  N7=f!,(df  g=f)Nz'test only works from main Python threadcx[R"S5 [R"[R5 grB)rDrEsignal raise_signalSIGINTr!rrsignal_from_thread*test_interrupt..signal_from_threade s 3FMM*rc>[R"TS9n[R"T5nUR 5 U"5 SSS5 UR 5 WRn[U[R[ 45(d2[UR[R[ 45(degg!,(df  N=f!WR 5 f=f![ a [R"S5 Nf=f)Nrz2KeyboardInterrupt didn't interrupt Flight read_all) r r rrr rKeyboardInterruptfailrrr=ArrowCancelled __context__)rtexc_infor- exc_typesr[s rrtest_interrupt..testk s N $$,>?]]9-GGIJ. NN!b//1BCDD q}}r'8'8:K&L M M NN ME.-  N KKL M Ns:*CCCC( C CC%%C(( D  D rrrk)r current_threadident main_threadrrr^r=r`rQr rr}r rrrqrrrrU)rrr rrrrdr[s @@rtest_interruptri` s!''9+@+@+B+H+HH =>+#B$5$56IN$   +v{{3 4v}}S12 V__,,88A ++J7 V__  LLN 5  "6#>#>?    5 4  sCE*A 9J J J J"% J3 /K6 K K K K K& "K)) K7c"\rSrSrSr/rSrSrg)ActionNoneFlightServeri z@A server that implements a side effect to a non iterable action.cURS:Xa0[R"UR5R S5/$URS:XaURR S5 g[ e)N get_valuerrgT)rHjsondumpsVALUESrrgrrs rr ActionNoneFlightServer.do_action sV ;;+ %JJt{{+227;< < [[H $ KK  t $!!rr!N)rrrrrryrrr!rrrtrt sJ F"rrtc[5n[SUR45nUR[R "SS55 UR[R "SS55n[ R"[U5RR55S/:XdeSSS5 SSS5 g!,(df  N=f!,(df  g=f)z}Ensure that actions are executed even when we don't consume iterator. See https://issues.apache.org/jira/browse/ARROW-14255 rrgrrvTN) rtr rrr rrwloadsrrJr)rr rs rtest_none_action_side_effectr~ s !V +v{{3 4x56   V]];< =zz$q',,1134>>> 5 " ! 4 4 " !s#CB C/C C C Cc V^ ^ SnSn[R"XS9m [RRS5n[R "S[R "54/5n"U 4SjS[R5nU"5n[SUR45nURX#5unm U 4S jn[R"US S 9n U R5 [R"[R5n UR![R""S //US 95 M(!,(df  O=fW R$R&U:Xde[R"[R5n UR)5 SSS5 O!,(df  O=fU R$R&U:XdeU R+5 UR-U5unm U 4Sjn[R"US S 9n U R5 [R"[R5n UR/S5 M!,(df  O=fU R$R&U:Xde[R"[R5n UR)5 SSS5 O!,(df  O=fU R$R&U:XdeU R+5 SSS5 O!,(df  O=fSSS5 g!,(df  g=f)zw Ensure that exceptions during writing preserve error context. See https://issues.apache.org/jira/browse/ARROW-16592. rDr)r,rrc.>\rSrSrU4SjrU4SjrSrg)0test_write_error_propagation..FailServeri c>Terr!rgrtrrrexcs rr7test_write_error_propagation..FailServer.do_put Irc>Terr!rs rr.FailServer.do_exchange rrr!N)rrrrrrr)rsr FailServerr s   rrrc`>TR5 M![Ra gf=fr)r&r FlightErrorrsr_reader-test_write_error_propagation.._reader s1 KKM%%   --TrrJr6Nc`>TR5 M![Ra gf=fr)rr rrsrrr s4 %%'%%  rr)r rUrqrr=rprr r rrr r r rrrrrr,rrrr) expected_message expected_inforrprrr rrrrcrrs @@rtest_write_error_propagationr sCM  % % 4C((44S9J YY"((*-. /FV,,  +v{{3 4z: !!>  ]]666 78""2??QC5#HI8 7~~((M999 ]]666 78 LLN8 7 7~~((M999  ++J7 !!>  ]]666 78%%d+8 7~~((M999 ]]666 78 LLN8 7 7~~((M999 ] 5 4 4 s| L!A$L*D// D= 9ALF L F( $BL:I I AL J:1 L: K 3L7 L L L L(c0[R"S5 g)zy 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  '(rc\rSrSrSrSrSrg)TracingFlightServeri9 z/A server that echoes back trace context values.cfURS5RnSUR55$)Ntracingc3R# UHupUSU3RS5v M g7f)z: rN)r).0r0rs r 0TracingFlightServer.do_action..A s2;$9LS55'"**733$9s%')r trace_contextr)rgrtrrs rrTracingFlightServer.do_action< s3..y9GG ;$1$7$7$9; ;rr!Nr r!rrrr9 s 9;rrc0[S[R"50S9n[SUR45n[R "SS/S9nUR SUS9HnM SSS5 SSS5 g!,(df  N=f!,(df  g=f) Nrrr)s traceparents700-000ff00f00f0ff000f0f00ff0f00fff0-000f0000f0f00000-00)s tracestaterr)rrry)rr TracingServerMiddlewareFactoryr rrr)rr rfrs r test_tracingrE s  v<<>)   +v{{3 4** 5 4   %%j'%BE C 5   5 4  s"B.A6%B6 B B Bc[SSS9nSn[R"[US9 UR [ R RS5SS9 SSS5 g!,(df  g=f)Nzgrpc+tls://localhost:9643TrzRArgument 'schema' has incorrect type \(expected pyarrow.lib.Schema, got NoneType\)r rDr6)r rrrrr rqr)r msgs r.test_do_put_does_not_crash_when_schema_is_nonerW sZ 56:.HeadersTrailersFlightServerid cURSS5 URSS5 URSS5 URSS5 [R"[R "/5U/5$) Nx-header header-value x-header-binz headervalue x-trailer trailer-value x-trailer-binz trailervalue) add_header add_trailerr ror=rprs rrJtest_headers_trailers..HeadersTrailersFlightServer.get_flight_infoe sh   z> :   ~/@ A    _ =   1C D$$ "  rr!N)rrrrrrr!rrHeadersTrailersFlightServerrd s rrc(>\rSrSrSrU4SjrSrg)?test_headers_trailers..HeadersTrailersMiddlewareFactoryip c/Ulgrrrs rrdHtest_headers_trailers..HeadersTrailersMiddlewareFactory.__init__q s DLrc>T"U5$rr!)rgrHeadersTrailersMiddlewares rrJtest_headers_trailers..HeadersTrailersMiddlewareFactory.start_callt s,T2 2rrN)rrrrrdrr)rsr HeadersTrailersMiddlewareFactoryrp s  3 3rrc \rSrSrSrSrSrg)8test_headers_trailers..HeadersTrailersMiddlewareiw cXlgrrrs rrdAtest_headers_trailers..HeadersTrailersMiddleware.__init__x s"LrcUR5H4up#UH)nURRRX$45 M+ M6 gr)rrrrg)rgrr0r rs rrItest_headers_trailers..HeadersTrailersMiddleware.received_headers{ s8&}} #ELL((// =$ /rrN)rrrrrdrrr!rrrrw s  # >rrrrr)rr)rs headervalue)rr)rs trailervalueN) r rrr rrr rqrrr)rrrrr rs @rtest_headers_trailersra s &6 3+B3>$4>/0G $ && +v{{3 Jfv66??CD+w>>>3wFFF-@@@5HHH K ' & J J ' &s%C2A7C!C2! C/ +C22 Dc N^ [SSSSSS9[SSSSSS9S.m "U 4SjS [5nU"5n[S UR45n[ 5nSGHn[ R RU5nURU[ R"[RRS S 9S 9S9upgU URUR[RRUS:HS 9S9 URU5 UR!5 UR#5nSSS5 WR%U5(deUR&T U:XaGMe SSS5 SSS5 g!,(df  NQ=f!,(df  N(=f!,(df  g=f)NrrLrKrJrr,) dict_deltasdict_replacementc">\rSrSrU4SjrSrg)Dtest_flight_dictionary_deltas_do_exchange..DeltaFlightServeri c>[5nUR5nURU5(deURTURR 5:XdeURS:XaH[ RRSS9nURURUS9 URU5 URS:Xa-URUR5 URU5 gg)Ns dict_deltasTemit_dictionary_deltasrysdict_replacement) rSrrrrdr r=r"r#rzrpr{) rgrtrrrexpected_tablereceived_tablerfexpected_statss rrPtest_flight_dictionary_deltas_do_exchange..DeltaFlightServer.do_exchange s/1N#__.N!((88 88<<>*2D2D2K2K2M#NN NN!!^3&&000M ^22G D"">2!!%88 ^223"">29rr!N)rrrrrr)rsrDeltaFlightServerr s  3 3rrrTr) write_optionsryr)rr]r rrSr rqrrrr=r"r#rzrpr{rrrr) rrr rrdrrrrrs @r)test_flight_dictionary_deltas_do_exchanger s  #$"#&'  & #$"#&'  N" 30 3   +v{{3 4+-:G00<2##%!'!2 "((88 88<<>'#:: ::%; 5   5 4  sCFA1F?A*E4 )1FF#F4 F >F F F F$cURS:XaSSSSS.$URS:XaaSS/[RRSS9[RR S [RR R S 9S.$0$) NdefaultrL)rrrrallr)sabcsdefzstd) compressionF) use_threadsensure_alignment)paramr=r"r#r AlignmentDataTypeSpecific)requests rcall_options_argsr s}} !!   % ()VV333GFF11!!#!1!1!B!B2    rrrrT)indirectc[S0UD6nUR5nUR5Hup4Uc X2;deMUSU3U;aMe g)N=r!)r__repr__r)r call_optionsrr vals rtest_call_options_reprr s_%9'89L  "D%++- ;; ; au~%%% .r)rFrr9rrrXrrZr rDr(rwrnumpyr ImportErrorrrr= pyarrow.librrr pyarrow.utilr pyarrow.testsrr rr r r r rrrrrrmark pytestmarkrr"r-r4rBrSr[r]rrrrrrr rrr)r3r=rJrgrrrrrrrrrr rr"r$r/r9r?rCrGrMrUrWr]rnrtrzrrrrrrrrslowrrrr&pandasr*r0r2r7r9r;r=rQrSrVskipifr]r_rcrernrrrtrvrxr{r}rrrrrrrrr%header_auth_server_middleware_factoryrrrrrrrrrequires_testing_datarrrrrrrrrrrrr rrrrr*r/r5r7r;r>r@rCrKrOrQrirk large_memoryrorrrtr~rrrrrrrfixturer parametrizerr!rrrs$     ::'"[[    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#Y'Y6'Y32Y36ZZ