U
|
¸ý°dt
ã @ sT d dl mZ d dlZd dlZd dlZd dlZd dlmZ d dlZ ddl
|
m Z ddl
|
m Z ddl
|
m Z ddl
|
mZ dd l mZ dd
|
l mZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddl
|
mZ ddl
|
mZ ddl
|
mZ ddl
|
mZ ddl
|
m Z ddl
|
m Z ddl
|
mZ ddl m Z ddl m Z! ddl m"Z" ddl m#Z# dd lm$Z$ dd l%m&Z& dd l'm(Z( e )¡ G d d d Z*da+G d!d" d"Z,G d#d$ d$e*Z-G d%d& d&Z.G d'd( d(Z/G d)d* d*Z0e1 Z2d+d, Z3d-d. Z4d/d0 Z5d1d2 Z6G d3d4 d4e*Z7G d5d6 d6e-e j8Z9G d7d8 d8e9Z:G d9d: d:e-Z;G d;d< d<Z<d@d>d?Z=dS )Aé )Ú annotationsN)ÚAnyé )Ú
|
assertions)Úconfig)Úmock©Úschema)Úeq_)Úne_)Ú BasicEntity)ÚComparableEntity)ÚComparableMixin)Úadict)Ú drop_all_tables_from_metadataé ©ÚColumn)Úevent)Úfunc©ÚInteger)Úselect©ÚTable)Úutil)ÚDeclarativeBase)Úevents)ÚMappedAsDataclass)Úregistry)Úsort_tables_and_constraints)Úvisitors)Ú ClauseElementc @ s e Zd ZdZdZdZdZdZd&ddZe
|
¡ dd Z e
|
¡ dd Z e
|
¡ d
|
d Z e
|
¡ d d Ze
|
¡ dd Ze j
|
dd Ze j
|
dd Ze
|
¡ dd Ze
|
¡ dd Ze
|
¡ dd Ze
|
¡ dd Ze j
|
d d Ze
|
¡ d d Ze j
|
d d! d"D d#d$d% ZdS )'ÚTestBase© NFc C s |s t |d S ©N)ÚAssertionError)ÚselfÚvalÚmsgr$ r$ úRd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/testing/fixtures.pyÚassert_C s zTestBase.assert_c c s" t jj}d t j_d V |t j_d S r% ©r ÚdbZ_compiled_cache©r' Ú_cacher$ r$ r* ÚnocacheF s zTestBase.nocachec c s0 t | dd ptj}| ¡ }|V W 5 Q R X d S ©NÚbind)Úgetattrr r- Úconnect)r' ÚengÚconnr$ r$ r* Úconnection_no_transM s
|
z TestBase.connection_no_transc c sJ t | dd ptj}| ¡ }| ¡ }|a|V d a|jr>| ¡ | ¡ d S r1 ) r3 r r- r4 ÚbeginÚ _connection_fixture_connectionZ is_activeÚrollbackÚclose)r' r5 r6 Útransr$ r$ r* Ú
|
connectionT s zTestBase.connectionc # sj g g d fdd }|V D ] }z | ¡ W q" Y q"X q" D ] }z | ¡ W qF Y qFX qFd S )NFc s | ¡ |r | ¡ d S r% )Úappend)ÚresultÚconsume©Úto_closeZ
|
to_consumer$ r* Úgon s
|
z/TestBase.close_result_when_finished.<locals>.go)F)Úallr; )r' rC Úrr$ rA r* Úclose_result_when_finishedi s z#TestBase.close_result_when_finishedc c s8 t |tt ¡ t d¡ddd¡id}|V | ¡ d S )Né2 ÚmysqlÚmariadbÚoracle)ÚmetadataÚtype_annotation_map)r ÚstrÚsaÚStringÚ with_variantÚdispose)r' rK Úregr$ r$ r* r s ÿÿþzTestBase.registryc # s, | G fdddt }|V |j ¡ d S )Nc s. e Zd Z Zee ¡ e d¡ddd¡iZdS )z TestBase.decl_base.<locals>.BaserG rH rI rJ N© Ú__name__Ú
|
__module__Ú __qualname__rK rM rN rO rP rL r$ ©Z_mdr$ r* ÚBase s ÿÿrX )r r rQ ©r' rK rX r$ rW r* Ú decl_base s zTestBase.decl_basec # s. | G fdddt t}|V |j ¡ d S )Nc s, e Zd Z Zee ¡ e d¡dd¡iZdS )z#TestBase.dc_decl_base.<locals>.BaserG rH rI NrS r$ rW r$ r* rX s ÿÿrX )r r r rQ rY r$ rW r* Ú dc_decl_base s zTestBase.dc_decl_basec c s
|
|V d S r% r$ )r' Ú future_enginer= r$ r$ r* Úfuture_connectionª s zTestBase.future_connectionc c s
|
d V d S r% r$ ©r' r$ r$ r* r\ ± s zTestBase.future_enginec # s0 ddl m d fdd }|V j d¡ d S )Nr ©ÚenginesFc s( |d kr i }d|d< j | ||||dS )NÚfixtureÚscope)ÚurlÚoptionsÚasyncioÚtransfer_staticpoolÚ
|
share_pool©Útesting_engine)rc rd Úfuturere rf rg r_ r$ r* Úgen_testing_engine¹ s ûz3TestBase.testing_engine.<locals>.gen_testing_enginera )NNNFFF)Ú r` Ztesting_reaperZ_drop_testing_engines)r' rk r$ r_ r* ri µ s úzTestBase.testing_enginec s fdd}|S )Nc s d| d< f | S )NTre r$ )Úkwrh r$ r* rC Ò s z)TestBase.async_testing_engine.<locals>.gor$ )r' ri rC r$ rh r* Úasync_testing_engineÐ s z TestBase.async_testing_enginec C s t S r% )Úfixture_sessionr^ r$ r$ r* ro Ø s zTestBase.fixture_sessionc c st ddl m} | ¡ }||j_|V |j`trdt ¡ rdt ¡ }| ¡ t ¡ t
|
|t W 5 Q R X n t
|
|t j dS )z>Provide bound MetaData for a single test, dropping afterwards.r r N) Úsqlr ÚMetaDataÚinstancerK r9 Zin_transactionZget_transactionr: r8 r r r- )r' Úrequestr rK r< r$ r$ r* rK Ü s" ÿþ
|
ÿzTestBase.metadatac C s* g | ]"}d D ]}dD ]}|||fqq qS ))ÚnoneÚexecuter8 ©TFr$ )Ú.0r: Úsecond_operationÚ begin_nestedr$ r$ r* Ú
|
<listcomp>õ s ýzTestBase.<listcomp>rv )Úparamsc sP |j \ td|tdtt| dd p.tj} |¡ fdd}|S )NÚtestÚdatar2 c sÜ | ¡ P} r@tjjjs&t d¡ |r4| ¡ }n| ¡ }|Æ |r\| ¡ ddi¡ n| ¡ ddi¡ r~| ¡ n|
|
¡ dkrt t jjd¡\ dkrÚ|rÄ| ¡ ddi¡ qø| ¡ ddi¡ n dkrø|rð| ¡ n| ¡ W 5 Q R X W 5 Q R X |r(| ¡ dd i¡ n| ¡ dd i¡ n|r\| ¡ ddi¡ n| ¡ ddi¡ |rr| ¡ n|
|
¡ nr | ¡ n|
|
¡ dkrVt t jjd¡ dkr |rê| ¡ ddi¡ n| ¡ ddi¡ nLdkr*t|dr | ¡ n| ¡ n"d
|
krL|rD| ¡ n| ¡ W 5 Q R X W 5 Q R X d } rr|d 7 }s|d 7 }|r¦t| tt ¡ ¡¡| n2| ¡ $}t| tt ¡ ¡¡| W 5 Q R X d S ) Nzsavepoints not enabledr} é
|
rt zCan't operate on closed transaction inside context manager. Please complete the context manager before emitting further commands.ru é r8 é ry r r )r8 r Ú requirementsÚ
|
savepointsÚenabledZ skip_testry ru Úinsertr: Úcommitr Zexpect_raises_messagerN ÚexcZInvalidRequestErrorÚhasattrr
|
Zscalarr r ÚcountÚ select_fromr4 )ÚsubjectZtrans_on_subjectZexecute_on_subjectr< Z nested_transZexpected_committedr6 ©ry r: rx Útr$ r* Úrun_test s
|
|
|
|
|
þ ÿ
|
|
|
|
|
þ
|
|
|
|
|
|
þ
|
þz4TestBase.trans_ctx_manager_fixture.<locals>.run_test)Úparamr r r r3 r r- Úcreate)r' rs rK r5 r r$ r r* Útrans_ctx_manager_fixtureô s
|
xz"TestBase.trans_ctx_manager_fixture)N)rT rU rV Ú __requires__Z__unsupported_on__Z __only_on__Z __skip_if__Z"__leave_connections_for_teardown__r+ r ra r0 r7 r= rF r rZ r[ r] r\ ri rn ro rK r r$ r$ r$ r* r# . sL
|
|
|
|
|
|
|
|
|
|
|
|
|
|
þÿ r# c @ s e Zd ZdZdS )ÚFutureEngineMixinz alembic's suite still using thisN)rT rU rV Ú__doc__r$ r$ r$ r* r
s r c @ s e Zd ZdZdZdZdZdZdZdZ dZ
|
dZ dZ dZ ejddddd Zejdd dd
|
d Zed d Zedd Zedd Zedd Zdd Zdd Zdd Zedd Zed d Z ed d Z ed d! Z ed"d# Z ed$d% Z d&d' Z!d(d) Z"ed*d+ Z#dS ),Ú
|
TablesTestÚonceÚeachNTÚclass©Úautouserb c c s0 | j }| ¡ | ¡ | ¡ d V | ¡ d S r% )Ú __class__Ú _init_classÚ_setup_once_tablesÚ_setup_once_insertsÚ _teardown_once_metadata_bind©r' Úclsr$ r$ r* Ú_setup_tables_test_class£ s z#TablesTest._setup_tables_test_classÚfunctionc c s" | ¡ | ¡ d V | ¡ d S r% )Ú_setup_each_tablesÚ_setup_each_insertsÚ_teardown_each_tablesr^ r$ r$ r* Ú_setup_tables_test_instance° s z&TablesTest._setup_tables_test_instancec C s | j S r% )Ú_tables_metadatar^ r$ r$ r* Útables_test_metadata¹ s z TablesTest.tables_test_metadatac C sX | j dkr(| jdkrd| _| jdks(tt | _t | _t | _| ¡ | _ t
|
¡ | _ d S )Nr r )r N) Úrun_define_tablesÚrun_create_tablesÚ run_insertsr& r ÚotherÚtablesÚ sequencesÚ
|
setup_bindr2 rN rq r§ ©r r$ r$ r* r ½ s
|
|
|
zTablesTest._init_classc C s6 | j dkr2| ¡ | j ¡ }| |¡ W 5 Q R X d S ©Nr ©r« Ú_load_fixturesr2 r8 Ú insert_data)r r6 r$ r$ r* r Ë s
|
z TablesTest._setup_once_insertsc C sR | j dkrN| | j¡ | jdkr.| j | j¡ | j | jj¡ | j | jj ¡ d S r± ©
|
r© Ú define_tablesr§ rª Ú
|
create_allr2 r Úupdater® Z
|
_sequencesr° r$ r$ r* r Ò s
|
|
z TablesTest._setup_once_tablesc C sl | j dkrP| | j¡ | jdkr.| j | j¡ | j | jj¡ | j | jj ¡ n| jdkrh| j | j¡ d S ©Nr rµ r^ r$ r$ r* r£ Û s
|
|
|
z TablesTest._setup_each_tablesc C s6 | j dkr2| ¡ | j ¡ }| |¡ W 5 Q R X d S r¹ r² )r' r6 r$ r$ r* r¤ å s
|
z TablesTest._setup_each_insertsc C sD | j dkr8| j ¡ | jdkr,t| j| j | j ¡ n| jdkrPt| j| j ttj dd}|rh|j
|
}| j dkr@| jdkr@| j dkr@| j ¡ ¤}t dd t| jj ¡ D D ]~}z:|rä| ¡ | | ¡ ¡ W 5 Q R X n| | ¡ ¡ W q¶ tjjk
|
r2 } ztd||f tjd W 5 d }~X Y q¶X q¶W 5 Q R X d S )Nr r Fc S s g | ]\}}|d k r|qS r% r$ )rw r Úfksr$ r$ r* rz s üz4TablesTest._teardown_each_tables.<locals>.<listcomp>zError emptying table %s: %r)Úfile)r© r Úclearrª r r§ r2 r3 r r r Ú run_deletesr8 Úreversedr Úvaluesry ru ÚdeleterN r Z
|
DBAPIErrorÚprintÚsysÚstderr)r' r r6 ÚtableÚexr$ r$ r* r¥ ë sD
|
|
|
|
ÿþý
|
ÿþÿ
|
|
þz TablesTest._teardown_each_tablesc C sF | j rt| j| j | jdkr*| | j¡ d | j_| jd k rBd | _d S r± )rª r r§ r2 Úrun_dispose_bindÚ dispose_bindÚrun_setup_bindr° r$ r$ r* r s
|
|
z'TablesTest._teardown_once_metadata_bindc C s t jS r% )r r- r° r$ r$ r* r¯ ! s zTablesTest.setup_bindc C s* t |dr| ¡ nt |dr&| ¡ d S )NrQ r; )r rQ r; )r r2 r$ r$ r* rÇ % s
|
|
|
zTablesTest.dispose_bindc C s d S r% r$ )r rK r$ r$ r* r¶ , s zTablesTest.define_tablesc C s i S r% r$ r° r$ r$ r* Úfixtures0 s zTablesTest.fixturesc C s d S r% r$ )r r= r$ r$ r* r´ 4 s zTablesTest.insert_datac C s | | j||¡ d S r% )Zassert_sql_countr2 )r' r Úfnr$ r$ r* Ú
|
sql_count_8 s zTablesTest.sql_count_c C s | | j||¡ d S r% )Z
|
assert_sqlr2 )r' Z callable_Z
|
statementsr$ r$ r* Úsql_eq_; s zTablesTest.sql_eq_c
|
sÊ i i }| ¡ ¡ D ]F\}t|dk r,qttr@| j |d < |dd
|< qt| jj ¡ D ]V\}dkrqn krqn| j
|
¡ *}| ¡ fdd| D ¡ W 5 Q R X qndS )z4Insert rows as represented by the fixtures() method.r r r Nc s g | ]}t t |qS r$ )ÚdictÚzip)rw Z column_values©ÚheadersrÄ r$ r* rz S s ÿz-TablesTest._load_fixtures.<locals>.<listcomp>) rÉ ÚitemsÚlenÚ
|
isinstancerM r r r§ r¿ r2 r8 ru r )r Úrowsr} rº r6 r$ rÏ r* r³ > s,
|
|
|
|
ÿ þþzTablesTest._load_fixtures)$rT rU rV rÈ r© rª r« r½ rÆ r2 r§ r r¬ r® r ra r¡ r¦ Úpropertyr¨ Ú classmethodr r r r£ r¤ r¥ r r¯ rÇ r¶ rÉ r´ rË rÌ r³ r$ r$ r$ r* r sT
|
|
|
|
|
|
|
)
|
|
|
|
|
|
r c @ s" e Zd Zejddddd ZdS )ÚNoCacheTr¢ r c c s" t jj}d t j_d V |t j_d S r% r, r. r$ r$ r* Ú_disable_cache[ s zNoCache._disable_cacheN)rT rU rV r ra rØ r$ r$ r$ r* r× Z s r× c @ s8 e Zd Zejdd Zdd Zejddddd Z d
|
S ) Ú RemovesEventsc C s t S r% )Úsetr^ r$ r$ r* Ú
|
_event_fnsd s zRemovesEvents._event_fnsc K s( | j |||f¡ tj|||f| d S r% )rÛ Úaddr Úlisten)r' ÚtargetÚnamerÊ rm r$ r$ r* Ú event_listenh s zRemovesEvents.event_listenTr¢ r c c s d V | j D ]}tj| q d S r% )rÛ r Úremove)r' Úkeyr$ r$ r* Ú_remove_eventsl s
|
z RemovesEvents._remove_eventsN)
|
rT rU rV r Zmemoized_propertyrÛ rà r ra rã r$ r$ r$ r* rÙ c s
|
|
rÙ c @ s e Zd Zejdddd ZdS )ÚRemoveORMEventsGloballyT)r c c s< d V t j ¡ t j ¡ t j ¡ t j ¡ t j ¡ d S r% )Ú
|
orm_eventsZ MapperEventsÚ_clearZInstanceEventsZ SessionEventsZInstrumentationEventsZ QueryEventsr^ r$ r$ r* Ú_remove_listenerst s
|
|
|
|
z)RemoveORMEventsGlobally._remove_listenersN)rT rU rV r ra rç r$ r$ r$ r* rä s s
|
rä c K sD | dd¡ | dd¡ | dtj¡}tjj|f| }t |¡ |S )NZ autoflushTZexpire_on_commitr2 ) Ú
|
setdefaultÚpopr r- rN ÚormÚSessionÚ_fixture_sessionsrÜ )rm r2 Úsessr$ r$ r* ro s
|
ro c C s t jj ¡ t ¡ d S r% )rN rê ÚsessionÚclose_all_sessionsrì r¼ r$ r$ r$ r* Ú_close_all_sessions s rð c C s t tj ¡ d S r% )rð rN rê Ú clear_mappersr° r$ r$ r* Ú stop_test_class_inside_fixtures s rò c C s t r
|
t d S r% )rì rð r$ r$ r$ r* Ú
|
after_test s ró c @ s e Zd ZdS )ÚORMTestN©rT rU rV r$ r$ r$ r* rô s rô c @ s e Zd ZU dZdZdZded< ejdddd d
|
Z ejdd dd d Z
|
e dd Z e dd Z e dd Zdd Zdd Ze dd Ze dd Zd d Zd d Ze d d! Ze d"d# ZdS )$Ú
|
MappedTestr r Nr ÚclassesTr r c c sZ | j }| ¡ |jd kr t |_| ¡ | ¡ | ¡ | ¡ d V | ¡ | ¡ d S r% )
|
r r r÷ r r Ú_setup_once_classesÚ_setup_once_mappersr Ú_teardown_once_classr r r$ r$ r* r¡ © s
|
z#MappedTest._setup_tables_test_classr¢ c c sN | ¡ | ¡ | ¡ | ¡ d V tjj ¡ | ¡ | ¡ |
|
¡ d S r% ) r£ Ú_setup_each_classesÚ_setup_each_mappersr¤ rN rê rî rï Ú_teardown_each_mappersÚ_teardown_each_classesr¥ r^ r$ r$ r* r¦ » s z&MappedTest._setup_tables_test_instancec C s | j ¡ d S r% )r÷ r¼ r° r$ r$ r* rú É s z MappedTest._teardown_once_classc C s | j dkr| | j¡ d S r± ©Úrun_setup_classesÚ_with_register_classesÚ setup_classesr° r$ r$ r* rø Í s
|
z MappedTest._setup_once_classesc C s* | j dkr&| ¡ \| _| _| | j¡ d S r± )Úrun_setup_mappersÚ_generate_registryÚmapper_registryÚmapperr Ú setup_mappersr° r$ r$ r* rù Ò s
|
z MappedTest._setup_once_mappersc C s8 | j dkr | ¡ \| j_| j_| j dkr4| | j¡ d S )Nr r )r r r r r r r r^ r$ r$ r* rü Ø s
|
ý
|
z MappedTest._setup_each_mappersc C s | j dkr| | j¡ d S r¹ rÿ r^ r$ r$ r* rû â s
|
z MappedTest._setup_each_classesc C s t | jd}||jfS )N)rK )r r§ Zmap_imperatively)r Údeclr$ r$ r* r æ s z MappedTest._generate_registryc sR | j G fddd}G dd dt|}G dd dt|}|| _|| _| dS )z¥Run a setup method, framing the operation with a Base class
|
that will catch new subclasses to be established within
|
the "classes" registry.
|
|
c s$ e Zd Zdd fdd Z ZS )z0MappedTest._with_register_classes.<locals>._BaseÚNone©Úreturnc s$ d k s t | | j< t ¡ d S r% ©r& rT ÚsuperÚ__init_subclass__r° ©r Ú cls_registryr$ r* r õ s
|
zBMappedTest._with_register_classes.<locals>._Base.__init_subclass__)rT rU rV r Ú __classcell__r$ ©r ©r r* Ú_Baseô s r c @ s e Zd ZdS )z0MappedTest._with_register_classes.<locals>.BasicNrõ r$ r$ r$ r* ÚBasicú s r c @ s e Zd ZdS )z5MappedTest._with_register_classes.<locals>.ComparableNrõ r$ r$ r$ r* Ú
|
Comparableý s r N)r÷ r r r r )r rÊ r r r r$ r r* r ë s z!MappedTest._with_register_classesc C s | j dkrtj ¡ d S r± )r rN rê rñ r^ r$ r$ r* rý s
|
z!MappedTest._teardown_each_mappersc C s | j dkr| j ¡ d S r± )r r÷ r¼ r^ r$ r$ r* rþ s
|
z!MappedTest._teardown_each_classesc C s d S r% r$ r° r$ r$ r* r s zMappedTest.setup_classesc C s d S r% r$ r° r$ r$ r* r s zMappedTest.setup_mappers)rT rU rV r r r÷ Ú__annotations__r ra r¡ r¦ rÖ rú rø rù rü rû r r rý rþ r r r$ r$ r$ r* rö s2
|
|
|
|
|
|
|
|
|
|
rö c s4 e Zd ZdZdZedd Ze fddZ ZS )ÚDeclarativeMappedTestr c C s d S r% r$ r° r$ r$ r* r s z(DeclarativeMappedTest._setup_once_tablesc sN j G fdddt}| _t |¡ jjrJ jrJ j t j
|
¡ d S )Nc sN e Zd ZejZjZee
|
¡ e
|
d¡ddd¡iZ dd fdd Z ZS ) z?DeclarativeMappedTest._with_register_classes.<locals>._DeclBaserG rH rI rJ r r
|
c s( d k s t | | j< t jf | d S r% r )r rm r r$ r* r - s
|
zQDeclarativeMappedTest._with_register_classes.<locals>._DeclBase.__init_subclass__)rT rU rV r r Z __table_cls__r§ rK rM rN rO rP rL r r r$ ©r r r r* Ú _DeclBase$ s ÿÿr ) r÷ r ZDeclarativeBasicr r r§ r rª r· r r- )r rÊ r r r r* r s z,DeclarativeMappedTest._with_register_classes) rT rU rV r r rÖ r r r r$ r$ r r* r s
|
r c @ s: e Zd Zd ZZdZdZe d¡Z dd Z
|
e dd Z dS ) Ú ComputedReflectionFixtureTestNT)Zcomputed_columnsZtable_reflectionz[\[\]\(\)\s`'\"]*c C s | j d|¡ ¡ S )Nrl )ÚregexpÚsubÚlower)r' Útextr$ r$ r* Ú normalizeD s z'ComputedReflectionFixtureTest.normalizec C sx ddl m} ddl m} ddlm} ddlm} ddlm} |d||d|d d
|
|d ||d ||d |d|dd |d||d|d d
|
|d ||d||d }|jjj rÜ|d||d|d d
|
|d ||d||dt
|
j d}|jj j r(| |d||ddd¡ |jjj r(| |d||ddd¡ |jjj rt| |d||dd d¡ |jjj rt| |d||d d d¡ d S ) Nr r )Útestingr )ÚComputedr Zcomputed_default_tableÚidT)Z primary_keyÚnormalZ computed_colz normal + 42Z with_defaultZ42)Zserver_defaultZcomputed_column_tableZcomputed_no_flagz normal / 42r Zcomputed_virtualz
|
normal + 2F)Z persistedz
|
normal / 2Zcomputed_storedz normal - 42z normal * 42)rl r r! r r r" r ÚrequiresZschemasr r Z test_schemaZcomputed_columns_virtualZ append_columnZcomputed_columns_stored) r rK r r! r r" r r Út2r$ r$ r* r¶ G sz ú û
|
ú
|
ýÿ
|
ýÿ
|
ýÿ
|
ýÿz+ComputedReflectionFixtureTest.define_tables) rT rU rV r« r½ Z __backend__r ÚreÚcompiler r rÖ r¶ r$ r$ r$ r* r < s
|
r c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚCacheKeyFixturec C s | ¡ }| ¡ }|d kr6|j d¡s(t|d kstnNt|j|j tt|jt|j t|j|jD ]\}}|j ||dshtqh||fS )Nr0 ©Úcompare_values)
|
Ú_generate_cache_keyÚ _annotationsÚgetr& r
|
râ ÚhashrÎ Ú
|
bindparamsÚcompare)r' ÚaÚbr+ Úa_keyÚb_keyÚa_paramÚb_paramr$ r$ r* Ú_compare_equal s z CacheKeyFixture._compare_equalc C sè | }| }t tt|d¡D ]À\}}||krX| || || |¡\}}|d krVq nº|| ¡ }|| ¡ }|d ks|d krº|d kr|| j d¡st|d kr || j d¡s tq |j |j krt
|
|j |j D ] \} }
|
| j |
|
|dsÖ qqÖt |j |j nt |j |j t|| tr t|| tr g } g } t || ¡D ] } | jdkrD| | ¡ qDt || ¡D ] } | jdkrp| | ¡ qptt|j dd dtt | ¡dd d tt|j d d dtt | ¡d
|
d d q d S ) Nr r0 r* Z bindparamc S s | j S r% ©râ ©r3 r$ r$ r* Ú<lambda>Ý ó z8CacheKeyFixture._run_cache_key_fixture.<locals>.<lambda>r9 c S s | j S r% r9 r: r$ r$ r* r; ß r< c S s | j S r% r9 r: r$ r$ r* r; ã r< c S s | j S r% r9 r: r$ r$ r* r; å r< )Ú itertoolsÚ combinations_with_replacementÚrangerÒ r8 r, r- r. r& râ rÎ r0 r1 r rÓ r" r! ZiterateZ__visit_name__r> r
|
Úsortedr Z unique_list)r' ra r+ Úcase_aÚcase_br2 r3 r4 r5 r6 r7 Zassert_a_paramsZassert_b_paramsÚelemr$ r$ r* Ú_run_cache_key_fixture sx
|
ÿ ÿ ÿ ÿ ÿ ÿþ ÿþz&CacheKeyFixture._run_cache_key_fixturec C sD | }| }t tt|d¡D ] \}}| || || |¡ q d S )Nr )r= r> r? rÒ r8 )r' ra r+ rA rB r2 r3 r$ r$ r* Ú _run_cache_key_equal_fixtureé s
|
ÿ z,CacheKeyFixture._run_cache_key_equal_fixtureN)rT rU rV r8 rD rE r$ r$ r$ r* r) s Kr) Fc sJ | j }|j| jG dd d fddfdd}|| _d S )Nc @ s e Zd ZdZdd Zdd ZdS )z.insertmanyvalues_fixture.<locals>.RandomCursor©Úcursorc S s
|
|| _ d S r% rF )r' rG r$ r$ r* Ú__init__þ s z7insertmanyvalues_fixture.<locals>.RandomCursor.__init__c S s | j ¡ }t|}t |¡ |S r% )rG ÚfetchallÚlistÚrandomÚshuffle)r' rÔ r$ r$ r* rI s
|
|
z7insertmanyvalues_fixture.<locals>.RandomCursor.fetchallN)rT rU rV Ú __slots__rH rI r$ r$ r$ r* Ú RandomCursorû s rN c 3 s@ r | } | ||||D ] }r4|j r4t d¡ |V q d S )Nz)Batches were downgraded for sorted INSERT)Z is_downgradedr Úwarn)rG Z statementÚ
|
parametersZgeneric_setinputsizesÚcontextÚbatch)rN Ú orig_dialectÚrandomize_rowsÚwarn_on_downgradedr$ r* Ú!_deliver_insertmanyvalues_batches s ÿ
|
|
zCinsertmanyvalues_fixture.<locals>._deliver_insertmanyvalues_batchesc
|
s4 t jj| d d | |W 5 Q R £ S Q R X d S )NrV )Únew)r ÚpatchÚobject)ÚdialectrQ )rV Ú orig_connr$ r* Ú_exec_insertmany_context s ýz:insertmanyvalues_fixture.<locals>._exec_insertmany_context)rZ rV r\ )r= rT rU rZ r\ r$ )rN rV r[ rS rT rU r* Úinsertmanyvalues_fixtureó s r] )FF)>Ú
|
__future__r r= rK r' r Útypingr Z
|
sqlalchemyrN rl r r r r r
|
r Úentitiesr r r r r r r r r r r r rê r r rå r r r rp r! Z sql.elementsr" Zmark_base_test_classr# r9 r r r× rÙ rä rÚ rì ro rð rò ró rô ZAssertsExecutionResultsrö r r r) r] r$ r$ r$ r* Ú<module>
|
sn U R x$Ph ÿ
|
