o gJT@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 zeroarg_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 --nomemorymemory_intensivezpDon't run memory profiling tests; this is now equivalent to the pytest -m 'not memory_intensive' mark expressionz--notimingintensivetiming_intensivezpDon't run timing intensive tests; this is now equivalent to the pytest -m 'not timing_intensive' mark expressionz--nomypymypyz_Don't run mypy typing tests; this is now equivalent to the pytest -m 'not mypy' mark expressionz--profile-sort 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-profileswrite_profilesFz$Write/update failing profiling data.)r rrr z--force-write-profilesforce_write_profilesz,Unconditionally write/update profiling data.z--dump-pyannotatedump_pyannotatez/Run pyannotate and dump json info to given filez--mypy-extra-test-pathmypy_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/api-tag/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)follower_identr/r+r+r,configure_followers  r3cCdS)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+dict_r+r+r,!memoize_important_follower_configr7cCr4)zRestore important configuration needed by a follower. This invokes in the follower process. Hook is currently not used. Nr+r5r+r+r,!restore_important_follower_configr8r9cCs*tatt|dt|dgdS)Nz setup.cfgztest.cfg) configparser ConfigParser file_configreadr") root_pathr+r+r, read_configsr?cCs|atD]}|ttqdS)z7things to set up early, before coverage might be setup.N)r pre_configurer<)optfnr+r+r, pre_begins rCcCs |t_dSN)r has_coverage)valuer+r+r,set_coverage_flag s rGcCsptD]}|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< sqlalchemyrIr0rJrKrLrMrNrOrPr/rQ setup_filtersrBr+r+r, post_begin$s    rVcCsXts ddlat|drt|tjdS|dr*t|tjdSdS)Nrz-infoz-debug)logging basicConfigendswith getLoggersetLevelINFODEBUGopt_strrFparserr+r+r,r#6s  r#cGsVtdur tttdttdD]}td|td|fqt ddS)Nz0Available --db options (use --dburi to override)r z%20s %sr) r<r?rcwdprintsortedrgetsysexit)argsmacror+r+r,r$Cs  r$cCs t|dSrD)_setup_requirementsr^r+r+r,r'Ns r'cfdd}|S)Nct||dSrD)r(r^tagr+r,_do_include_tagSz)_set_tag_include.._do_include_tagr+)rmrnr+rlr,r%R r%crj)NcrkrD)r)r^rlr+r,_do_exclude_tagZroz)_set_tag_exclude.._do_exclude_tagr+)rmrqr+rlr,r&Yrpr&cCt|dddSN-_) exclude_tagsaddreplacer^r+r+r,r)`r)cCrrrs) include_tagsrwrxr^r+r+r,r(dryr(cCt||SrD)r@r rUr+r+r,prel r|cCr{rD)rRr rUr+r+r,postqr}r~cC|adSrDrrAr<r+r+r,_setup_optionsvsrcCs,ddlm}|ddd|ddddS)Nr)registryzsqlite.pysqlite_numericz#sqlalchemy.dialects.sqlite.pysqlite_SQLiteDialect_pysqlite_numericzsqlite.pysqlite_dollar_SQLiteDialect_pysqlite_dollar)sqlalchemy.dialectsrregister)rAr<rr+r+r, _register_sqlite_numeric_dialect|s rc CsVtjdddkr)ddlm}z|ddWdSty(}ztd|d}~wwdS) NREQUIRE_SQLALCHEMY_CEXT01r)has_compiled_extT)raise_zEREQUIRE_SQLALCHEMY_CEXT is set but can't import the cython extensions)osenvironrdsqlalchemy.utilr ImportErrorAssertionError)rAr<rerrr+r+r, __ensure_cexts rcCsddlm}t|_dSNrrP)r0rP_fixture_fn_class_fixture_functions)rr<rPr+r+r, _init_symbolss  rcCs|jrdt_dSdSNF)disable_asynciorENABLE_ASYNCIOrr+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)NrrHrr.)urlz[,\s]+r z:Unknown URI specifier '%s'. Specify --dbs for known uris.rcsg|]}|qSr+)make_url).0db_urlsa_urlr+r, sz_engine_uri..a F) hide_password zAdding database URL: %s)rSrIr0rPr/sqlalchemy.enginerrlistrr resplitr RuntimeErrorr rd_currentrr1openwriterender_as_stringgenerate_db_urlsloginfo setup_configset_as_current) rr<rIrPr/db_urls extra_driversdb_tokenr rfile_ expanded_urlscfgr+rr, _engine_urisT             rcCs|dd}t|dS)N sqla_testingrequirement_cls)rdri)rr<rr+r+r, _requirementss  rcCsrddlm}ddlm}|d\}}t|}|dddD]}t||}q t||}||_|_t |_ dS)NrrrH:.) r0rPrSrIr __import__getattr requirementsrequiresbootstrapped_as_sqlalchemy)argumentrPrImodnameclsnamemod componentreq_clsr+r+r,ris     ricCsDddlm}|jrddlm}|jD] }|||jqdSdS)Nrrr.)r0rPrr/Config all_configsdrop_all_schema_objectsr )rr<rPr/rr+r+r,_prep_testing_databases  rcCsddlm}t|_||_dSr)r0rPrr<)rAr<rPr+r+r,_post_setup_optionss  rcCs.ddlm}|j|dd|j|jd|_dS)Nr)rOr profile_file)sortdump)r0rOProfileStatsFilerdrr_profile_stats)rr<rOr+r+r,_setup_profilings   rcCs"t|tjsdS|drdSdS)NFruT) issubclassrJTestBase startswith)nameclsr+r+r, want_class  rcCs"|jdsdS|jdurdSdS)Ntest_FT)__name__r __module__)rrBr+r+r, want_methodrrc csd|vs d|vrDd|v}t||dD].}|j}tdd|j}tdd|}d|j|f}t||f||d }t||||VqdS|VdS) Nrsparse_backend)sparsez [_\[\]\.]+ruz_+$z%s_%s)_sa_orig_cls_name__only_on_config__)_possible_configs_for_clsrrsubrr setattr) rmodulemarkersrr orig_name alpha_namersubclsr+r+r,generate_sub_tests!s"  rcCst|t|dSrD) _do_skips _setup_enginerr+r+r,!start_test_class_outside_fixtures9s rcCst|tjdSrD)rJstop_test_class_inside_fixturesrKtesting_reaperrr+r+r,stop_test_class>s rcCsJtjtttj|ztjst Wt dSWt dSt wrD) rKr stop_test_class_outside_fixturesr/rPr rrrMglobal_cleanup_assertions_restore_enginerr+r+r,rGs   rcCstjr tjtdSdSrD)rPrresetrIr+r+r+r,rQsrcCstjttdSrD)rKr final_cleanuprMrrr+r+r+r,final_process_cleanupVs  rcCs@t|ddrt|j}d|d<tj|d}tj|tdSdS)N__engine_options__classscoper) rdictrrKtesting_enginerPr push_enginerI)roptsengr+r+r,r\s   rcCs*t|d|j}d|||f}t|dS)Nrz%s.%s.%s)rrrO_start_current_test)testtest_module_name test_class test_namerid_r+r+r, before_testdsrcCsttjdSrD)rJ after_testrKrr r+r+r,rosrcCstj|dSrD)rKrafter_test_outside_fixturesrr+r+r,after_test_fixturestrorcCsttj}|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|jjjfSrD)r rdriverdialectserver_version_info)rr+r+r,sz+_possible_configs_for_cls..)key)setrPrr__unsupported_on__rLdb_specrremoverrQto_listrintersection_updaterhasattrrrmatching_config_reasonsextendrenabled_for_configrw differencedifference_updatereversedrcr rvalues)rreasonsrrspec config_objr requirementcheck skip_reasons non_preferred 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) N __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|VqdSrD)r")rdigr+r+r, s  z&_do_skips...N)r rjoinrL_server_versionr)rr+r+r+r,r3s    z_do_skips..__prefer_backends__)rrrP skip_testrrr4rrr!rrLrrQrr6rwr%r&r _setup_configpop)rr)rcmsgr/r*r+r+r+r,rs>           rcCstj|tdSrD)rPrpushrI)r+ctxr+r+r,r8sr8c@sjeZdZejddZejddZejddZejddZd d Z ejdd dZ ej ddZ dS)FixtureFunctionscOtrDNotImplementedErrorselfargkwr+r+r,skip_test_exceptionz$FixtureFunctions.skip_test_exceptioncOr?rDr@rCrgrEr+r+r, combinationsrGzFixtureFunctions.combinationscOr?rDr@rHr+r+r, param_identrGzFixtureFunctions.param_identcOr?rDr@rBr+r+r,fixturerGzFixtureFunctions.fixturecCr?rDr@rCr+r+r,get_current_test_namesz&FixtureFunctions.get_current_test_namereturnrcCr?rDr@rLr+r+r,mark_base_test_classrGz%FixtureFunctions.mark_base_test_classcCr?rDr@rLr+r+r, add_to_markerrGzFixtureFunctions.add_to_markerN)rNr) rr __qualname__abcabstractmethodrFrIrJrKrMrOabstractpropertyrPr+r+r+r,r>s     r>cCrrD)r)fixture_fn_classr+r+r,set_fixture_functions srVr)U __future__rrRargparserr:rWrpathlibrrretypingrr0rrrZrrJrKrLrNrOr/rMrrPrIrQr<rrzrvr__annotations__r-r3r7r9r?rCrGrVr#r$r'r%r&r)r(r@rRr|r~rrrrrrrrirrrrrrrrrrrrrrrrrr8ABCr>rrVr+r+r+r,s       6           1           B+