a 7 d dl Zd dl mZ d dl mZ d dl mZ d dl mZ ddlmZ dd lm Z dd
l m
Z
dd l m Z dd l m Z dd
l m
Z
ddl mZ ddl mZ ddl mZ G d dee ZdS ) N)Column)Integer)MetaData)Table )AutogenFixtureTest )testing)config)eq_)
exclusions)is_)is_true)mock)TestBasec e Zd ZdZdZd Zd Z ej d d d d e j
j d Z ej d
d d d
e j
j
f d Ze j
j d ZdS )AutogenerateComputedTest)computed_columnsTc t }t }t d|t dt d t d|t dt d t dt t j d | || }t |d d d t |d d d t |d d
j d |d d
j }t t |t j t |j
d t t |j d d S ) NuseridT primary_keyfoo5r
add_column r )r r r r saComputed_fixturer namecomputedr
isinstancer persistedstrsqltextselfm1m2diffscs }/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/alembic/testing/suite/test_autogen_computed.pytest_add_computed_columnz1AutogenerateComputedTest.test_add_computed_column s2
ZZ
ZZ
fb&wDAAABBB
4d 3 3 3 5'2;s#3#3 4 4
b"%% E!HQK &&& E!HQK E!HQK e$$$ !HQK
1bk**+++ AK C NNC c H t }t }t d|t dt d t dt t j d t d|t dt d | || }t |d d d t |d d d |d d
}t |j d t j
j j r-t t |j t j nt! |j d t j
j j rZt t |j t j t t) |j j j d d S t j
j j r.t t |j t j d S t! |j d d S ) Nr r Tr r r r
remove_columnr r )r r r r r r r r r! r requirementscomputed_reflects_normallyenabledr r# r" r #computed_reflects_as_server_defaultserver_default
DefaultClauser% argtextr' s r- test_remove_computed_columnz4AutogenerateComputedTest.test_remove_computed_column. s
ZZ
ZZ
4d 3 3 3 5'2;s#3#3 4 4
fb&wDAAABBB
b"%% E!HQK ))) E!HQK !HQK AFE 9 A " Jqz2;77 8 8 8 8
D ! ! ! B J " Jq /1ABB C C C A$(-.. 4 4 4 4 4
;
C " Jqz2;77 8 8 8 8 8
D ! ! ! ! !r/ c . d t j d fS Nbar*5r r r/ r- z!AutogenerateComputedTest.P s r{7++, r/ c . t j d d fS r< r> r? r/ r- r@ z!AutogenerateComputedTest.Q s "+g&&- r/ c V t j d t j dd fS )Nr= bar * 42T)r$ r> r? r/ r- r@ z!AutogenerateComputedTest.R s( K K
d 3 3 3
r/ c R t j d t j d fS )Nr= rC r> r? r/ r- r@ z!AutogenerateComputedTest.V s W%%r{:'>'>? r/ c
t j |fi t \ }}t }t }|g n|g}|g n|g}t d|t dt d t dt t dt g|R t d|t dt d t dt t dt g|R t j d 5 }| || }d d d n # 1 swxY w Y t |j t j d g t t | g d S ) Nr r Tr barr alembic.util.warnz/Computed default on user.foo cannot be modified)r
resolve_lambdalocalsr r r r r patchr r
mock_callscalllistr( test_case
arg_before arg_afterr) r* mock_warnr+ s r- !test_cant_change_computed_warningz:AutogenerateComputedTest.test_cant_change_computed_warningO s !( 6y M MFHH M M
I
ZZ
ZZ%-RRJ<
#+BB)
4d 3 3 3 5' " " 5' /J / / /
4d 3 3 3 5' " " 5' .I . . .
Z+
,
, * MM"b))E * * * * * * * * * * * * * * *
YH
I
I J
DKK 8DD "D c dS )N)NNr? r? r/ r- r@ z!AutogenerateComputedTest.| s r/ c R t j d t j d fS )Nr r> r? r/ r- r@ z!AutogenerateComputedTest.} s S!!2;s#3#34 r/ c R t j d t j d fS r< r> r? r/ r- r@ z!AutogenerateComputedTest.~ s W%%r{7';';< r/ c . t j d d fS r< r> r? r/ r- r@ z!AutogenerateComputedTest. s R[))40 r/ c
t j |fi t \ }}t }t }|g n|g}|g n|g}t d|t dt d t dt t dt g|R t d|t dt d t dt t dt g|R t j d 5 }| || }d d d n # 1 swxY w Y t |j g t t | g d S )Nr r Tr rF r rG )
r
rH rI r r r r r rJ r r rK rM rN s r- test_computed_unchangedz0AutogenerateComputedTest.test_computed_unchanged{ s !( 6y M MFHH M M
I
ZZ
ZZ%-RRJ<
#+BB)
4d 3 3 3 5' " " 5' /J / / /
4d 3 3 3 5' " " 5' .I . . .
Z+
,
, * MM"b))E * * * * * * * * * * * * * * * I "%%% DKK rT c
^ t }t }t d|t dt d t dt t dt t j d t d|t dt d t dt t dt | || }t |d d d d t |d d d
d t |d d d d |d d d }|d d d
}t |d t t |t j t j
t d r)t t |j j d dS t j
t d r)t t |j j d dS dS )zAsserts the current behavior which is that on PG and Oracle,
the GENERATED ALWAYS AS is reflected as a server default which we can't
tell is actually "computed", so these come out as a modification to
the server default.
r r Tr rF r zbar + 42r modify_defaultr r N
postgresqlz
(bar + 42)oraclez"BAR"+42)r r r r r r r r r r r# r7 r
againstr r% r8 r9 )r( r) r* r+ oldnews r- (test_remove_computed_default_on_computedzAAutogenerateComputedTest.test_remove_computed_default_on_computed s ZZ
ZZ
4d 3 3 3 5' " " 5'2;z#:#: ; ;
4d 3 3 3 5' " " 5' " "
b"%% E!HQKN ,--- E!HQKNF### E!HQKNE"""Ahqk"oAhqk"o C
3 011222 fl 3 3 / CGL!!< 0 0 0 0 0
1
1 / CGL!!: . . . . . / /r/ N)__name__
__module__ __qualname__ __requires__ __backend__r. r: r
combinationsr r2 r3 rS )computed_doesnt_reflect_as_server_defaultrZ r5 rd r? r/ r- r r s (LK! ! !0 " " "B W,, - -
@? 3 43 D W 44<< 0 0 I
< <(/ (/ =<(/ (/ (/r/ r )
sqlalchemyr r r r r _autogen_fixturesr r
r r r
r r r r r r? r/ r- ro s< 1 1 1 1 1 1 ! ! ! ! ! ! y/ y/ y/ y/ y/ 18 y/ y/ y/ y/ y/r/