U ovgJT@stUddlmZddlZddlmZddlZddladdlZddlm Z ddl Z ddl Z ddl m Z ddlmZdZtdZdadadadadadadadZdadadadadaeZ eZ!da"d e#d <d d Z$d dZ%ddZ&ddZ'ddZ(ddZ)ddZ*ddZ+ddZ,ddZ-dd Z.d!d"Z/d#d$Z0d%d&Z1d'd(Z2gZ3gZ4d)d*Z5d+d,Z6e5d-d.Z7e5d/d0Z8e6d1d2Z9e6d3d4Z:e5d5d6Z;e6d7d8Ze6d=d>Z?e6d?d@Z@e6dAdBZAdCdDZBdEdFZCdGdHZDdIdJZEdKdLZFdMdNZGdOdPZHdQdRZIdSdTZJdUdVZKdWdXZLdYdZZMded[d\ZNd]d^ZOd_d`ZPGdadbdbejQZRdaSdcddZTdS)f) annotationsN) Namespace)Path)Any)asyncioFzsqlalchemy.testing.plugin_baseroptionscCs|ddttdd|ddttdd|ddtd d d |d dtd d|ddtddd |ddtddd |ddddd|dddd|ddtddd|d dtd!d"d|d#dtd$d%d|d&dtd'd(d|d)td*d+d,d-|d.td/d0d1|d2dd3d4d|d5td6d7d1|d8dttd9d|d:dttd;d<|d=dttd>d<|d?dd@dAdBdC|dDddEdAdFdC|dGtdHdId1|dJtdgdKdLdM|dNtdOdP|dQddRddS)SNz --log-infocallbackz,turn on info logging for (multiple OK))actiontyperhelpz --log-debugz-turn on debug logging for (multiple OK)z--dbappenddbzBUse prefab database uri. Multiple OK, first one is run by default.)r r destr z--dbszList available prefab dbs)r Zzeroarg_callbackr z--dburidburiz8Database uri. Multiple OK, first one is run by default.z --dbdriverdbdriverz|Additional database drivers to include in tests. These are linked to the existing database URLs by the provisioning system.z --dropfirst store_true dropfirstz,Drop all tables in the target database first)r rr z--disable-asyncioz8disable test / fixtures / provisoning running in asyncio)r r z--backend-onlybackendz}Run only tests marked with __backend__ or __sparse_backend__; this is now equivalent to the pytest -m backend mark expressionz --nomemoryZmemory_intensivezpDon't run memory profiling tests; this is now equivalent to the pytest -m 'not memory_intensive' mark expressionz--notimingintensiveZtiming_intensivezpDon't run timing intensive tests; this is now equivalent to the pytest -m 'not timing_intensive' mark expressionz--nomypyZmypyz_Don't run mypy typing tests; this is now equivalent to the pytest -m 'not mypy' mark expressionz--profile-sortZ cumulative profilesortz*Type of sort for profiling standard output)r defaultrr z--profile-dump profiledumpz2Filename where a single profile run will be dumped)r rr z--low-connectionslow_connectionsz>Use a low number of distinct connections - i.e. for Oracle TNSz--write-idents write_identszCwrite out generated follower idents to , when -n is usedz--requirementsz3requirements class for testing, overrides setup.cfgz --include-tagzAInclude tests with tag ; legacy, use pytest -m 'tag' instead)r rr r z --exclude-tagzEExclude tests with tag ; legacy, use pytest -m 'not tag' insteadz--write-profilesZwrite_profilesFz$Write/update failing profiling data.)r rrr z--force-write-profilesZforce_write_profilesz,Unconditionally write/update profiling data.z--dump-pyannotateZdump_pyannotatez/Run pyannotate and dump json info to given filez--mypy-extra-test-pathZmypy_extra_test_pathszlAdditional test directories to add to the mypy tests. This is used only when running mypy tests. Multiple OK)r r rrr z--postgresql-templatedbz^name of template database to use for PostgreSQL CREATE DATABASE (defaults to current database))r r z--oracledb-thick-modez:enables the 'thick mode' when testing with oracle+oracledb)str_log _list_dbs_set_tag_include_set_tag_exclude_requirements_opt _include_tag _exclude_tag) make_optionr"I/tmp/pip-unpacked-wheel-uqd9otp_/sqlalchemy/testing/plugin/plugin_base.py setup_options;s@   r$cCsddlm}||_dS)zConfigure required state for a follower. This invokes in the parent process and typically includes database creation. r provisionN)sqlalchemy.testingr&FOLLOWER_IDENT)Zfollower_identr&r"r"r#configure_followers r)cCsdS)zStore important configuration we will need to send to a follower. This invokes in the parent process after normal config is set up. Hook is currently not used. Nr"Zdict_r"r"r#!memoize_important_follower_configsr+cCsdS)zRestore important configuration needed by a follower. This invokes in the follower process. Hook is currently not used. Nr"r*r"r"r#!restore_important_follower_configsr,cCs*tatt|dt|dgdS)Nz setup.cfgztest.cfg) configparser ConfigParser file_configreadr) root_pathr"r"r# read_configsr2cCs|atD]}|ttqdS)z7things to set up early, before coverage might be setup.N)r pre_configurer/)optfnr"r"r# pre_beginsr6cCs |t_dSN)rZ has_coverage)valuer"r"r#set_coverage_flag sr9cCsptD]}|ttqddlmaddlmamamaddlm a m a m a ddlm a m a ddlmat dS)z9things to set up later, once we know coverage is running.rtesting)fixturesengines exclusions) assertionswarnings profiling)configr&)utilN)post_configurerr/ sqlalchemyr;r'r<r=r>r?r@rArBr&rCZ setup_filtersr5r"r"r# post_begin$s   rGcCsRtsddlat|dr2t|tjn|drNt|tjdS)Nrz-infoz-debug)logging basicConfigendswith getLoggersetLevelINFODEBUGopt_strr8parserr"r"r#r6s  rcGsVtdkrtttdttdD]}td|td|fq*t ddS)Nz0Available --db options (use --dburi to override)r z%20s %sr) r/r2rcwdprintsortedrgetsysexit)argsmacror"r"r#rCs  rcCs t|dSr7)_setup_requirementsrOr"r"r#rNsrcsfdd}|S)Ncst||dSr7)rrOtagr"r#_do_include_tagSsz)_set_tag_include.._do_include_tagr")r\r]r"r[r#rRs rcsfdd}|S)Ncst||dSr7)r rOr[r"r#_do_exclude_tagZsz)_set_tag_exclude.._do_exclude_tagr")r\r^r"r[r#rYs rcCst|dddSN-_) exclude_tagsaddreplacerOr"r"r#r `sr cCst|dddSr_) include_tagsrcrdrOr"r"r#rdsrcCst||Sr7)r3r rFr"r"r#prels rfcCst||Sr7)rDr rFr"r"r#postqs rgcCs|adSr7rr4r/r"r"r#_setup_optionsvsrjcCs,ddlm}|ddd|ddddS)Nr)registryzsqlite.pysqlite_numericz#sqlalchemy.dialects.sqlite.pysqliteZ_SQLiteDialect_pysqlite_numericzsqlite.pysqlite_dollarZ_SQLiteDialect_pysqlite_dollar)Zsqlalchemy.dialectsrkregister)r4r/rkr"r"r# _register_sqlite_numeric_dialect|s rmc Cs^tjdddkrZddlm}z|ddWn,tk rX}ztd|W5d}~XYnXdS) NZREQUIRE_SQLALCHEMY_CEXT01r)has_compiled_extT)Zraise_zEREQUIRE_SQLALCHEMY_CEXT is set but can't import the cython extensions)osenvironrUZsqlalchemy.utilrp ImportErrorAssertionError)r4r/rperrr"r"r# __ensure_cexts rvcCsddlm}t|_dSNrrB)r'rB_fixture_fn_classZ_fixture_functions)rr/rBr"r"r# _init_symbolss rzcCs|jr dt_dS)NF)Zdisable_asynciorZENABLE_ASYNCIOrir"r"r#_set_disable_asynciosr{c spddlm}ddlm}ddlm}ddlm|jrBt|j}ng}|j pNg}|j r|j D]D}t d|D]2}|| dkrtd|ql||d|qlq\|s||dd d|_|jr|jrfd d |DD]<} t|jd &} | |jd | jdddW5QRXqt|||} | D]8} td| || |||j} |js2| | |q2dS)Nrr:rxr%)urlz[,\s]+r z:Unknown URI specifier '%s'. Specify --dbs for known uris.rcsg|]}|qSr")Zmake_url).0db_urlZsa_urlr"r# sz_engine_uri..a F)Z hide_password zAdding database URL: %s)rEr;r'rBr&Zsqlalchemy.enginer|rlistrr resplitr RuntimeErrorr rU_currentrr(openwriteZrender_as_stringZgenerate_db_urlsloginfoZ setup_configZset_as_current) rr/r;rBr&Zdb_urlsZ extra_driversZdb_tokenr r~file_Z expanded_urlscfgr"rr# _engine_urisL        rcCs|dd}t|dS)N sqla_testingrequirement_cls)rUrZ)rr/rr"r"r# _requirementss rcCsrddlm}ddlm}|d\}}t|}|dddD]}t||}q@t||}||_|_t |_ dS)Nrrxr::.) r'rBrEr;r __import__getattr requirementsrequiresbootstrapped_as_sqlalchemy)argumentrBr;modnameZclsnamemod componentZreq_clsr"r"r#rZs    rZcCs@ddlm}|jrs rcCs:tjtttj|ztjs*t W5tXdSr7) r=r stop_test_class_outside_fixturesr&rBr _restore_enginerrr?global_cleanup_assertionsrr"r"r#rGs   rcCstjrtjtdSr7)rBrresetr;r"r"r"r#rQsrcCstjttdSr7)r=rZ final_cleanupr?rrr"r"r"r#final_process_cleanupVs rcCs<t|ddr8t|j}d|d<tj|d}tj|tdS)N__engine_options__classZscoperh) rdictrr=Ztesting_enginerBrZ push_enginer;)roptsZengr"r"r#r\s    rcCs*t|d|j}d|||f}t|dS)Nrz%s.%s.%s)rrrAZ_start_current_test)testZtest_module_nameZ test_classZ test_namerZid_r"r"r# before_testdsrcCsttjdSr7)r< after_testr=rrr"r"r#rosrcCstj|dSr7)r=rZafter_test_outside_fixturesrr"r"r#after_test_fixturestsrcCsttj}|jr@tj|j}t|D]}||r(||q(t |ddr~tjt |j }t|D]}||sf||qft |ddr| |jgt|drtj}t|D]H}|jD]<}t ||}||} | r|||dk r|| qqqt|drft} tj}t|D]4}|jD]&}t ||}||s$| |q$q|| rf|| |ri} tt|dddD] } | jj} | | kr| | | <q| S|S)N __only_on__r __requires____prefer_requires__cSs|jj|jj|jjjfSr7)r rdriverdialectZserver_version_info)rr"r"r#sz+_possible_configs_for_cls..)key)setrBrrZ__unsupported_on__r>db_specrremoverrCto_listrintersection_updaterhasattrrrZmatching_config_reasonsextendrZenabled_for_configrc differencedifference_updatereversedrTr rvalues)rreasonsrrspec config_objr requirementcheckZ skip_reasons non_preferredZ per_dialectrr r"r"r#rxs^                        rcCsg}t||}t|ddrFt|dD] }|r$td|j|jfq$|sd|j|jdddtjDd|f}t|nRt |drt }t j t |j}|D]}||s||q||r||tj|krt||dS) NZ __skip_if__Fz'%s' skipped by %sz1'%s.%s' unsupported on any DB implementation %s%sz, c ss<|]4}d|jjdddt|jD|jjfVqdS)z '%s(%s)+%s'rcss|]}t|VqdSr7)r)r}digr"r"r# sz&_do_skips...N)r rjoinr>Z_server_versionr)r}rr"r"r#rs   z_do_skips..__prefer_backends__)rrrBZ skip_testrrrrrrrr>rrCrrrcrrr _setup_configpop)rrrcmsgrrrr"r"r#rs8          rcCstj|tdSr7)rBrpushr;)rctxr"r"r#rsrc@sneZdZejddZejddZejddZejddZd d Z ejd d d dZ ej ddZ dS)FixtureFunctionscOs tdSr7NotImplementedErrorselfargkwr"r"r#skip_test_exceptionsz$FixtureFunctions.skip_test_exceptioncOs tdSr7rrrXrr"r"r# combinationsszFixtureFunctions.combinationscOs tdSr7rrr"r"r# param_identszFixtureFunctions.param_identcOs tdSr7rrr"r"r#fixtureszFixtureFunctions.fixturecCs tdSr7rrr"r"r#get_current_test_namesz&FixtureFunctions.get_current_test_namer)returncCs tdSr7rrr"r"r#mark_base_test_classsz%FixtureFunctions.mark_base_test_classcCs tdSr7rrr"r"r# add_to_markerszFixtureFunctions.add_to_markerN) rr __qualname__abcabstractmethodrrrrrrabstractpropertyrr"r"r"r#rs    rcCs|adSr7)ry)Zfixture_fn_classr"r"r#set_fixture_functions sr)NF)U __future__rrargparserr-rHrqpathlibrrrVtypingrr'rrrKrr<r=r>r@rAr&r?rrBr;rCr/rrerbr__annotations__r$r)r+r,r2r6r9rGrrrrrr rr3rDrfrgrjrmrvrzr{rrrZrrrrrrrrrrrrrrrrrrABCrryrr"r"r"r# s       6           1          B+