S8( d dl mZ d dl mZ d dl mZ d dl mZ d dl mZ d dl mZ ddlmZ d d
l m
Z
d d l m Z d d l m Z d d
l m
Z
d dl mZ G d dee Z G d dee Z G d dee ZdS ) )Column)ForeignKeyConstraint)Integer)MetaData)String)Table )AutogenFixtureTest ) combinations)configeq_)mock)TestBasec p e Zd ZdZdZd Zd Zd Zd Zd Z d Z
e j j
d Zd
Zd Zd S )
AutogenerateForeignKeysTestT)!foreign_key_constraint_reflectionc
t }t }t d|t dt d d t d|t dt d t dt d d
t d t d d
t dt d t dgdg t d|t dt d d t d|t dt d t dt d d
t d t d d
t dt d | || }| |d dddgddgd d S )N
some_tabletest
T primary_keyuseridname2 Fnullablea1xserver_defaulttest2some_table.testr remove_fkservergeneratedconditional_namer r r r r r _fixture_assert_fk_diffselfm1m2diffss x/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/alembic/testing/suite/test_autogen_fks.pytest_remove_fkz*AutogenerateForeignKeysTest.test_remove_fk s
ZZ
ZZ
66"::4 8 8 8
4d 3 3 3 66":: 6 6 6 4C 8 8 8 7F2JJ ' ' '->,? @ @
66"::4 8 8 8
4d 3 3 3 66":: 6 6 6 4C 8 8 8 7F2JJ ' '
b"%% !H
I
H .
c
t }t }t d|t dt d t dt d t d|t dt d t dt d d
t d t d d
t dt d t d|t dt d t dt d t d|t dt d t dt d d
t d t d d
t dt d t dgdg | || }| |d dddgddg d S )Nr r Tr r r r r r Fr r! r" r# r% r& r add_fk)r r r r r r r, r- r. s r3 test_add_fkz'AutogenerateForeignKeysTest.test_add_fkC s
ZZ
ZZ
4d 3 3 3 66":: & &
4d 3 3 3 66":: 6 6 6 4C 8 8 8 7F2JJ ' '
4d 3 3 3 66":: & &
4d 3 3 3 66":: 6 6 6 4C 8 8 8 7F2JJ ' ' '->,? @ @
b"%% !Hh <&
r5 c t }t }t d|t dt d t dt d t d|t dt d t dt d d
t d t d d
t dt t dgdg t d|t dt d t dt d t d|t dt d t dt d d
t d t d d
t dt t dgdg | || }t |g d S )Nr r Tr r r r r r Fr r! r" r# r%
some_table.id)r r r r r r r, r r. s r3 test_no_changez*AutogenerateForeignKeysTest.test_no_changen s
ZZ
ZZ
4d 3 3 3 66":: & &
4d 3 3 3 66":: 6 6 6 4C 8 8 8 7G $ $ '_,= > >
4d 3 3 3 66":: & &
4d 3 3 3 66":: 6 6 6 4C 8 8 8 7G $ $ '_,= > >
b"%% E2r5 c n t }t }t d|t dt d d t dt d d t d|t dt d t d t d
d t d
t d d t dt d t dt d t ddgddg t d|t dt d d t dt d d t d|t dt d t d t d
d t d
t d d t dt d t dt d t ddgddg | || }t |g d S )Nr id_1r Tr id_2r r r r Fr r! r" r#
other_id_1
other_id_2some_table.id_1some_table.id_2r r r r r r r, r r. s r3 test_no_change_composite_fkz7AutogenerateForeignKeysTest.test_no_change_composite_fk s
ZZ
ZZ
66"::4 8 8 8 66"::4 8 8 8
4d 3 3 3 66":: 6 6 6 4C 8 8 8 < , , < , , |,"$56
66"::4 8 8 8 66"::4 8 8 8
4d 3 3 3 66":: 6 6 6 4C 8 8 8 < , , < , , |,"$56
b"%% E2r5 c
& t }t }t d|t dt d d t d|t dt d t dt d d
t d t d d
t dt d t dgdgd t d|t dt d d t d|t dt d t dt d d
t d t d d
t dt d t d gdgd | || }| |d dddgddgt j j
j rdnd | |d ddd gddgd d S )Nr r r Tr r r r r Fr r! r" r# r% r& MyFKr myfkr r' r r7 ) r r r r r r r, r- r
requirementsfk_namesenabledr. s r3 1test_casing_convention_changed_so_put_drops_firstzMAutogenerateForeignKeysTest.test_casing_convention_changed_so_put_drops_first s+
ZZ
ZZ
66"::4 8 8 8
4d 3 3 3 66":: 6 6 6 4C 8 8 8 7F2JJ ' ' '->,?f M M M
66"::4 8 8 8
4d 3 3 3 66":: 6 6 6 4C 8 8 8 7F2JJ ' ' $*;)<6 J J J
b"%% !H
I
H!.7?IT
!H
F
H
r5 c x t }t }t d|t dt d d t dt d d t d|t dt d t d t d
d t d
t d d t dt d t dt d t d|t dt d d t dt d d t d|t dt d t d t d
d t d
t d d t dt d t dt d t ddgddgd | || }| |d ddddgdddgd d S )Nr r= r Tr r> r r r r Fr r! r" r# r? r@ rA rB fk_test_namerG r r7 r+ r. s r3 test_add_composite_fk_with_namez;AutogenerateForeignKeysTest.test_add_composite_fk_with_name s/
ZZ
ZZ
66"::4 8 8 8 66"::4 8 8 8
4d 3 3 3 66":: 6 6 6 4C 8 8 8 < , , < , ,
66"::4 8 8 8 66"::4 8 8 8
4d 3 3 3 66":: 6 6 6 4C 8 8 8 < , , < , , |,"$56#
b"%% !H
< (
V
r5 c x t }t }t d|t dt d d t dt d d t d|t dt d t d t d
d t d
t d d t dt d t dt d t ddgddgd t d|t dt d d t dt d d t d|t dt d t d t d
d t d
t d d t dt d t dt d | || }| |d ddddgdddgd d S )Nr r= r Tr r> r r r r Fr r! r" r# r? r@ rA rB rN rG r r' r) r+ r. s r3 test_remove_composite_fkz4AutogenerateForeignKeysTest.test_remove_composite_fk= s/
ZZ
ZZ
66"::4 8 8 8 66"::4 8 8 8
4d 3 3 3 66":: 6 6 6 4C 8 8 8 < , , < , , |,"$56#
66"::4 8 8 8 66"::4 8 8 8
4d 3 3 3 66":: 6 6 6 4C 8 8 8 < , , < , ,
b"%% !H
< (
V +
r5 c t }t }t d|t dt d d t dt d d t d|t dt d t d t d t d
t d t d|t dt d d d t dt d d
d t d|t dt d t d t d d t d
t d d t ddgddgd | || }| |d ddd d
gdddgd d S )Nr r= r Tr r> r r r? r@ tid1keyr tid2oid1rU oid2some_table.tid1some_table.tid2rN rG r r7 r+ r. s r3 test_add_fk_colkeysz/AutogenerateForeignKeysTest.test_add_fk_colkeysu s
ZZ
ZZ
66"::4 8 8 8 66"::4 8 8 8
4d 3 3 3 < , , < , ,
66"::6t D D D 66"::6t D D D
4d 3 3 3 < 8 8 8 < 8 8 8 "$56#
b"%% !H
< (
V
r5 c t }t }t d|t dt d d t dt d d t d|t dt d t d t d t d
t d t d d
gd d g t d|t dt d d
d t dt d dd t d|t dt d t d t d d t d
t d d t ddgddg | || }t |g d S )Nr r= r Tr r> r r r? r@ rA rB rS rT rV rW rX rY rZ r[ rC r. s r3 test_no_change_colkeysz2AutogenerateForeignKeysTest.test_no_change_colkeys s
ZZ
ZZ
66"::4 8 8 8 66"::4 8 8 8
4d 3 3 3 < , , < , , |,"$56
66"::6t D D D 66"::6t D D D
4d 3 3 3 < 8 8 8 < 8 8 8 #46G"H
b"%% E2r5 N)__name__
__module__ __qualname__ __backend__ __requires__r4 r8 r; rD rL rO r
rI no_name_normalizerQ r\ r^ r5 r3 r r s K9L-
-
-
^)
)
)
V( ( (T0 0 0d;
;
;
z4
4
4
l *5
5
+*5
n1
1
1
f+ + + + +r5 r c e Zd ZdZdZ edd ej j d Z d Z
edd ej j d Z dS ) IncludeHooksTestT)rJ )objectrG c t }t }t d|t dt d }t d|t dt t dt }| t |j j g|j j gd | t |j j g|j j gd
t d|t dt d }t d|t dt t dt |d k r d }|
|||
}n!|dk rd }|
||| }| |d dddgddgd
t t | d d S )Nrefr Tr tr" yfk1rG fk2rh c J t | t o
|dk o|o|dk S Nforeign_key_constraintrm
isinstancer object_r type_ reflected
compare_tos r3 include_objectzBIncludeHooksTest.test_remove_connection_fk..include_object s> w(<== & !99&!&
r5 object_filtersr c n | dk r.|dk rdS t |d t |d ddd dS dS )Nrm indexTrq rk schema_name
table_nameschema_qualified_table_nameFr r rv parent_namess r3 include_namez@IncludeHooksTest.test_remove_connection_fk..include_name s` 5== '' #t 7888$+/*-;> !5 4r5 name_filtersr r' r) r r r r r append_constraintr cr" r rl r, r- r len) r/ hook_typer0 r1 rj t1ry r2 r s r3 test_remove_connection_fkz*IncludeHooksTest.test_remove_connection_fk s ZZ
ZZ 4d 3 3 3
3 3
"$&CEH:E B B B
"$&CEH:E B B B
4d 3 3 3
3 3
MM"b MHHEE
&
" MM"b|MDDE !H
E
F "
CJJr5 c Z t }t }t d|t dt d t d|t dt t dt t d|t dt d }t d|t dt t dt }| t |j j g|j j gd | t |j j g|j j gd
d }|
||| }| |d
dddgddgd
t t | d d S )Nrj r Tr rk r" rl rm rG rn c L t | t o|dk o| o|dk S rp rr rt s r3 ry z=IncludeHooksTest.test_add_metadata_fk..include_objectK s@ 7$899 " 55"!M" EM
r5 rz r r7 r r )r/ r0 r1 rj t2ry r2 s r3 test_add_metadata_fkz%IncludeHooksTest.test_add_metadata_fk) s
ZZ
ZZ
4d 3 3 3
3 3
4d 3 3 3
3 3
"$&CEH:E B B B
"$&CEH:E B B B
b"^
DD !HhcUED6
CJJr5 c
t }t }t d|t dt d }t d|t dt d t dt d t d|t dt t d t t d
t }| t |j j g|j j gd | t |j j g|j j gd
t d|t dt d t d|t dt d t dt d }t d|t dt t d t t d
t }| t |j j |j j
g|j j |j j gd | t |j j |j j
g|j j |j j gd
|dk r d }| ||| } n!|dk rd }
| |||
} |dk ri|
| d ddd gddgd
|
| d ddd d
gdddgd
t t | d d S |dk r t d | D h d d S d S )Nref_aaTr ref_bbrk r" rl zrm rG rn rh c F t | t o |dk o|dk S rp rr rt s r3 ry z7IncludeHooksTest.test_change_fk..include_object s4 w(<== & !99&
r5 rz r c n |dk rdS | dk r&t |d t |d ddd dS dS )Nr} Trm rq rk r~ Fr r s r3 r z5IncludeHooksTest.test_change_fk..include_name s` G## 45== 7888$+/*-;> !5 4r5 r r r' r r7 c 8 h | ]}|d |d j fS )r r rG ).0ds r3 z2IncludeHooksTest.test_change_fk.. s' 222q!A$! "222r5 > r7 rm r7 rn r' rn )r r r r r r r r" r rl r r r, r- r r ) r/ r r0 r1 r1ar r2br ry r2 r s r3 test_change_fkz IncludeHooksTest.test_change_fkZ s ZZ
ZZ 3T 2 2 2
3T 2 2 2 3T 2 2 2
3 3 3
"$&CEG95 A A A
"$&CEG95 A A A
3T 2 2 2
3T 2 2 2 3T 2 2 2
3 3 3
357CEG"45
357CEG"45
MM"b MHHEE
&
$ MM"b|MDDE a+sSE7SE
!
ac
c
!
E
A
&
22E222LLL
!
r5 N) r_ r` ra rb rc r r
rI rd r r r re r5 r3 rg rg s K L\+y)) *K K +* *)KZ/ / /b \+y)) *l l +* *)l l lr5 rg c h e Zd ZdZd Zej j d Zej j d Z d Z
ej j d Z ej j d Z
ej j d Zej j d Zej j d
Zej j d Zej j d Zej j d
Zej j d Zej j ej j d Zej j d Zej j d Z ej j ej j d Z ej j ej j d Z ej j ej j d Z ej j ej j d Z!ej j ej j d Z"ej j d Z#ej j$ d Z%dS )AutogenerateFKOptionsTestTc
t }t }t d|t dt d t dt d t d|t dt d t dt d d
t d t t d gd
gfi | t d|t dt d t dt d t d|t dt d t dt d d
t d t t d gd
gfi | | || S )Nr r Tr r r r r r Fr tidr: )r r r r r r r, )r/ old_optsnew_optsr0 r1 s r3 _fk_opts_fixturez*AutogenerateFKOptionsTest._fk_opts_fixture s|
ZZ
ZZ
4d 3 3 3 66":: & &
4d 3 3 3 66":: 6 6 6 5' " " %?*; H Hx H H
4d 3 3 3 66":: & &
4d 3 3 3 66":: 6 6 6 5' " " %?*; H Hx H H
}}R$$$r5 c
| i ddi }| |d dddgddgd d
| |d d ddgddgd
d S )Nondeletecascader r' r r r r r( r r* r r7 r r r- r/ r2 s r3 test_add_ondeletez+AutogenerateFKOptionsTest.test_add_ondelete %%b:y*ABB !H
G
F .
!H
G
F
r5 c
| ddii }| |d dddgddgdd
| |d d ddgddgd
d S )Nr CASCADEr r' r r r r r( r r r7 r r r s r3 test_remove_ondeletez.AutogenerateFKOptionsTest.test_remove_ondelete
%%z9&=rBB !H
G
F .
!H
G
F
r5 c Z | ddiddi }t |g dS )test case sensitivityr caSCAdeCasCadeNr r r s r3 test_nochange_ondeletez0AutogenerateFKOptionsTest.test_nochange_ondelete& s9 %%
#j)%<
E2r5 c
| i ddi }| |d dddgddgd d
| |d d ddgddgd
d S )Nonupdater r r' r r r r r( r r* r r7 r r r s r3 test_add_onupdatez+AutogenerateFKOptionsTest.test_add_onupdate- r r5 c
| ddii }| |d dddgddgdd
| |d d ddgddgd
d S )Nr r r r' r r r r r( r r r7 r r r s r3 test_remove_onupdatez.AutogenerateFKOptionsTest.test_remove_onupdateF r r5 c Z | ddiddi }t |g dS )r r r r Nr r s r3 test_nochange_onupdatez0AutogenerateFKOptionsTest.test_nochange_onupdate_ s; %%
#j)%<
E2r5 c Z | ddiddi }t |g dS )6test the RESTRICT option which MySQL doesn't report onr restrictNr r s r3 test_nochange_ondelete_restrictz9AutogenerateFKOptionsTest.test_nochange_ondelete_restrictg ; %%
$z:&>
E2r5 c Z | ddiddi }t |g dS )r r r Nr r s r3 test_nochange_onupdate_restrictz9AutogenerateFKOptionsTest.test_nochange_onupdate_restrictp r r5 c Z | ddiddi }t |g dS )>
!H
G
F .
!H
G
F!
r5 c | dddi }| |d dddgdd gd dd
| |d d
ddgdd gd d d S )Nr Tr r r' r r r r r( r r r7 r r s r3 /test_remove_initially_immediate_plus_deferrablezIAutogenerateFKOptionsTest.test_remove_initially_immediate_plus_deferrable: s %%%T : :B
!H
G
F .
!H
G
F
r5 c ^ | dddddd }t |g d S )NTr r r r r s r3 *test_add_initially_deferrable_nochange_onezDAutogenerateFKOptionsTest.test_add_initially_deferrable_nochange_oneX sB %% k : : k : :
E2r5 c ^ | dddddd }t |g d S )NTr r r r s r3 *test_add_initially_deferrable_nochange_twozDAutogenerateFKOptionsTest.test_add_initially_deferrable_nochange_twob B %% j 9 9 j 9 9
E2r5 c ^ | d ddd dd }t |g d S )Nr r r r s r3 ,test_add_initially_deferrable_nochange_threezFAutogenerateFKOptionsTest.test_add_initially_deferrable_nochange_threel r r5 c
| i ddi }| |d dddgddgd d
| |d d ddgddgd
d S Nr Tr r' r r r r r( )r r* r r7 )r r r s r3 test_add_deferrablez-AutogenerateFKOptionsTest.test_add_deferrablev s %%b<*>?? !H
G
F .
!H
G
F
r5 c
| ddii }| |d dddgddgdd
| |d d ddgddgd
d S r r r s r3 test_remove_deferrablez0AutogenerateFKOptionsTest.test_remove_deferrable s %%|T&:B?? !H
G
F .
!H
G
F
r5 N)&r_ r` ra rb r r
rI fk_ondelete_is_reflectedr r r fk_onupdate_is_reflectedr r fk_onupdater fk_ondelete_restrictr fk_onupdate_restrictr fk_ondelete_noactionr r r r r fk_initiallyr r
fk_deferrabler r r r r r fk_deferrable_is_reflectedr re r5 r3 r r sh K$% $% $%L 1
21
0 1
21
0 1
21
0 1
21
0 $ %$ - .- - .- - .- $ %$ -
.-
> -
.-
> 1 1
21 21
: %
&%
0 %
&%
2 & %
&% '&
6 & %
&% '&
8 % & '& &% % & '& &% % & '& &% &
'&
0 3
43
r5 r N)
sqlalchemyr r r r r r _autogen_fixturesr
testingr r
r r r r rg r re r5 r3 r s + + + + + + 1 1 1 1 1 1 # # # # # # C C C C C"4h C C CLr r r r r)8 r r rj[
[
[
[
[
2H [
[
[
[
[
r5