o ZhJT@svUddlmZddlZddlmZddlZddladdlZddlm Z ddl Z ddl Z ddl m Z ddlmZ dZtdZdadadadadadadadZdadadadadaeZ 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"\/var/www/html/lang_env/lib/python3.10/site-packages/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)cCdS)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_configr,cCr*)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_configr-r.cCs*tatt|dt|dgdS)Nz setup.cfgztest.cfg) configparser ConfigParser file_configreadr) root_pathr"r"r# read_configsr4cCs|atD]}|ttqdS)z7things to set up early, before coverage might be setup.N)r pre_configurer1)optfnr"r"r# pre_begins r8cCs |t_dSN)rZ has_coverage)valuer"r"r#set_coverage_flag s r;cCsptD]}|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_configurerr1 sqlalchemyr=r'r>r?r@rArBrCrDr&rEZ setup_filtersr7r"r"r# post_begin$s    rIcCsXts ddlat|drt|tjdS|dr*t|tjdSdS)Nrz-infoz-debug)logging basicConfigendswith getLoggersetLevelINFODEBUGopt_strr:parserr"r"r#r6s  rcGsVtdur tttdttdD]}td|td|fqt ddS)Nz0Available --db options (use --dburi to override)r z%20s %sr) r1r4rcwdprintsortedrgetsysexit)argsmacror"r"r#rCs  rcCs t|dSr9)_setup_requirementsrQr"r"r#rNs rcfdd}|S)Nct||dSr9)rrQtagr"r#_do_include_tagSz)_set_tag_include.._do_include_tagr")r`rar"r_r#rR rcr])Ncr^r9)r rQr_r"r#_do_exclude_tagZrbz)_set_tag_exclude.._do_exclude_tagr")r`rdr"r_r#rYrcrcCt|dddSN-_) exclude_tagsaddreplacerQr"r"r#r `r cCrerf) include_tagsrjrkrQr"r"r#rdrlrcCt||Sr9)r5r rHr"r"r#prel rocCrnr9)rFr rHr"r"r#postqrprqcC|adSr9rr6r1r"r"r#_setup_optionsvsrucCs,ddlm}|ddd|ddddS)Nr)registryzsqlite.pysqlite_numericz#sqlalchemy.dialects.sqlite.pysqliteZ_SQLiteDialect_pysqlite_numericzsqlite.pysqlite_dollarZ_SQLiteDialect_pysqlite_dollar)Zsqlalchemy.dialectsrvregister)r6r1rvr"r"r# _register_sqlite_numeric_dialect|s rxc CsVtjdddkr)ddlm}z|ddWdSty(}ztd|d}~wwdS) NZREQUIRE_SQLALCHEMY_CEXT01r)has_compiled_extT)Zraise_zEREQUIRE_SQLALCHEMY_CEXT is set but can't import the cython extensions)osenvironrWZsqlalchemy.utilr{ ImportErrorAssertionError)r6r1r{errr"r"r# __ensure_cexts rcCsddlm}t|_dSNrrD)r'rD_fixture_fn_classZ_fixture_functions)rr1rDr"r"r# _init_symbolss  rcCs|jrdt_dSdSNF)Zdisable_asynciorZENABLE_ASYNCIOrtr"r"r#_set_disable_asyncios rc szddlm}ddlm}ddlm}ddlm|jr!t|j}ng}|j p'g}|j rP|j D]!}t d|D]}|| dvrEtd|||d|q6q.|s[||dd d|_|jr|jrfd d |DD](} t|jd } | |jd | jdddWdn1swYqmt|||} | D]} td| || |||j} |js| | |qdS)Nrr<rr%)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)rGr=r'rDr&Zsqlalchemy.enginerrlistrr resplitr RuntimeErrorr rW_currentrr(openwriteZrender_as_stringZgenerate_db_urlsloginfoZ setup_configZset_as_current) rr1r=rDr&Zdb_urlsZ extra_driversZdb_tokenr rfile_Z expanded_urlscfgr"rr# _engine_urisT             rcCs|dd}t|dS)N sqla_testingrequirement_cls)rWr\)rr1rr"r"r# _requirementss  rcCsrddlm}ddlm}|d\}}t|}|dddD]}t||}q t||}||_|_t |_ dS)Nrrr<:.) r'rDrGr=r __import__getattr requirementsrequiresbootstrapped_as_sqlalchemy)argumentrDr=modnameZclsnamemod componentZreq_clsr"r"r#r\s     r\cCsDddlm}|jrddlm}|jD] }|||jqdSdS)Nrrr%)r'rDrr&Config all_configsZdrop_all_schema_objectsr )rr1rDr&rr"r"r#_prep_testing_databases  rcCsddlm}t|_||_dSr)r'rDrr1)r6r1rDr"r"r#_post_setup_optionss  rcCs.ddlm}|j|dd|j|jd|_dS)Nr)rCrZ profile_file)sortdump)r'rCZProfileStatsFilerWrrZ_profile_stats)rr1rCr"r"r#_setup_profilings   rcCs"t|tjsdS|drdSdS)NFrhT) issubclassr>ZTestBase startswith)nameclsr"r"r# want_class  rcCs"|jdsdS|jdurdSdS)NZtest_FT)__name__r __module__)rr7r"r"r# want_methodrrc csd|vs d|vrDd|v}t||dD].}|j}tdd|j}tdd|}d|j|f}t||f||d }t||||VqdS|VdS) NrZsparse_backend)sparsez [_\[\]\.]+rhz_+$z%s_%s)_sa_orig_cls_name__only_on_config__)_possible_configs_for_clsrrsubrr setattr) rmodulemarkersrrZ orig_nameZ alpha_namersubclsr"r"r#generate_sub_tests!s"  rcCst|t|dSr9) _do_skips _setup_enginerr"r"r#!start_test_class_outside_fixtures9s rcCst|tjdSr9)r>Zstop_test_class_inside_fixturesr?testing_reaperrr"r"r#stop_test_class>s rcCsJtjtttj|ztjst Wt dSWt dSt wr9) r?r stop_test_class_outside_fixturesr&rDr rrrAglobal_cleanup_assertions_restore_enginerr"r"r#rGs   rcCstjr tjtdSdSr9)rDrresetr=r"r"r"r#rQsrcCstjttdSr9)r?rZ final_cleanuprArrr"r"r"r#final_process_cleanupVs  rcCs@t|ddrt|j}d|d<tj|d}tj|tdSdS)N__engine_options__classscopers) rdictrr?Ztesting_enginerDrZ push_enginer=)roptsZengr"r"r#r\s   rcCs*t|d|j}d|||f}t|dS)Nrz%s.%s.%s)rrrCZ_start_current_test)testZtest_module_nameZ test_classZ test_namerZid_r"r"r# before_testdsrcCsttjdSr9)r> after_testr?rrr"r"r#rosrcCstj|dSr9)r?rZafter_test_outside_fixturesrr"r"r#after_test_fixturestrbrcCsttj}|jr tj|j}t|D] }||r||qt |ddr?tjt |j }t|D] }||s>||q3t |ddrL| |jgt|dr}tj}t|D]$}|jD]}t ||}||} | r{|||dury|| nq]qXt|drt} tj}t|D]}|jD]}t ||}||s| |qq|| r|| |ri} tt|dddD]} | jj} | | vr| | | <q| S|S)N __only_on__r __requires____prefer_requires__cSs|jj|jj|jjjfSr9)r rdriverdialectZserver_version_info)rr"r"r#sz+_possible_configs_for_cls..)key)setrDrrZ__unsupported_on__r@db_specrremoverrEto_listrintersection_updaterhasattrrrZmatching_config_reasonsextendrZenabled_for_configrj differencedifference_updatereversedrVr rvalues)rreasonsrrspec config_objr requirementcheckZ skip_reasons non_preferredZ per_dialectrr r"r"r#rxsl                       rcCsg}t||}t|ddr#t|dD]}|r"td|j|jfq|sCd|j|jdddtjDd|f}t|n)t |drlt }t j t |j}|D] }||sa||qV||rl||tj|vrzt||dSdS) NZ __skip_if__Fz'%s' skipped by %sz1'%s.%s' unsupported on any DB implementation %s%sz, c ss>|]}d|jjdddt|jD|jjfVqdS)z '%s(%s)+%s'rcss|]}t|VqdSr9)r)rdigr"r"r# s  z&_do_skips...N)r rjoinr@Z_server_versionr)rrr"r"r#rs    z_do_skips..__prefer_backends__)rrrDZ skip_testrrrrrrrr@rrErrrjrrr _setup_configpop)rrrcmsgrrrr"r"r#rs>           rcCstj|tdSr9)rDrpushr=)rctxr"r"r#rsrc@sjeZdZejddZejddZejddZejddZd d Z ejdd dZ ej ddZ dS)FixtureFunctionscOtr9NotImplementedErrorselfargkwr"r"r#skip_test_exceptionz$FixtureFunctions.skip_test_exceptioncOrr9rrrZrr"r"r# combinationsrzFixtureFunctions.combinationscOrr9rrr"r"r# param_identrzFixtureFunctions.param_identcOrr9rrr"r"r#fixturerzFixtureFunctions.fixturecCrr9rrr"r"r#get_current_test_namesz&FixtureFunctions.get_current_test_namereturnrcCrr9rr r"r"r#mark_base_test_classrz%FixtureFunctions.mark_base_test_classcCrr9rr r"r"r# add_to_markerrzFixtureFunctions.add_to_markerN)r r) rr __qualname__abcabstractmethodrrr r r rabstractpropertyrr"r"r"r#rs     rcCrrr9)r)Zfixture_fn_classr"r"r#set_fixture_functions srr)U __future__rrargparserr/rJr|pathlibrrrXtypingrr'rrrMrr>r?r@rBrCr&rArrDr=rEr1rrmrir__annotations__r$r)r,r.r4r8r;rIrrrrrr rr5rFrorqrurxrrrrrr\rrrrrrrrrrrrrrrrrrABCrrrr"r"r"r#s       6           1           B+