U 饡c~@s^ddlZddlZddlZddlZddlmZddlZddlZddl m Z m Z ddl m Z ejjdddejjd d dejjd d ejjd ejejdd dejd<ddZGdddZddZddZejddZejddddZejdd d!d"Zejddd#d$Zejddd%d&Zejddd'd(Zejddd)d*Z dS)+N)TemporaryDirectory)groupsdefaultsfind_free_portcii) max_examplesdev2debug )r verbosityZHYPOTHESIS_PROFILEz /dev/nullZAWS_CONFIG_FILEcCshd dd}tD]T}|d|t|}|jd|d|d|d|jd|dd d |dqdS) NcSsPtj|}|s|S|}|dkr,dS|dkr8dStd||dS)N>ontrueyesy1T>0nooffnfalseFz {}={} is not parsable as boolean)osenvirongetupperlower ValueErrorformat)namedefaultvaluer#:/tmp/pip-unpacked-wheel-qoi2rb4q/pyarrow/tests/conftest.pybool_env2sz"pytest_addoption..bool_envzPYARROW_TEST_{} --enable-{} store_truezEnable the {} test group)actionr!help --disable-{}FzDisable the {} test group)N)rrrZ addoption)parserr%groupr!r#r#r$pytest_addoption0s   r-c@s$eZdZddZddZddZdS) PyArrowConfigcCs i|_dSN) is_enabled)selfr#r#r$__init__JszPyArrowConfig.__init__cCs|j}|tkr||dSr/)r rrequires)r1markr,r#r#r$ apply_markMszPyArrowConfig.apply_markcCs|j|std|dS)Nz{} NOT enabled)r0pytestskipr)r1r,r#r#r$r3Rs zPyArrowConfig.requiresN)__name__ __module__ __qualname__r2r5r3r#r#r#r$r.Isr.cCsXt|_tD]F}|d|d|}d|}||oD|| }||jj|<q dS)Nmarkersr&r*)r.pyarrowrZaddinivalue_linerZ getoptionr0)configr4Z enable_flagZ disable_flagr0r#r#r$pytest_configureWs   r>cCs |D]}|jj|qdSr/)Z iter_markersr=r<r5)itemr4r#r#r$pytest_runtest_setuphs r@cCs t|jSr/)pathlibPathZstrpath)Ztmpdirr#r#r$tempdirnsrCsession)ZscopecCsttjdS)Ndata)rArB__file__parentr#r#r#r$ base_datadirtsrHT)ZautousecCs|dddS)zStop the AWS SDK from trying to contact the EC2 metadata server. Otherwise, this causes a 5 second delay in tests that exercise the S3 filesystem. ZAWS_EC2_METADATA_DISABLEDrN)Zsetenv)Z monkeypatchr#r#r$disable_aws_metadataysrIcCs8tjdd}ttjdd}tjdd}|||fS)NZARROW_HDFS_TEST_HOSTr!ZARROW_HDFS_TEST_PORTrZARROW_HDFS_TEST_USERZhdfs)rrrint)hostportuserr#r#r$hdfs_connectionsrNcCs dt}}d\}}||||fS)N localhost)ZarrowZ apachearrowr)rKrL access_key secret_keyr#r#r$ s3_connections rRc cs|\}}}}d||}tj}|||dtt}ddddd||g}d} zDztj||d} Wnt k rt d YnX|| |d VW5| dk r| XW5QRXdS) Nz{}:{})ZMINIO_ACCESS_KEYZMINIO_SECRET_KEYZminioz--compatserverz--quietz --addressenvz!`minio` command cannot be located) connectionprocessrC) rrrcopyupdaterkill subprocessPopenOSErrorr6r7) rRrKrLrPrQaddressrUrCargsprocr#r#r$ s3_servers.     rac cst}tj}tjdddt|g}d}z`ztj ||d}Wn8t k rt}zt d|d|W5d}~XYnXd|f|dVW5|dk r|XdS) Nz-mZ testbenchz--portrTzCommand z failed to execute: rO)rVrW) rrrrXsys executablestrrZr[r\r]r6r7)rLrUr_r`er#r#r$ gcs_servers ( rf)!rrAr[rbtempfilerr6Z hypothesishZconftestrrZ pyarrow.utilrsettingsZregister_profileZ VerbosityverboseZ load_profilerrr-r.r>r@ZfixturerCrHrIrNrRrarfr#r#r#r$sB