U
|
¸ý°d+V ã @ sR d dl mZ d dlZd dlZd dlad dlZd dlZd dlZd dlm Z d dl
|
m Z dZ t d¡ZdadadadadadadadZdadadadadae Z e Z da dd Z d d
|
Z d d Z!d d Z"dd Z#dd Z$dd Z%dd Z&dd Z'dd Z(dd Z)d d Z*d d Z+d!d" Z,d#d$ Z-g Z.g Z/d%d& Z0d'd( Z1e0d)d* Z2e0d+d, Z3e1d-d. Z4e1d/d0 Z5e0d1d2 Z6e1d3d4 Z7e1d5d6 Z8d7d8 Z9e1d9d: Z:e1d;d< Z;e1d=d> Z<d?d@ Z=dAdB Z>dCdD Z?dEdF Z@dGdH ZAdIdJ ZBdKdL ZCdMdN ZDdOdP ZEdQdR ZFdSdT ZGdUdV ZHdadWdXZIdYdZ ZJd[d\ ZKG d]d^ d^ejLZMdaNd_d` ZOdS )bé )Ú annotationsN)ÚAny)ÚasyncioFz sqlalchemy.testing.plugin_basec C s¸ | ddt tdd | ddt tdd | ddt d d
|
d | d dtd d | ddt ddd | ddt ddd | ddddd | dddd | d dtd d d | d dtd!d"d | d#dtd$d%d | d&dtd'd(d | d)t d*d+d,d- | d.t d/d0d1 | d2dd3d4d | d5t d6d7d1 | d8dt td9d | d:dtt d;d< | d=dtt d>d< | d?dd@dAdBdC | dDddEdAdFdC | dGt dHdId1 | dJt dg dKdLdM | dNt dOdP | dQddRd d S )SNz
|
--log-infoÚcallbackz,turn on info logging for <LOG> (multiple OK))ÚactionÚtyper Úhelpz --log-debugz-turn on debug logging for <LOG> (multiple OK)z--dbÚappendÚdbzBUse prefab database uri. Multiple OK, first one is run by default.)r r Údestr z--dbszList available prefab dbs)r Zzeroarg_callbackr z--dburiÚdburiz8Database uri. Multiple OK, first one is run by default.z
|
--dbdriverÚdbdriverz|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 r r z--disable-asyncioz8disable test / fixtures / provisoning running in asyncio)r r z--backend-onlyÚbackendz}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-connectionsÚlow_connectionsz>Use a low number of distinct connections - i.e. for Oracle TNSz--write-identsÚ write_identszCwrite out generated follower idents to <file>, when -n<num> is usedz--requirementsz3requirements class for testing, overrides setup.cfgz --include-tagzAInclude tests with tag <tag>; legacy, use pytest -m 'tag' instead)r r r r z --exclude-tagzEExclude tests with tag <tag>; legacy, use pytest -m 'not tag' insteadz--write-profilesZwrite_profilesFz$Write/update failing profiling data.)r r r r 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 r r 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_option© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/testing/plugin/plugin_base.pyÚ setup_options9 s@ ûûûüûû üýû üüüûüüüûûûûûüú
|
ýýr! c C s ddl m} | |_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_followerî s r& c C s dS )z¯Store 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_configú s r( c C s dS )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_config s r) c C s* t ¡ at t| d t| d g¡ d S )Nz setup.cfgztest.cfg)Ú configparserÚ ConfigParserÚ file_configÚreadr )Ú root_pathr r r Ú read_config s ÿr/ c C s | a tD ]}|t t qdS )z7things to set up early, before coverage might be setup.N)ÚoptionsÚ pre_configurer, )ÚoptÚfnr r r Ú pre_begin s r4 c C s
|
| t _d S ©N)r0 Z has_coverage)Úvaluer r r Úset_coverage_flag s r7 c C sp t D ]} | tt qddlma ddlmamama ddlm a m
|
a
|
m a ddlm a m a ddlma t
|
¡ dS )z9things to set up later, once we know coverage is running.r ©Útesting)ÚfixturesÚenginesÚ
|
exclusions)Ú
|
assertionsÚwarningsÚ profiling)Úconfigr# )ÚutilN)Úpost_configurer0 r, Ú
|
sqlalchemyr9 r$ r: r; r< r= r> r? r@ r# rA Z setup_filters©r3 r r r Ú
|
post_begin" s rE c C sR t sdd l a t ¡ | d¡r2t |¡ t j¡ n | d¡rNt |¡ t j¡ d S )Nr z-infoz-debug)ÚloggingÚ basicConfigÚendswithÚ getLoggerÚsetLevelÚINFOÚDEBUG©Úopt_strr6 Úparserr r r r 4 s
|
|
r c G sB t d tt d¡D ] }t d|t d|¡f qt d¡ d S )Nz0Available --db options (use --dburi to override)r
|
z%20s %sr )ÚprintÚsortedr, r0 ÚgetÚsysÚexit)ÚargsÚmacror r r r A s r c C s t | d S r5 )Ú_setup_requirementsrM r r r r H s r c s fdd}|S )Nc s t | | d S r5 )r rM ©Útagr r Ú_do_include_tagM s z)_set_tag_include.<locals>._do_include_tagr )rY rZ r rX r r L s r c s fdd}|S )Nc s t | | d S r5 )r rM rX r r Ú_do_exclude_tagT s z)_set_tag_exclude.<locals>._do_exclude_tagr )rY r[ r rX r r S s r c C s t | dd¡¡ d S ©Nú-Ú_)Ú exclude_tagsÚaddÚreplacerM r r r r Z s r c C s t | dd¡¡ d S r\ )Ú include_tagsr` ra rM r r r r ^ s r c C s t | ¡ | S r5 )r1 r rD r r r Úpref s
|
rc c C s t | ¡ | S r5 )rB r rD r r r Úpostk s
|
rd c C s | a d S r5 ©r0 ©r2 r, r r r Ú_setup_optionsp s rg c C s, ddl m} | ddd¡ | ddd¡ d S )Nr )Úregistryzsqlite.pysqlite_numericz#sqlalchemy.dialects.sqlite.pysqliteZ _SQLiteDialect_pysqlite_numericzsqlite.pysqlite_dollarZ _SQLiteDialect_pysqlite_dollar)Zsqlalchemy.dialectsrh Úregister)r2 r, rh r r r Ú _register_sqlite_numeric_dialectv s ýýrj c
|
C s^ t j dd¡dkrZddlm} z|dd W n, tk
|
rX } ztd|W 5 d }~X Y nX d S ) NZREQUIRE_SQLALCHEMY_CEXTÚ0Ú1r )Úhas_compiled_extT)Zraise_zEREQUIRE_SQLALCHEMY_CEXT is set but can't import the cython extensions)ÚosÚenvironrR Zsqlalchemy.utilrm Ú ImportErrorÚAssertionError)r2 r, rm Úerrr r r Ú __ensure_cext s ÿýrs c C s ddl m} t |_d S ©Nr ©r@ )r$ r@ Ú_fixture_fn_classZ_fixture_functions)r0 r, r@ r r r Ú _init_symbols s rw c C s | j r dt_d S )NF)Zdisable_asyncior ZENABLE_ASYNCIOrf r r r Ú_set_disable_asyncio s rx c
|
C sH ddl m} ddlm} ddlm} | jr6t| j}ng }| jpBg }| jr| jD ]D}t
|
d|¡D ]2}|| d¡krt d| q`| | d|¡¡ q`qP|s¬| | dd¡¡ d |_t| ||¡} | D ]|}
|
t d |
|
¡ | jr |jr t| jd
|
&} | |j d |
|
jd d d¡ W 5 Q R X | |
|
| ||j¡} |jsÆ| | |¡ qÆd S )Nr r8 ru r" z[,\s]+r
|
z:Unknown URI specifier '%s'. Specify --dbs for known uris.r zAdding database URL: %sÚaú F)Z hide_passwordÚ
|
)rC r9 r$ r@ r# r Úlistr r
|
ÚreÚsplitr0 Ú RuntimeErrorr rR Ú_currentZgenerate_db_urlsÚlogÚinfor r% ÚopenÚwriteZrender_as_stringZ setup_configZset_as_current) r0 r, r9 r@ r# Zdb_urlsZ extra_driversZdb_tokenr
|
Z expanded_urlsZdb_urlÚfile_Úcfgr r r Ú _engine_uri¢ sH
|
|
ÿÿ ÿ ÿr c C s | dd¡}t| d S )NÚ sqla_testingÚrequirement_cls)rR rW )r0 r, r r r r Ú _requirementsÓ s r c C s ddl m} ddlm} |jd k r&d S | d¡\}}t|}| d¡dd
D ]}t||}qNt||}| |_|_t |_ d S )Nr ru r8 ú:Ú.é )
|
r$ r@ rC r9 Ú requirementsr~ Ú
|
__import__ÚgetattrÚrequiresÚbootstrapped_as_sqlalchemy)Zargumentr@ r9 ÚmodnameZclsnameÚmodÚ componentZreq_clsr r r rW Ú s
|
|
rW c C s@ ddl m} | jr<ddl m} |j ¡ D ]}| ||j¡ q(d S )Nr ru r" )r$ r@ r r# ÚConfigÚ all_configsZdrop_all_schema_objectsr
|
)r0 r, r@ r# r r r r Ú_prep_testing_databaseï s
|
r c C s ddl m} t|_||_d S rt )r$ r@ r0 r, )r2 r, r@ r r r Ú_post_setup_optionsú s r c C s. ddl m} |j| dd¡| j| jd|_d S )Nr )r? r Z profile_file)ÚsortÚdump)r$ r? ZProfileStatsFilerR r r Z_profile_stats)r0 r, r? r r r Ú_setup_profiling s
|
ýr c C s& t |tjsdS | d¡r dS dS d S )NFr^ T)Ú
|
issubclassr: ZTestBaseÚ
|
startswith)ÚnameÚclsr r r Ú
|
want_class s
|
|
r¡ c C s& |j d¡sdS |jd kr dS dS d S )NZtest_FT)Ú__name__r Ú
|
__module__)r r3 r r r Ú want_method s
|
|
r¤ c c s d|ksd|krd|k}t | |dD ]\}| j}t dd|j¡}t dd|¡}d| j|f }t|| f||d }t||| |V q$n| V d S )
|
Nr Zsparse_backend)Úsparsez
|
[_\[\]\.]+r^ z_+$Ú z%s_%s)Ú_sa_orig_cls_nameÚ__only_on_config__)Ú_possible_configs_for_clsr¢ r} Úsubr r Úsetattr) r ÚmoduleÚmarkersr¥ r Ú orig_nameZ
|
alpha_namer Úsubclsr r r Úgenerate_sub_tests s ý
|
r° c C s t | t| d S r5 )Ú _do_skipsÚ _setup_engine©r r r r Ú!start_test_class_outside_fixtures7 s r´ c C s t | ¡ tj ¡ d S r5 )r: Z stop_test_class_inside_fixturesr; Útesting_reaperr³ r r r Ústop_test_class< s
|
r¶ c C s: t j ¡ t ttj| ¡ ztjs*t
|
¡ W 5 t X d S r5 ) r; rµ Ú stop_test_class_outside_fixturesr# r@ r
|
Ú_restore_enginer0 r r= Úglobal_cleanup_assertionsr³ r r r r· E s
|
r· c C s t jrt j t¡ d S r5 )r@ r Úresetr9 r r r r r¸ O s r¸ c C s t j ¡ t ¡ t d S r5 )r; rµ Z final_cleanupr= r¹ r¸ r r r r Úfinal_process_cleanupT s
|
r» c C s< t | dd r8t| j}d|d< tj|d}tj |t¡ d S )NÚ__engine_options__ÚclassZscopere ) r Údictr¼ r; Ztesting_enginer@ r Z push_enginer9 )r ÚoptsZengr r r r² Z s
|
|
r² c C s* t |d|j}d|||f }t |¡ d S )Nr§ z%s.%s.%s)r r¢ r? Z_start_current_test)ÚtestZtest_module_nameZ
|
test_classZ test_namer Zid_r r r Ú before_testb s rÁ c C s t ¡ tj ¡ d S r5 )r: Ú
|
after_testr; rµ ©rÀ r r r r n s r c C s t j | ¡ d S r5 )r; rµ Zafter_test_outside_fixturesrà r r r Úafter_test_fixturess s rÄ c C s² t tj ¡ }| jr@tj| j }t|D ]}||r(| |¡ q(t | dd r~tjt
|
| j ¡ }t|D ]}||sf| |¡ qft | dd r| | jg¡ t| drútj}t|D ]H}| jD ]<}t ||}| |¡} | rº| |¡ |d k rò| | ¡ q°qºq°t| drft }
|
tj}t|D ]4}| jD ]&}t ||}| |¡s$|
|
|¡ q$q| |
|
¡rf| |
|
¡ |r®i } tt|dd dD ] } | jj } | | kr| | | < q| ¡ S |S )NÚ __only_on__r¨ Ú __requires__Ú__prefer_requires__c S s | j j| j j| j jjfS r5 )r
|
r ÚdriverÚdialectZserver_version_info)r r r r Ú<lambda>ª s ýz+_possible_configs_for_cls.<locals>.<lambda>)Úkey) Úsetr@ r r Z__unsupported_on__r< Údb_specr| Úremover rA Úto_listrÅ Úintersection_updater¨ Úhasattrr rÆ Zmatching_config_reasonsÚextendrÇ Zenabled_for_configr` Ú
|
differenceÚdifference_updateÚreversedrQ r
|
r Úvalues)r Úreasonsr¥ r ÚspecÚ
|
config_objr Ú requirementÚcheckZ skip_reasonsÚ non_preferredZ per_dialectr r
|
r r r r© w s^
|
|
|
|
|
|
|
|
|
þÿ
|
|
r© c C sô g }t | |}t| ddrFt| dD ] }| r$t d| j|jf ¡ q$|sd| j| jd dd tj ¡ D ¡d |¡f }t |¡ nRt | drØt
|
}t j t | j¡ }|D ]}||s¬| |¡ q¬| |¡rØ| |¡ tj|krðt| ¡ | d S ) NZ __skip_if__Fz'%s' skipped by %sz1'%s.%s' unsupported on any DB implementation %s%sz, c s s< | ]4}d |j jd dd t |j ¡D ¡|j jf V qdS )z '%s(%s)+%s'r c s s | ]}t |V qd S r5 )r )Ú.0Údigr r r Ú <genexpr>Ì s ÿz&_do_skips.<locals>.<genexpr>.<genexpr>N)r
|
r Újoinr< Z_server_versionrÈ )rÝ rÙ r r r rß È s
|
÷
|
|
þúÿz _do_skips.<locals>.<genexpr>Ú__prefer_backends__)r© r r@ Z skip_testr¢ r£ rà r r rÑ rÌ r< rÍ rA rÏ rá r` rÓ rÔ r Ú _setup_configÚpop)r r× r ÚcÚmsgrÜ rØ rÙ r r r r± ¹ s8
|
ÿ
|
|
ö ñ
|
|
|
|
r± c C s t j | t¡ d S r5 )r@ r Úpushr9 )rÙ Úctxr r r râ ä s râ c @ sn e Zd Zejdd Zejdd Zejdd Zejdd Zd d
|
Z ejd d d dZ
|
ej dd Z dS )ÚFixtureFunctionsc O s
|
t d S r5 ©ÚNotImplementedError©ÚselfÚargÚkwr r r Úskip_test_exceptioné s z$FixtureFunctions.skip_test_exceptionc O s
|
t d S r5 ré ©rì rU rî r r r Ú combinationsí s z FixtureFunctions.combinationsc O s
|
t d S r5 ré rð r r r Ú param_identñ s z FixtureFunctions.param_identc O s
|
t d S r5 ré rë r r r Úfixtureõ s zFixtureFunctions.fixturec C s
|
t d S r5 ré ©rì r r r Úget_current_test_nameù s z&FixtureFunctions.get_current_test_namer )Úreturnc C s
|
t d S r5 ré rô r r r Úmark_base_test_classü s z%FixtureFunctions.mark_base_test_classc C s
|
t d S r5 ré rô r r r Ú add_to_marker s z FixtureFunctions.add_to_markerN) r¢ r£ Ú __qualname__ÚabcÚabstractmethodrï rñ rò ró rõ r÷ Úabstractpropertyrø r r r r rè è s
|
|
|
|
rè c C s | a d S r5 )rv )Zfixture_fn_classr r r Úset_fixture_functions s rý )NF)PÚ
|
__future__r rú r* rF rn r} rS Útypingr r$ r r rI r r: r; r< r> r? r# r= r r@ r9 rA r, rÌ rb r_ r0 r! r& r( r) r/ r4 r7 rE r r r r r r r r1 rB rc rd rg rj rs rw rx r r rW r r r r¡ r¤ r° r´ r¶ r· r¸ r» r² rÁ r rÄ r© r± râ ÚABCrè rv rý r r r r Ú<module>
|
sª
|
6
|
|
|
|
|
|
|
|
0
|
|
|
|
|
|
|
|
B+
|
