bgddlZddlmZddlmZddlmZddlmZddlmZdd lm Z dd l m Z dd l m Z dd l m Z dd l m Z Gddee ZdS)N)Column)Integer)MetaData)Table)AutogenFixtureTest)testing)config)eq_)is_true)TestBasec eZdZdZdZdZdZdZej de dfe ddfe de dd fe d e d fe d ddde d ddd fe ddde dd dfe j j dZdZdZdZdS)AutogenerateIdentityTest)identity_columnsTc t}t}td|tdtjtd|tdtjtdt tjddd|||}t|d d d t|d d dt|d d j d|d d j }tt|tjt|j dt|jddS) Nuserotheridstart incrementT primary_keyr add_columnr )rrrsaTextrIdentity_fixturer nameidentityr isinstancerr)selfm1m2diffsis |/builddir/build/BUILD/cloudlinux-venv-1.0.7/venv/lib/python3.11/site-packages/alembic/testing/suite/test_autogen_identity.pytest_add_identity_columnz1AutogenerateIdentityTest.test_add_identity_columns7 ZZ ZZ fb&"'22333   7BG $ $  !q111      b"%% E!HQK&&& E!HQK    E!HQK d### !HQK  1bk**+++ AGQ AKc vt}t}td|tdtt jdddtd||||}t|ddd t|ddd|dd}t|jdtt|j tjt|j j dt|j j ddS) Nrrrr rTrr remove_column)rrrrrr!r"r r#r r%r$rr)r&r'r(r)cs r+test_remove_identity_columnz4AutogenerateIdentityTest.test_remove_identity_column0s ZZ ZZ    !q111      fb b"%% E!HQK))) E!HQK   !HQK AFD 1:r{33444 AJ a    AJ !$$$$$r-c t}t}||fD]:}td|tdtt jd;|||}t|gdS)Nrrrr)rrrrrr!r"r )r&r'r(mr)s r+test_no_change_identity_columnz7AutogenerateIdentityTest.test_no_change_identity_columnLs ZZ ZZb  A tWbk&:&:&:;;      b"%% E2r-Nrr3rrF)alwaysrrd)rminvaluemaxvaluecycle r i')rrr9i c |rtjdi|fnd}|rtjdi|fnd}t}t}td|t dt g|Rt dtjtd|t dt g|Rt dtj|||}tt|dd|dd}t|ddt|ddt|d d|d }|d } d } | ||| || dS) Nrrrrrmodify_defaultrr rc|ratt|tj|D]#\}}t t |||$dSt|dvdS)N)NF)r r%rr!itemsr getattr)kwidtkvs r+checkz.checks . 3 44555HHJJ,,DAqQ++++,,},-----r-) rr!rrrrr r"r len) r&beforeafter arg_before arg_afterr'r(r)oldnewrGs r+test_change_identityz-AutogenerateIdentityTest.test_change_identity[s 28?bk++F++--R /4r rr?)rrrrrr r!r"r rHr r%rr9)r&r'r(r)addeds r+test_add_identity_to_columnz4AutogenerateIdentityTest.test_add_identity_to_columnsi ZZ ZZ   4 ! ! 7BG $ $        4"+A"E"E"E F F 7BG $ $     b"%% CaMM1a  E!H&''' E!Hf E!Hd E!Hda 5"+../// EK END!!!!!r-c t}t}td|tdtt jddtdtjtd|tdttdtj|||}tt|dd|dd}t|dd t|ddt|d dt|d d|d }tt|tjdS) NrrrrQrRrrrr>r r?r) rrrrrr!r r"r rHr r%)r&r'r(r)removeds r+ test_remove_identity_from_columnz9AutogenerateIdentityTest.test_remove_identity_from_columnsE ZZ ZZ   4"+A"E"E"E F F 7BG $ $        4 ! ! 7BG $ $     b"%% CaMM1a  E!H&''' E!Hf E!Hd E!Hd( 7BK0011111r-c t}t}td|tdtt jddtdtjtd|tdtt jddtdtj|||}tj j j st|gdStt|dd |dd}t|dd t|ddt|d d|d }|d }tt|tjtt|tjdS)NrrrT)ron_nullrFrrr>r rr?)rrrrrr!r r"r requirementssupports_identity_on_nullenabledr rHr r%)r&r'r(r)rMrNs r+test_identity_on_nullz.AutogenerateIdentityTest.test_identity_on_nulls ZZ ZZ   4"+At"D"D"D E E 7BG $ $        4"+Au"E"E"E F F 7BG $ $     b"%%"<D 2 rNNNNN E!H q ! ! !!HQKE a* + + + a& ! ! ! a$   (C(C JsBK00 1 1 1 JsBK00 1 1 1 1 1r-)__name__ __module__ __qualname__ __requires__ __backend__r,r1r5r combinationsdictr rZidentity_columns_alterrOrTrWr]r=r-r+rrs(LK:%%%8   W tt!}}} A A12223 U   TT.../ Dq1s$ ? ? ? Dq1s% @ @ @ DrQ 6 6 6 DrQ 6 6 6    /((0/  (T"""@222<22222r-r) sqlalchemyrrrrr_autogen_fixturesrr r r r rrr=r-r+ris111111b2b2b2b2b218b2b2b2b2b2r-