o E6d^V@sjUddlmZddlZddlmZddlZddladdlZddlZddl Z ddl m Z ddl m Z  dZtdZdadadadadadadadZdadadadadaeZeZda de!d <d d Z"d d Z#ddZ$ddZ%ddZ&ddZ'ddZ(ddZ)ddZ*ddZ+ddZ,d d!Z-d"d#Z.d$d%Z/d&d'Z0gZ1gZ2d(d)Z3d*d+Z4e3d,d-Z5e3d.d/Z6e4d0d1Z7e4d2d3Z8e3d4d5Z9e4d6d7Z:e4d8d9Z;d:d;Zd?Z>e4d@dAZ?dBdCZ@dDdEZAdFdGZBdHdIZCdJdKZDdLdMZEdNdOZFdPdQZGdRdSZHdTdUZIdVdWZJdXdYZKdddZd[ZLd\d]ZMd^d_ZNGd`dadaejOZPdaQdbdcZRdS)e) annotationsN) Namespace)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.)rr destr z--dbszList available prefab dbs)rZzeroarg_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)rr r z--disable-asyncioz8disable test / fixtures / provisoning running in asyncio)rr 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 defaultr r z--profile-dump profiledumpz2Filename where a single profile run will be dumped)r r r 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)rrr r z --exclude-tagzEExclude tests with tag ; legacy, use pytest -m 'not tag' insteadz--write-profilesZwrite_profilesFz$Write/update failing profiling data.)rr rr 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 rrr r 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!|C:\Users\jesus\OneDrive\Desktop\erpjis_fastapi\backend\jisbackend\Lib\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_configsr3cCs|atD]}|ttqdS)z7things to set up early, before coverage might be setup.N)r pre_configurer0)optfnr!r!r" pre_begins r7cCs |t_dSN)rZ has_coverage)valuer!r!r"set_coverage_flags 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_configurerr0 sqlalchemyr<r&r=r>r?r@rArBrCr%rDZ setup_filtersr6r!r!r" post_begin#s    rHcCsXts ddlat|drt|tjdS|dr*t|tjdSdS)Nrz-infoz-debug)logging basicConfigendswith getLoggersetLevelINFODEBUGopt_strr9parserr!r!r"r5s  rcGsBtdttdD]}td|td|fq tddS)Nz0Available --db options (use --dburi to override)r z%20s %sr)printsortedr0rgetsysexit)argsmacror!r!r"rBsrcCs t|dSr8)_setup_requirementsrPr!r!r"rIs rcfdd}|S)Nct||dSr8)rrPtagr!r"_do_include_tagNz)_set_tag_include.._do_include_tagr!)r^r_r!r]r"rM rcr[)Ncr\r8)rrPr]r!r"_do_exclude_tagUr`z)_set_tag_exclude.._do_exclude_tagr!)r^rbr!r]r"rTrarcCt|dddSN-_) exclude_tagsaddreplacerPr!r!r"r[rcCrcrd) include_tagsrhrirPr!r!r"r_rjrcCt||Sr8)r4r rGr!r!r"preg rmcCrlr8)rEr rGr!r!r"postlrnrocC|adSr8rr5r0r!r!r"_setup_optionsqsrscCs,ddlm}|ddd|ddddS)Nr)registryzsqlite.pysqlite_numericz#sqlalchemy.dialects.sqlite.pysqliteZ_SQLiteDialect_pysqlite_numericzsqlite.pysqlite_dollarZ_SQLiteDialect_pysqlite_dollar)Zsqlalchemy.dialectsrtregister)r5r0rtr!r!r" _register_sqlite_numeric_dialectws rvc 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)osenvironrUZsqlalchemy.utilry ImportErrorAssertionError)r5r0ryerrr!r!r" __ensure_cexts rcCsddlm}t|_dSNrrC)r&rC_fixture_fn_classZ_fixture_functions)rr0rCr!r!r" _init_symbolss  rcCs|jrdt_dSdSNF)Zdisable_asynciorZENABLE_ASYNCIOrrr!r!r"_set_disable_asyncios rc CsVddlm}ddlm}ddlm}|jrt|j}ng}|jp!g}|jrJ|jD]!}t d|D]}|| dvr?t d|| |d|q0q(|sU| |ddd|_t|||} | D]F} td | |jr|jrt|jd } | |jd | jd d dWdn1swY|| |||j} |js| | |qbdS)Nrr;rr$z[,\s]+r z:Unknown URI specifier '%s'. Specify --dbs for known uris.rzAdding database URL: %sa F)Z hide_password )rFr<r&rCr%rlistrr resplitr RuntimeErrorr rU_currentZgenerate_db_urlsloginforr'openwriteZrender_as_stringZ setup_configZset_as_current) rr0r<rCr%Zdb_urlsZ extra_driversZdb_tokenr Z expanded_urlsZdb_urlfile_cfgr!r!r" _engine_urisN            rcCs|dd}t|dS)N sqla_testingrequirement_cls)rUrZ)rr0rr!r!r" _requirementss  rcCsddlm}ddlm}|jdurdS|d\}}t|}|dddD]}t||}q't||}||_|_t |_ dS)Nrrr;:.) r&rCrFr< requirementsr __import__getattrrequiresbootstrapped_as_sqlalchemy)ZargumentrCr<modnameclsnamemod componentZreq_clsr!r!r"rZs      rZcCsDddlm}|jrddlm}|jD] }|||jqdSdS)Nrrr$)r&rCrr%Config all_configsZdrop_all_schema_objectsr )rr0rCr%rr!r!r"_prep_testing_databases  rcCsddlm}t|_||_dSr)r&rCrr0)r5r0rCr!r!r"_post_setup_optionss  rcCs.ddlm}|j|dd|j|jd|_dS)Nr)rBrZ profile_file)sortdump)r&rBZProfileStatsFilerUrrZ_profile_stats)rr0rBr!r!r"_setup_profilings   rcCs"t|tjsdS|drdSdS)NFrfT) issubclassr=ZTestBase startswith)nameclsr!r!r" want_class   rcCs"|jdsdS|jdurdSdS)NZtest_FT)__name__r __module__)rr6r!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 [_\[\]\.]+rfz_+$z%s_%s)_sa_orig_cls_name__only_on_config__)_possible_configs_for_clsrrsubrr setattr) rmodulemarkersrrZ orig_nameZ alpha_namersubclsr!r!r"generate_sub_testss"  rcCst|t|dSr8) _do_skips _setup_enginerr!r!r"!start_test_class_outside_fixtures5s rcCst|tjdSr8)r=Zstop_test_class_inside_fixturesr>testing_reaperrr!r!r"stop_test_class:s rcCsJtjtttj|ztjst Wt dSWt dSt wr8) r>r stop_test_class_outside_fixturesr%rCr rrr@global_cleanup_assertions_restore_enginerr!r!r"rCs   rcCstjr tjtdSdSr8)rCrresetr<r!r!r!r"rMsrcCstjttdSr8)r>rZ final_cleanupr@rrr!r!r!r"final_process_cleanupRs  rcCs@t|ddrt|j}d|d<tj|d}tj|tdSdS)N__engine_options__classscoperq) rdictrr>Ztesting_enginerCrZ push_enginer<)roptsZengr!r!r"rXs   rcCs*t|d|j}d|||f}t|dS)Nrz%s.%s.%s)rrrBZ_start_current_test)testZtest_module_nameZ test_classZ test_namerZid_r!r!r" before_test`srcCsttjdSr8)r= after_testr>rrr!r!r"rksrcCstj|dSr8)r>rZafter_test_outside_fixturesrr!r!r"after_test_fixturespr`rcCsttj}|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|jjjfSr8)r rdriverdialectZserver_version_info)rr!r!r"sz+_possible_configs_for_cls..)key)setrCrrZ__unsupported_on__r?db_specrremoverrDto_listrintersection_updaterhasattrrrZmatching_config_reasonsextendrZenabled_for_configrh differencedifference_updatereversedrTr rvalues)rreasonsrrspec config_objr requirementcheckZ skip_reasons non_preferredZ per_dialectrr r!r!r"rtsl                       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|VqdSr8)r).0digr!r!r" s  z&_do_skips...N)r rjoinr?Z_server_versionr)rrr!r!r"rs    z_do_skips..__prefer_backends__)rrrCZ skip_testrrrrrrrr?rrDrrrhrrr _setup_configpop)rrrcmsgrrrr!r!r"rs>           rcCstj|tdSr8)rCrpushr<)rctxr!r!r"rsrc@sjeZdZejddZejddZejddZejddZd d Z ejdd dZ ej ddZ dS)FixtureFunctionscOtr8NotImplementedErrorselfargkwr!r!r"skip_test_exceptionz$FixtureFunctions.skip_test_exceptioncOrr8rrrXrr!r!r" combinationsrzFixtureFunctions.combinationscOrr8rrr!r!r" param_identrzFixtureFunctions.param_identcOrr8rrr!r!r"fixturerzFixtureFunctions.fixturecCrr8rrr!r!r"get_current_test_namesz&FixtureFunctions.get_current_test_namereturnrcCrr8rrr!r!r"mark_base_test_classrz%FixtureFunctions.mark_base_test_classcCrr8rrr!r!r" add_to_markerrzFixtureFunctions.add_to_markerN)rr) rr __qualname__abcabstractmethodrrrrrrabstractpropertyr r!r!r!r"rs     rcCrpr8)r)Zfixture_fn_classr!r!r"set_fixture_functionssrr)S __future__rr argparserr.rIrzrrVtypingrr&rrrLrr=r>r?rArBr%r@rrCr<rDr0rrkrgr__annotations__r#r(r+r-r3r7r:rHrrrrrrrr4rErmrorsrvrrrrrrZrrrrrrrrrrrrrrrrrrABCrrrr!r!r!r"s      6          /           B+