50T ` d dl Z d dlZd dlZd dlZd dlZd dlZddlmZ ddlmZ ddlm Z ddlm
Z
ddlm Z ddlm Z e
rd Z
e
xZZne rd
ZexZZn
ej Zd Zd Zd
Z G d de Zd Zd Zd Zd Zd Zd Zed Z d Z d Z d dZ d Z G d d e! Z"d!d Z#d Z$dS )" N decorator) defaultdict)inspect_getfullargspec)jython)py2k)pypyc t j t j d t j t j dS )z aggressive gc.collect for tests.g?r )gccollecttimesleepargss k/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/sqlalchemy/testing/util.pyjython_gc_collectr s5
3
q c R t j t j d S N)r r
r s r pypy_gc_collectr % s
r c d S r r r r lazy_gcr . s r c # K t } t r+ dd l}| | n# t $ r Y nw xY wdd l}| | | D ])t
d|j D ]j fdfV *d S )Nr c 0 | S r )dumps)dpickle_protocols r zpicklers..C s 7==H+E+E r ) setr cPickleadd ImportErrorpicklerangeHIGHEST_PROTOCOLloads)picklersr$ r' r r! s @@r r+ r+ 2 s uuH NNN LL ! ! ! ! D MMM LL F F b&"9:: F FH-!E!E!E!E!EE E E E E FF Fs 5
AAc t | t rt | | S | t j dd|z z z t j t d| z S )N10
)
isinstancefloatrounddecimalDecimal to_integral ROUND_FLOORpow)valueprecs r
round_decimalr: F sl % "UD!!!
GOC#*$455 5 B B B
r c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS ) RandomSetc t t | }t j | t | S r )listr# __iter__randomshuffleiter)selfls r r? zRandomSet.__iter__Q s6 d## $ $qAwwr c t j dt | dz
}t t | | }| | |S )Nr )r@ randintlenr> r# r? remove)rC indexitems r popz
RandomSet.popV sR q#d))a-00CLL&&''. D r c R t t | | S r )r< r# unionrC others r rN zRandomSet.union\ s 4//000r c R t t | | S r )r< r#
differencerO s r rR zRandomSet.difference_ s e44555r c R t t | | S r )r< r# intersectionrO s r rT zRandomSet.intersectionb s ))$66777r c t | S r )r< )rC s r copyzRandomSet.copye s r N) __name__
__module__ __qualname__r? rL rN rR rT rV r r r r< r< P sn
1 1 16 6 68 8 8 r r< c t t }| D ] \ }}|| | !t | D ] \ }}||d D ]}||| v r dS !dS )zATrue if the given sorting conforms to the given partial ordering.NFT)r r# r% enumerate)tuplessorted_elementsdepsparentchildinodens r conforms_partial_orderingrd i s s D
V _-- 4 $ AtAw uuu tr c t t | D ] \ }}| | !fdt | S )Nc 3
K t | dk rt | V d S | D ]V}t | |g }| | s | D ]
}|g|z V Wd S )NrF )rH r> r# rR rT )elementselemsubset sub_ordering_all_orderingsedgess r rk z-all_partial_orderings.._all_orderings| s x==A x..
4
4 X 114&99 **5;77 4(6v(>(> 4 4 #f| 33333
4
4r )r r# r% rB )r\ rg r_ r` rk rl s @@r all_partial_orderingsrm w sw E ! !
e 4 4 4 4 4 4 x(( ) ))r c || _ n<# t $ r/ t j | j | j || j | j } Y nw xY w| S )ac Return 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.
)rW TypeErrortypes FunctionType__code__ __globals__ __defaults__ __closure__)fnnames r function_namedrx s[
K r
Is
6AAc | } ||g|R i |}| ddd |S # t j } | j | }|s |cY S xY w)a Run 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)ctxrv argkwobjresultr} raise_s r run_as_contextmanagerr s --//C
C$#$$$$$ T4&&&
<>> x( MMMs $; $A!c d | D S )zConverts the results of sql execution into a plain set of column tuples.
Useful for asserting the results of an unordered query.
c , h | ]}t | S r tuple).0rows r zrowset.. s * * *3E#JJ * * *r r )resultss r rowsetr s + *' * * **r c J | r r )msgs r failr s #5r c > ddl m} ddl m} ddlm} | |j }|d }t |dd }||_ | |i || ||j ||_ S # | ||j ||_ w xY w)z>Provide bound MetaData for a single test, dropping afterwards.rF config)enginesr schemametadataN)
r r
sqlalchemyr MetaDatadbgetattrr drop_all_tables) rv r r r r r r rC prev_metas r provide_metadatar s !!!!!!vy))H 7Dj$//I DM"r42 &)444!
&)444!
!!!!s A8 8$B c ddl m} t | D ]} | t |j fd| D ddt
z z d dS )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'
)
rF r c g | ]@d fdD ft fdD z AS )_c 3 H K | ] } |d |V dS FNgetr kr s r z/flag_combinations... s4 99Aq% 9a999999r c 3 D K | ]} |d V dS r r r s r r z/flag_combinations... s/ 22AEE!UOO222222r )joinr )r r keyss @r
z%flag_combinations.. su
XX9999999
9
9 ;2222T22222
3
r ra a,)id_argnames)r r r# updatesorted combinationsrH r )r r r r s @r flag_combinationsr s < 55D
A $<z"resolve_lambda.. s% :::#S"&&++:::r r )r dictrs r rp rq rr )__fnr pos_args
pass_pos_argsglbnew_fns ` r resolve_lambdar sy &d++A.H:::::::M
t
CJJrNNN
s
3
3F 6 " "M " ""r functionc " ddl m fd}|S )z&Provide MetaData for a pytest fixture.rF r c P fd} | S )Nc 3 K ddl m} | x}| _ | | }| j |V | j d S # | j w xY w)Nr r )r r r r
create_allr drop_all)rC r r r r rv s r run_ddlz3metadata_fixture..decorate..run_ddl s ) ) ) ) ) )'-'8'8 8Ht}
-D(++##FI... !!&),,,,,!!&),,,,s *A, , B)scope)fixture)rv r r ddls ` r decoratez"metadata_fixture..decorate sB - - - - - - )v~~C~((111r )r r )r r r s ` @r metadata_fixturer s= 2 2 2 2 2 2 Or c L ddl m ddlm t fd }|S )ziForce the given table names to be dropped after test complete,
isolating for foreign key cycles
rF r r )inspectc | |i |t j j S # t j j w xY w)N)
include_names)r r )rv r r r r namess r goz force_drop_names..go: sg P2t?r?? FIwwvy'9'9 O O O O OOFIwwvy'9'9 O O O O Os 0 'A)r r r r r )r r r r s ` @@r force_drop_namesr 2 sf
""""""P P P P P P YP Ir c " e Zd ZdZd Zd ZeZdS )adictz,Dict keys available as attributes. Shadows.c j | | S # t $ r t | | cY S w xY wr )KeyErrorr __getattribute__)rC keys r r zadict.__getattribute__H sH 49 4 4 4((s33 3 3 3 4s
%22c : t fd|D S )Nc g | ]
}| S r r )r r rC s r r z"adict.__call__..O s 000Cd3i000r r )rC r s ` r __call__zadict.__call__N s& 00004000111r N)rW rX rY __doc__r r get_allr r r r r E s9 664 4 4 2 2 2 GGGr r c ddl m}m}m}m}m} ddlm} m}
|t | }|
5 } t | | D ]\ } }
| r:|| |vr|
| || | | A|
r| j j sP|
D ]x\ } }|| |vr || | |d| |d| | }|
|
||j j g|j j g| y d d d d S # 1 swxY w Y d S )Nr )ColumnTableIntegerr ForeignKeyConstraint) DropTableDropConstraintr xy)rw )r r r r r r sqlalchemy.schemar r r# connectreversed get_sorted_table_and_fkc_namesexecutedialectsupports_altercr r )engine inspectorr r r r r r r r r conntnamefkcsfkctbs r r r T sJ <;;;;;;; M**
T# 4 4F 4 C C
KE4
,m1K1K IeeE88::f E E EFF
~4 "& JE3%1 ! 6 6
sG,, sG,, % B LL& 0 0"$&BDF8# N N N 3 s C0D::D>D>c , t fd }|S )Nc p | |i | S # w xY wr )_clear)rv r r event_clss r r z!teardown_events..decorate sD 2s>b>> I s 5r )r r s ` r teardown_eventsr s- Y Or )r )NN)%r3 r r@ r| r rp utilr r r r r r
r
gc_collectr r r
r+ r: r# r< rd rm rx r r r r r r r r r r r r r r r r sg
) ) ) ) ) )
-,J
+*J J
F F F( 2 * * *& ( 2+ + + " " "$/ / /d# # #" 0 & D + + + +\ r