L ia0ddlmZddlZddlmZddlmZddlmZddlm Z ddlm Z d d lm Z d d lm Z d d lm Z d d lmZd dlmZd dlmZd dlmZd dlmZej(GddZdaGddZy)) annotationsN) assertions)config)eq_)drop_all_tables_from_metadata)picklers)Column)func)Integer)select)Table)DeclarativeBase)MappedAsDataclass)registryc eZdZdZdZdZdZdZddZe jdZ e jdZ e jdZ e jdZe jd Ze jd Ze jd Ze jd Ze jd Ze jdZe jdZe jedZe jdZe jdDcgc]}dD]}dD]}|||f c}}}}dZycc}}}}w)TestBaseNFc|sJ|yNr)selfvalmsgs f/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/sqlalchemy/testing/fixtures/base.pyassert_zTestBase.assert_1s Csc#Ktjj}dtj_d|tj_ywr)rdb_compiled_cache)r_caches rnocachezTestBase.nocache4s0**$( ! $* !sA A c#Kt|ddxstj}|j5}|dddy#1swYyxYwwNbind)getattrrrconnect)rengconns rconnection_no_transzTestBase.connection_no_trans;sAdFD)6VYY [[] dJ   s0A A A A A c#Kt|ddxstj}|j}|j }|a|da|j r|j|jywr$) r&rrr'begin_connection_fixture_connection is_activerollbackclose)rr(r)transs r connectionzTestBase.connectionBsbdFD)6VYY{{} )-& )-& ?? NN  sA5A7c#Kggdfd }|D]} |jD]} |jy#Y3xYw#Y"xYww)NcPj||rj|yyr)append)resultconsumeto_close to_consumes rgoz/TestBase.close_result_when_finished..go\s% OOF #!!&)r)F)allr0)rr:rr8r9s @@rclose_result_when_finishedz#TestBase.close_result_when_finishedWsk  *  A    A     s2AAAA AA A AAc #Kt|ttjj tjddddi}||j yw)N2mysqlmariadboracle)metadatatype_annotation_map)rstrsaString with_variantdispose)rrCregs rrzTestBase.registrymsORYY[--IIbM7Ix!   sA A"c#pK|Gfddt}||jjyw)Nc~eZdZWZeej jej ddddiZy) TestBase.decl_base..Baser?r@rArBN __name__ __module__ __qualname__rCrErFrGrHrD_mdsrBaserM~s;HYRYY[--BIIbM7Ix# rrT)rrrIrrCrTrSs @r decl_basezTestBase.decl_basezs/ ?   36c#zK|Gfddtt}||jjyw)Nc|eZdZWZeej jej dddiZy)#TestBase.dc_decl_base..Baser?r@rANrNrRsrrTrZs9HYRYY[--BIIbM7I# rrT)rrrrIrUs @r dc_decl_basezTestBase.dc_decl_bases2 $o   s8;c#K|ywrr)r future_enginer2s rfuture_connectionzTestBase.future_connections c#Kdywrr)rs rr]zTestBase.future_engines  r_c#pKddlm dfd }|jjdyw)Nr)enginescB|i}d|d<j|||||S)Nfixturescope)urloptionsasynciotransfer_staticpool share_pool)testing_engine)rfrgfuturerhrirjrbs rgen_testing_enginez3TestBase.testing_engine..gen_testing_engines@(GG ))$7% * rrd)NNNFFF)rbtesting_reaper_drop_testing_engines)rrmrbs @rrkzTestBase.testing_engines? %  &! 44Y?rWcfd}|S)Ncd|d<di|S)NTrhrr)kwrks rr:z)TestBase.async_testing_engine..gos ByM!'B' 'rr)rrkr:s ` rasync_testing_enginezTestBase.async_testing_engines ( r)paramsc#(K|jywr)param)rrequests rr zTestBase.picklerssmmsc#Kddlm}|j}||j_||j`t rft j rRt j}|jt j5t|t dddyt|tjy#1swYyxYww)z>Provide bound MetaData for a single test, dropping afterwards.r )schemaN) sqlrzMetaDatainstancerCr-in_transactionget_transactionr/r,rrr)rrxrzrCr1s rrCzTestBase.metadatas "??$$,!    % +.==?2BBDE NN /557 -<   *(FII >   sB C  C$C C C )TF)noneexecuter,c|j\td|tdtt |ddxst j }j|fd}|S)Ntestdatar%c |j5}rtjjjstj d|r|j }n|j }|5|r#|j jddin"|j jddir|jn|j dk7rtjtjjd5 dk(rH|r#|j jddinK|j jddin( dk(r#|r|j n|j dddddd|r$|j jdd in|j jdd int|r#|j jddin"|j jddi|r$r|jn4|jn#r|jn|j dk7rtjtjjd5 dk(rH|r#|j jddin~|j jddin[ dk(r.t!|dr|jn9|jn( d k(r#|r|j n|j ddddddd }r|d z }s|d z }|rFt#|j%t't)j*j- |y|j/5}t#|j%t't)j*j- |dddy#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYyxYw) Nzsavepoints not enabledr rzCan't operate on closed transaction inside context manager. Please complete the context manager before emitting further commands.r r, begin_nestedr)r,r requirements savepointsenabled skip_testrrinsertr/commitrexpect_raises_messagerFexcInvalidRequestErrorhasattrrscalarrr count select_fromr') subjecttrans_on_subjectexecute_on_subjectr1 nested_transexpected_committedr)rr/second_operationts rrun_testz4TestBase.trans_ctx_manager_fixture..run_tests^ 9E!..99AA(()AB)'.';';'= ','9'9'; %$=-#OOAHHJ E!MM!((*vrlC$(113(//1+v5!+!A!A " : :!D"=$4y#@'9(/,-HHJ )*). ahhj62,(O%5%@'9(/(<(<(>(-(:(:(<'=#$=R* VRLA ahhj62,?* VRLA ahhj62,?'##,,.#NN,#!NN,!LLN'61'==FF66@9 09<#5$+OOAHHJ $M$)MM!((*vrl$K!1W!<#*5'#:$)KKM$+MMO!1^!C#5$+$8$8$:$)$6$6$8+9S^ 9@"# #a'"#a'"!NN6$**,#7#C#CA#FG& __&$ F4::<$8$D$DQ$GH*m==#$=$=@99S^ 9^ 9`seA"P;4BP"A6PP"DP;%B)P/P;AQP P""P, 'P;/P8 4P;;QQ)rwrr r r&rrcreate) rrxrCr(rrr/rrs @@@@rtrans_ctx_manager_fixturez"TestBase.trans_ctx_manager_fixtures[4;==0"L &(F67$; <dFD)6VYY  u nrr)rOrPrQ __requires____unsupported_on__ __only_on__ __skip_if__"__leave_connections_for_teardown__rrrdr"r*r2r=rrVr[r^r]rkrtr rCr).0r/rrs0000rrrsLKK */&V^^++ V^^ V^^(V^^*V^^   ^^     ^^    V^^ V^^V^^@@4V^^V^^8:&'V^^??.V^^*  $@ !!  ' 6 6 6    sE:rceZdZdZy)FutureEngineMixinz alembic's suite still using thisN)rOrPrQ__doc__rrrrrrs*rr) __future__r sqlalchemyrFrnrrrutilrr r r r rrormrrrmark_base_test_classrr-rrrrrsn#0"$OOOd "&++r