NEw/Qw d dl Z d dlZd dlZddlmZ ddlmZ ddlmZ ddlm Z ddl
m Z ddl
m Z dd l
mZ d
d lmZ d
d lm
Z
d
d
lmZ d
dlmZ d
dlmZ G d de Z G d de Z G d de Z G d de Z G d dee Z G d deeej Z G d d e Z G d d e Z dS ) N )
assertions)configschema)drop_all_tables) BasicEntity)ComparableEntity)adict )event)util)declarative_base)DeclarativeMeta)sort_tables_and_constraintsc b e Zd ZdZdZdZdZddZd Zd Z e
j d Z dS )TestBase Nc |s
J | d S Nr )selfvalmsgs o/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/sqlalchemy/testing/fixtures.pyassert_zTestBase.assert_- s Cs c R t | d r| d S d S )NsetUp)hasattrr r s r setupzTestBase.setup1 s/ 4 ! ! JJLLLLL r c R t | d r| d S d S )NtearDown)r r# r s r teardownzTestBase.teardown5 s/ 4 $ $ MMOOOOO r c # " K t j }| } |V | | d S # | | w xY wr )r dbconnectbeginrollbackclose)r conntranss r
connectionzTestBase.connection9 sz y ""
JJJ NN JJLLLLL
NN JJLLLLs A$ $*Br )
__name__
__module__ __qualname__ __requires____unsupported_on__ __only_on__ __skip_if__r r! r$ r fixturer- r r r r r s L K K V^ r r c h e Zd ZdZdZdZdZdZdZdZ dZ
dZ dZ e
d Ze
d Ze
d Ze
d Zd Zd Zd
Zd Zd Ze
d
Ze
d Ze
d Ze
d Ze
d Ze
d Z e
d Z d Z d Z e
d Z dS )
TablesTestonceeachNc ~ | | | d S r ) _init_class_setup_once_tables_setup_once_insertsclss r setup_classzTablesTest.setup_classm s< !!!!!r c & | j dk r | j dk rd| _ | j dv sJ t | _ t | _ | | _ t j | _
| j | j
_ d S )Nr9 r8 )r9 N) run_define_tablesrun_create_tables run_insertsr othertables
setup_bindbindsaMetaDatametadatar> s r r; zTablesTest._init_classu s F * *$..(.%?n4444GG WW
>>##{}} H r c | j dk r]| | j 5 }| | d d d d S # 1 swxY w Y d S d S Nr8 rD _load_fixturesrH r( insert_data)r? r+ s r r= z TablesTest._setup_once_inserts s ?f $ $ !!
&T%%%
&
&
&
&
&
&
&
&
&
&
&
&
&
&
&
&
&
& % $ A A #A c | j dk rj| | j | j dk r | j | j | j | j j d S d S rM rB
define_tablesrK rC
create_allrH rF updater> s r r< z TablesTest._setup_once_tables su F * * cl + + +$.. ''111 J cl 1 2 2 2 2 2 + *r c H | j dk rj| | j | j dk r | j | j | j | j j d S | j dk r!| j | j d S d S Nr9 rS r s r _setup_each_tablesz TablesTest._setup_each_tables s !V + + t} - - -%//
((333 K t} 3 4 4 4 4 4
#v
-
- M $ $TY / / / / / .
-r c | j dk r]| | j 5 }| | d d d d S # 1 swxY w Y d S d S rX rN )r r+ s r _setup_each_insertsz TablesTest._setup_each_inserts s v % % ! ! ! ""
'd &&&
'
'
'
'
'
'
'
'
'
'
'
'
'
'
'
'
'
' & %rQ c | j dk rX| j | j dk rt | j | j | j n%| j dk rt | j | j | j dk r| j dk r| j 5 }t d t | j j D D ]q} | |
+# t j j $ r0}t# j d|d|t& j Y d }~jd }~ww xY w d d d d S # 1 swxY w Y d S d S d S )Nr9 c g | ] \ }}||
S r r ).0tfkss r
z4TablesTest._teardown_each_tables.. s- $Q = )==r zError emptying table z: )file)rB rF clearrC r rK rH run_deletesr' reversedr valuesexecutedeleterI exc
DBAPIErrorr print_sysstderr)r r+ tableexs r _teardown_each_tablesz TablesTest._teardown_each_tables s !V + + K %//
ty999 M ! ! ! !
#v
-
- DM49 5 5 5 !V + +0@F0J0J""$$
% (C M 0 7 7 9 9) ) E U\\^^4444 6, >CeeRR H!$
, +0J0Js= 8AE;>'D&%E;&E*:&E% E;%E**E;;E?E?c V | | d S r )rY r[ r s r r! zTablesTest.setup s, !!! """""r c . | d S r )rp r s r r$ zTablesTest.teardown s ""$$$$$r c | j rt | j | j | j dk r| | j d | j _ | j d | _ d S d S rM )rC r rK rH run_dispose_bind dispose_bindrun_setup_bindr> s r _teardown_once_metadata_bindz'TablesTest._teardown_once_metadata_bind sk 4 CL#( 3 3 3 6 ) ) SX & & & )CHHH * )r c . | d S r )rw r> s r teardown_classzTablesTest.teardown_class s ((*****r c t j S r )r r& r> s r rG zTablesTest.setup_bind s
yr c t |d r| d S t |d r| d S d S )Ndisposer* )r r| r* )r? rH s r ru zTablesTest.dispose_bind sR 4 # # LLNNNNN
T7
#
# JJLLLLL r c d S r r )r? rK s r rT zTablesTest.define_tables r c i S r r r> s r fixtureszTablesTest.fixtures s r c d S r r )r? r- s r rP zTablesTest.insert_data r~ r c > | | j || d S r )assert_sql_countrH )r countfns r
sql_count_zTablesTest.sql_count_ s" diU33333r c > | | j || d S r )
assert_sqlrH )r callable_
statementss r sql_eq_zTablesTest.sql_eq_ s 9j99999r c i i c}| D ]X\ }t | dk rt t j r
| j |d < |dd |< Yt | j j D ]M\ }vr
| j
fd| D NdS )z4Insert rows as represented by the fixtures() method.r r r Nc V g | ]%}t t | &S r )dictzip)r^
column_valuesheadersrn s r ra z-TablesTest._load_fixtures.. s? % WU^];;<< r )
r itemslen
isinstancer string_typesrF r rK rf rH rg insert)r? rowsdatar` r rn s @@r rO zTablesTest._load_fixtures s7 B
<<>>//11 # #KE44yy1}}%!233
*
5) !!WGEN qrr(DKK5 L & & ( (
JE3 }G## H )-e
r )!r. r/ r0 rv rB rC rD rd rt rH rK rF rE classmethodr@ r; r= r< rY r[ rp r! r$ rw ry rG ru rT r rP r r rO r r r r7 r7 T s N K K DH
F E" " [" % % [ % & & [& 3 3 [30 0 0' ' ' :# # #% % %
[
+ + [+ [ [
[
[
[
4 4 4: : : [ r r7 c D e Zd Zej d Zd Z fdZ xZS )
RemovesEventsc t S r )setr s r
_event_fnszRemovesEvents._event_fns s uu r c h | j |||f t j |||fi | d S r )r addr
listen)r targetnamer kws r event_listenzRemovesEvents.event_listen sA VT2 .///
VT2,,,,,,,r c | j D ]}t j | t t | }t |d r| d S d S )Nr$ )r r
removesuperr r r$ )r keysuper_ __class__s r r$ zRemovesEvents.teardown se ? C L# }d++ 6: & & OO r ) r. r/ r0 r memoized_propertyr r r$
__classcell__r s @r r r sd - - - r r c $ e Zd Zed ZdS )_ORMTestc t j j t j d S r )rI ormsessionclose_all_sessions
clear_mappersr> s r ry z_ORMTest.teardown_class s2
))+++
r N)r. r/ r0 r ry r r r r r s- [ r r c e Zd ZdS )ORMTestNr. r/ r0 r r r r r & s Dr r c e Zd ZdZdZdZed Zed Zd Z d Z
ed Z ed Z ed
Z
d Zd Zed
Zd Zd Zed Zed ZdS )
MappedTestr8 r9 Nc | | j t | _ | | | | d S r )r; classesr r< _setup_once_classes_setup_once_mappersr= r> s r r@ zMappedTest.setup_class3 ss ; ''CK !!! !!! !!!!!r c V | | d S r )_teardown_once_classrw r> s r ry zMappedTest.teardown_class? s, """ ((*****r c | | | | d S r )rY _setup_each_classes_setup_each_mappersr[ r s r r! zMappedTest.setupD sP !!! """ """ """""r c t j j | | | d S r )rI r r r _teardown_each_mappers_teardown_each_classesrp r s r r$ zMappedTest.teardownJ sT
))+++ ##%%% ##%%% ""$$$$$r c j | j t d S r )r rc r ry r> s r r z MappedTest._teardown_once_classP s. !!!!!r c T | j dk r | | j d S d S rM run_setup_classes_with_register_classes
setup_classesr> s r r z MappedTest._setup_once_classesU 5 F * * & &s'8 9 9 9 9 9 + *r c T | j dk r | | j d S d S rM run_setup_mappersr
setup_mappersr> s r r z MappedTest._setup_once_mappersZ r r c T | j dk r | | j d S d S rX r r s r r z MappedTest._setup_each_mappers_ 5 !V + + ' '(: ; ; ; ; ; , +r c T | j dk r | | j d S d S rX r r s r r z MappedTest._setup_each_classesc r r c | j G fddt } G d dt j |t } G d dt
| } G d dt | }|| _ || _ | d S )
zRun a setup method, framing the operation with a Base class
that will catch new subclasses to be established within
the "classes" registry.
c e Zd Z fdZdS )6MappedTest._with_register_classes..FindFixturec L | |< t | ||| d S r )type__init__r? classnamebasesdict_ cls_registrys r r z?MappedTest._with_register_classes..FindFixture.__init__q s+ *- Y'
c9eU;;;;;r Nr. r/ r0 r r s r FindFixturer p s.
<
<
<
<
<
<
._BaseNr r r r _Baser u Dr r c e Zd ZdS )0MappedTest._with_register_classes..BasicNr r r r Basicr x r r r c e Zd ZdS )5MappedTest._with_register_classes..ComparableNr r r r
Comparabler { r r r N) r r r with_metaclassobjectr r
r r )r? r r r r r r s @r r z!MappedTest._with_register_classesg s { < < < < < < <$ < < <
D' V<< K )5 #
r c \ | j dk r t j d S d S rM )r rI r r r s r r z!MappedTest._teardown_each_mappers s4 !V + + F " " " " " , +r c R | j dk r| j d S d S rM )r r rc r s r r z!MappedTest._teardown_each_classes s2 !V + + L , +r c d S r r r> s r r zMappedTest.setup_classes r~ r c d S r r r> s r r zMappedTest.setup_mappers r~ r )r. r/ r0 r r r r r@ ry r! r$ r r r r r r r r r r r r r r r * sW G " " [ " + + [+# # # % % % " " [" : : [: : : [:< < << < <
[
4# # #! ! !
[
[
r r c L e Zd ZdZdZed Ze fd Z xZS )DeclarativeMappedTestr8 c d S r r r> s r r< z(DeclarativeMappedTest._setup_once_tables r~ r c j | j G fddt } G d dt }t | j || }|| _ t
t | | | j j r-| j
r(| j t j
d S d S d S )Nc e Zd Z fdZdS )LDeclarativeMappedTest._with_register_classes..FindFixtureDeclarativec @ | |< t j | ||| d S r )r r r s r r zUDeclarativeMappedTest._with_register_classes..FindFixtureDeclarative.__init__ s* *- Y' (iFFFFFr Nr r s r FindFixtureDeclarativer s5
G
G
G
G
G
G
Gr r c e Zd Zej ZdS )FDeclarativeMappedTest._with_register_classes..DeclarativeBasicN)r. r/ r0 r Table
__table_cls__r r r DeclarativeBasicr s "LMMMr r )rK metaclassr? )r r r r rK r r r r rF rC rU r r& )r? r r r _DeclBaser r s @r r z,DeclarativeMappedTest._with_register_classes s { G G G G G G G_ G G G
) ) ) ) )v ) ) ) %\,
) #S))@@DDD < /3#8 / L # #FI . . . . . / / / /r ) r. r/ r0 r r r r< r r r s @r r r si
[
/ / / / [/ / / / /r r c Z e Zd ZdxZZdZdZ ej d Z d Z
e d Z dS ) ComputedReflectionFixtureTestNT)computed_columnstable_reflectionz[\[\]\(\)\s`'\"]*c \ | j d| S )N )regexpsublower)r texts r normalizez'ComputedReflectionFixtureTest.normalize s$ {r4((..000r c
ddl m} ddl m} ddlm} ddlm} ddlm} |d| |d|d
|d | |d | |d
|d|d |d| |d|d
|d | |d| |d
}|j j j rE |d| |d|d
|d | |d| |d t j }|j j j rg|
|d| |dd |j j j r+|
|d| |dd |j j j ri|
|d| |dd |j j j r/|
|d| |d d d S d S d S ) Nr )Integer)testing)Column)Computed)r computed_default_tableidT) primary_keynormal computed_colz normal + 42 with_default42)server_defaultcomputed_column_tablecomputed_no_flagz normal / 42r computed_virtualz
normal + 2F) persistedz
normal / 2computed_storedz normal - 42z normal * 42)r r r r r r r requiresschemasenabledr test_schemacomputed_columns_virtual
append_columncomputed_columns_stored) r? rK r r r r r r_ t2s r rT z+ComputedReflectionFixtureTest.define_tables s ######%%%%%%""""""
$ F4d 3 3 3 F8W % % F>7HH],C,C D D F>74 @ @ @
E # F4d 3 3 3 F8W % % F%w 0G0G H H
# + 'tW$777x)))7HH]4K4KLL )
B 4 <
OO& H\U;;;
'/
F* ??? 3 ;
OO% H]d;;;
'/
F) $???
r )
r. r/ r0 rD rd __backend__r1 recompiler r r rT r r r r r si $$K+K;L
RZ,
-
-F1 1 1 A A [A A Ar r ) r( rl
sqlalchemyrI r r r r enginesr entitiesr r
r r r
ext.declarativer r r r r r7 r r r AssertsExecutionResultsr r r r r r r/ s~
$ $ $ $ $ $ ! ! ! ! ! ! & & & & & & . . . . . . - - - - - - 0 0 0 0 0 0# # # # #v # # #lw w w w w w w wt F " v h i
i
i
i
i
:z'I i
i
i
X / / / / /J / / /FM M M M MJ M M M M Mr