+h6ddlmZddlZddlZddlmZddlZddlZddlmZddl Z ddl m Z ddl m Z dd l m Z dd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZesdZ e xZ!Z"nejFZ!dZ"dZ$erd3dZ%nd3dZ%dZ&Gdde'Z(d Z)d!Z*d"Z+d#Z,d$Z-d%Z.ed&Z/d'Z0d(Z1d)Z2d4d*Z3d+Z4Gd,d-e5Z6d.Z7d5d/Z8d0Z9d1Z:d2Z;y)6)dequeN)chain) getsizeof)config)mock)inspect) Connection)Column)DropConstraint) DropTable)ForeignKeyConstraint)MetaData)Tableschema)Integer decorator) defaultdict)has_refcount_gc)inspect_getfullargspec)py2kcTtjtjyN)gccollect)argss J/var/www/html/venv/lib/python3.12/site-packages/sqlalchemy/testing/util.pynon_refcount_gc_collectr!&s  cyrr$r"r lazy_gcr%/s r"c#Kt}tr ddl}|j|ddl}|j||D]1t d|jD]jfdf3y#t$rYWwxYww)Nrc(j|Sr)dumps)dpickle_protocols r zpicklers..Ds7==H+Er") setrcPickleadd ImportErrorpicklerangeHIGHEST_PROTOCOLloads)picklersr/r2r+r,s @@r r6r63suH    LL ! LLFb&"9"9: FH--!EE E FF   s(BA7A B7 BBBBcLt|}tj||d|SNr)listrandomshuffle) populationkpops r random_choicesr?Is#:s1Qxr"c0tj||S)N)r=)r:choices)r<r=s r r?r?Rs~~jA..r"ct|tr t||S|tjdd|zzzj tj td|z S)N10 ) isinstancefloatrounddecimalDecimal to_integral ROUND_FLOORpow)valueprecs r round_decimalrPVs\%UD!! GOOC#*$45 5 B B  B  r"c0eZdZdZdZdZdZdZdZy) RandomSetc~ttj|}tj|t |Sr)r9r.__iter__r:r;iter)selfls r rTzRandomSet.__iter__as* d# $qAwr"ctjdt|dz }ttj ||}|j ||S)Nrr)r:randintlenr9r.rTremove)rVindexitems r r>z RandomSet.popfsCq#d)a-0CLL&'. D r"c@ttj||Sr)rRr.unionrVothers r r_zRandomSet.unionls4/00r"c@ttj||Sr)rRr. differencer`s r rczRandomSet.differenceose455r"c@ttj||Sr)rRr. intersectionr`s r rezRandomSet.intersectionrs))$677r"ct|Sr)rR)rVs r copyzRandomSet.copyus r"N) __name__ __module__ __qualname__rTr>r_rcrergr$r"r rRrR`s   168r"rRctt}|D]\}}||j|t|D]\}}||dD] }|||vs yy)zATrue if the given sorting conforms to the given partial ordering.NFT)rr.r0 enumerate)tuplessorted_elementsdepsparentchildinodens r conforms_partial_orderingruyst s D  V  _-4 $ AtAw  r"ctt|D]\}}|j|fdt|S)Nc3Kt|dk(rt|y|D]G}t|j|g}|j |r3|D] }|g|z Iyw)Nr)rZr9r.rcre)elementselemsubset sub_ordering_all_orderingsedgess r r|z-all_partial_orderings.._all_orderingssu x=A x.   4X114&9**5;7(6v(>4 #f|334 4s AA-A-)rr.r0rU)rmrxrprqr|r}s @@r all_partial_orderingsr~sI  E!  e  ! 4 x( ))r"c ||_|S#t$rEtj|j|j ||j |j}Y|SwxYw)acReturn a function with a given __name__. Will assign to __name__ and return the original function if possible on the Python implementation, otherwise a new function will be constructed. This function should be phased out as much as possible in favor of @decorator. Tests that "generate" many named tests should be modernized. )rh TypeErrortypes FunctionType__code__ __globals__ __defaults__ __closure__)fnnames r function_namedrsX  I     KKr  I  s A AAc|j} ||g|i|}|jddd|S#tj}|j|}|s|cYSxYw)aRun the given function under the given contextmanager, simulating the behavior of 'with' to support older Python versions. This is not necessary anymore as we have placed 2.6 as minimum Python version, however some tests are still using this structure. N) __enter____exit__sysexc_info)ctxrargkwobjresultrraise_s r run_as_contextmanagerrsj --/C C$#$$ T4& <<>x( Ms 3*Ac>|Dchc] }t|c}Scc}w)zConverts the results of sql execution into a plain set of column tuples. Useful for asserting the results of an unordered query. tuple)resultsrows r rowsetrs #* *3E#J ** *scJ|rr$)msgs r failrs #5r"cddlm}tj}|d}t |dd}||_ ||i||j |jrFt||j|jjj||_St|tj||_S#|j |jrFt||j|jjj||_wt|tj||_wxYw)ztProvide bound MetaData for a single test, dropping afterwards. Legacy; use the "metadata" pytest fixture. r)fixturesrmetadataN) rrrgetattrr_close_all_sessions_connection_fixture_connectiondrop_all_tables_from_metadataget_transactioncommitrdb)rrrrrrV prev_metas r provide_metadatars H 7Dj$/IDM"42 $$&  2 2 *(AA   3 3 C C E L L N"  *(FII >! % $$&  2 2 *(AA   3 3 C C E L L N"  *(FII >! s CBEc Jt}|D]|jt|}tj|Dcgc]2dj fd|Dft fd|Dz4c}ddt|zzdj |dScc}w)a A facade around @testing.combinations() oriented towards boolean keyword-based arguments. Basically generates a nice looking identifier based on the keywords and also sets up the argument names. E.g.:: @testing.flag_combinations( dict(lazy=False, passive=False), dict(lazy=True, passive=False), dict(lazy=False, passive=True), dict(lazy=False, passive=True, raiseload=True), ) would result in:: @testing.combinations( ('', False, False, False), ('lazy', True, False, False), ('lazy_passive', True, True, False), ('lazy_passive', True, True, True), id_='iaaa', argnames='lazy,passive,raiseload' ) _c3HK|]}j|ds|ywFNget.0r=r*s r z$flag_combinations..(s9Aq%a9s""c3BK|]}j|dywrrrs r rz$flag_combinations..)s2AEE!UO2srra,)id_argnames)r.updatesortedr combinationsjoinrrZ)rkeysr*s `r flag_combinationsrs< 5D  A $.create_fixture..fixture7s"(R(- -r"z fixture_%3.3d)rh)rrrs` r create_fixturez+lambda_combinations..create_fixture6s .+S0r")rrMockrrr3rZ)rrrrarg_setsrrrs` r lambda_combinationsr1sx !/ 2DQ @ @AH   (-c(m(< =1>!   = AC !A >s A6A;c t|d}|Dcic]}||j|}}t|j}|j |t j |j|}|di|Scc}w)zGiven a no-arg lambda and a namespace, return a new lambda that has all the values filled in. This is used so that we can have module-level fixtures that refer to instance-level variables using lambdas. rr$)rr>dictrrrrr)__fnrpos_argsr pass_pos_argsglbnew_fns r resolve_lambdarBsw&d+A.H19:#S"&&+%:M: t CJJrN    s 3F  "M "" ;sA<cfd}|S)z&Provide MetaData for a pytest fixture.cHfd}tj|S)Nc3$Ktjx}|_ ||}|jtj ||j tj y#|j tj wxYwwr)rrr create_allrrdrop_all)rVrrrs r run_ddlz3metadata_fixture..decorate..run_ddlWsh'-'8 8Ht} -D(+##FII. !!&)),!!&)),sB,A, B,!B  B)scope)rr)rrddls` r decoratez"metadata_fixture..decorateVs  -)v~~C(11r"r$)rrs` r metadata_fixturerSs 2 Or"c$tfd}|S)ziForce the given table names to be dropped after test complete, isolating for foreign key cycles c ||i|ttjttjS#ttjttjwxYw)N) include_names)drop_all_tablesrrr )rrrnamess r gozforce_drop_names..gonsG Pt?r? FIIwvyy'9 OOFIIwvyy'9 Os >5A3r)rrs` r force_drop_namesrhs PP Ir"c eZdZdZdZdZeZy)adictz,Dict keys available as attributes. Shadows.cX ||S#t$rtj||cYSwxYwr)KeyErrorr__getattribute__)rVkeys r rzadict.__getattribute__|s3 49  4((s3 3 4s ))cDt|Dcgc]}|| c}Scc}wrr)rVrrs r __call__zadict.__call__s40Cd3i0110s N)rhrirj__doc__rrget_allr$r"r rrys64 2Gr"rcddlmfd}t|ts"|j 5}||dddy||y#1swYyxYw)Nr)enginesc jj||jjs4ddlm}|j dd5j|dddyj|y#1swYyxYw)Nr) assertionszCan't sort tablesF)assert_)testing_reaperprepare_for_drop_tablesdialectsupports_alterrrexpect_warningsr) connectionrrrs r rz)drop_all_tables_from_metadata..gos{66zB!!00 $++#U, .!!*- . .   j )  . .s A99B)rrrFr begin)rengine_or_connectionrrrs` @r rrsR * *J 7 ! ' ' ) Z zN      s A  Ac ~| t|}|j5}t|j|D]\}}|r6|||vr|j t t |t|>|sA|jjsX|D]\}}|||vr t |ttdttdt|}|j tt|jjg|jj g| dddy#1swYyxYw)Nrxy)r)r.rreversedget_sorted_table_and_fkc_namesexecuterrrrrr rr rcrr) engine inspectorrrconntnamefkcsfkctbs r rrs' M* 4#  4 4F 4 C  KE4 ,m1K eE8:fEF~~44"&JE3%1!6  sG,sG,% BLL&0"$$&&BDDFF8#N sAD3;B.D33D<c$tfd}|S)Nc` ||i|jS#jwxYwr)_clear)rrr event_clss r rz!teardown_events..decorates, s>b>    I   s-r)r rs` r teardown_eventsr s Or"c d}ttttttt|t tt tit tdfd|S)zReturns the approximate memory footprint an object and all of its contents. source: https://code.activestate.com/recipes/577504/ cHtj|jSr)r from_iterableitems)r*s r dict_handlerz total_size..dict_handlers""1779--r"rc t|vryjt|t|}jD]2\}}t ||s|t t ||z }|S|Sr8)idr0rrrFsummap)ostyphandler all_handlers default_sizeseensizeofs r rztotal_size..sizeofs{ a5D= A a &(..0 LC!S!SVWQZ011  r")rrUr9rrr. frozensetr)rrrrrrs @@@@r total_sizersR. t d t l T4 L 5DQrFrr9)tupstacksentinel num_elementsrys r count_cache_key_tuplesr&shEExHL yy| 8  A L e $T hZ/%7  r")r)function)NN)< collectionsrrIr itertoolsrr:rrrrrrr rr rr r rrrrsql sql.sqltypesrutilrrrrrr! gc_collectr%rr6r?rPr.rRrur~rrrrrrrrrrrrrrr rr&r$r"r r.s   #)"")32JJ F(/2 *&(2+ )" )"X-`"#"*" D !.#L#Lr"