&˔^ t U d dl mZ d dlZd dlmZ d dlZd dlZd dlmZ d dlm Z d dl
m Z d dl
m Z d dl
m
Z
d d l
mZ d d
l
mZ d d l
mZ d d l
mZ d d
lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dl Z ddl m Z ddl m Z ddl!m"Z" ddl#m$Z$ ddl%m&Z& ddl'm(Z( ddl'm)Z) ddl'm*Z* ej+ Z,e,- d g G d d e Z G d d ee Ze)rd d!lm.Z. n
G d" d# Z.d$e._/ d0d&Z0i Z1d'e2d(< ed) Z3ed* Z4 d1d-Z5 G d. d/ Z6dS )2 ) annotationsN)contextmanager)Any)Dict)Column)inspect)MetaData)String)Table)testing)text)config)mock)eq_)
TablesTest)TestBase ) _get_dialect )EnvironmentContext)MigrationContext)
Operations) sqla_compatcreate_mock_engine)sqla_14)sqla_2ztest.cfgc e Zd ZeZ ej d Zej d Zej d Z dS )r c # K | d 5 t | V d d d d S # 1 swxY w Y d S )NT)_per_migration)begin_transactionr )selfmigration_contexts j/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/alembic/testing/fixtures.py ops_contextzTestBase.ops_context) s
0
0
0
E
E 0 0 .// / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s 8<<c J t j |t d S )NT)transaction_per_migration)opts)r configuredict)r"
connections r$ r# zTestBase.migration_context. s, ) TD A A A
c # K t j 5 }|V d d d d S # 1 swxY w Y d S N)r dbconnect)r" conns r$ r+ zTestBase.connection4 s
Y
DJJJ s 377N)
__name__
__module__ __qualname__r is_sqlalchemy_futurer fixturer% r# r+ r, r$ r r & sx !W_0 0 0
_
_
_ _ r, r c e Zd ZdS )r N)r2 r3 r4 r7 r, r$ r r : s Dr, r )FutureEngineMixinc e Zd ZdZdS )r9 )
sqlalchemy_14N)r2 r3 r4 __requires__r7 r, r$ r9 r9 B s ) r, r9 T
postgresql://c > g fd}t | | fS )Nc ~ t | j d S N)dialect)appendstrcompilerA )sql multiparamsparamsbufengines r$ dumpzcapture_db..dumpL s3
3s{{6>{::;;<<<<.configure` 0
RtS C r, r) )
popioBytesIOStringIOrU r r) r patchobject)rZ rH r) rY s ` @r$ capture_context_bufferrb V s vvj% jllkmmIIx#>>??? 'D! ! ! ! ! !
-{I F F s B$$B(+B(c + K ddl m} ddlm} t j | }| }| |d fd } d|i t j
fd}t j
t d | 5 V d d d d S # 1 swxY w Y d S )
Nr )_sqlite_file_dbr )eventbefore_cursor_executec 8 |dz d S )N
)write)r1 cursor statement
parameterscontext executemanyrH s r$ bcez*capture_engine_context_buffer..bces s )d"#####r, r+ c . |j di | i |S rS rT rV s r$ r) z0capture_engine_context_buffer..configurez r[ r, r) )envrd
sqlalchemyre r] r_ r0 listens_forrU r r) r r` ra ) rZ rd re engr1 ro r) rH rY s ` @@r$ capture_engine_context_bufferru h sP $$$$$$
+--C
/
C ;;==D
t 455$ $ $ $ 65$ II|T"### 'D! ! ! ! ! !
-{I F F s 3CC C defaultFc i }|rt | |d< G fdd } | G fddt }r|d< |r||d< t s | d k rt d
d _ nt | || _ d _ s fd
}t j | } n |d< d } | | | }
t |
t j _ |
S )N)naming_conventiontarget_metadatac & e Zd Zd Z fdZd ZdS )op_fixture..buffer_c g | _ d S r. linesr" s r$ __init__z$op_fixture..buffer_.__init__ s
DJJJr, c | }t j dd| }r,t j dd| }t j dd| }| j | d S )N[\n\t] z z\;\n*$)stripresubr~ rB )r" msgas_sqls r$ ri z!op_fixture..buffer_.write so ))++C&B,,C
1 fWb#..fYC00 J c " " " " "r, c d S r. r7 r s r$ flushz!op_fixture..buffer_.flush s Dr, N)r2 r3 r4 r ri r )r s r$ buffer_r{ sL # # # # # r, r c 8 e Zd Z fdZ fdZ fdZ fdZdS )op_fixture..ctxc S r. r7 r" rH s r$ get_bufz op_fixture..ctx.get_buf s Jr, c g j d d < d S r. r} r s r$ clear_assertionsz(op_fixture..ctx.clear_assertions s CIaaaLLLr, c F t j d |D d S )Nc : g | ]}t j d d| S )r r )r r ).0ss r$
z3op_fixture..ctx.assert_.. s& BBBBF9b! 4 4BBBr, )r r~ )r" rE rH s r$ assert_z op_fixture..ctx.assert_ s,
BBcBBB C C C C Cr, c x j D ] }t j dd| |v r d S J d|dj )Nr r FzCould not locate fragment z in )r~ r r )r" rE stmtrH s r$ assert_containsz'op_fixture..ctx.assert_contains sb
6)R--55FF 6 CCII ur, N)r2 r3 r4 r r r r )rH s r$ ctxr s~ D D D D D
r, r r
literal_bindsmariadbmysql)
r MariaDBTc t | t rt | } | j sJ t | } | d S r@ )
isinstancerC r
supports_executionrD ri )r
multiparamparamrE rH ctx_dialects r$ executezop_fixture..execute s] $$$
"Dzz* * **dll;l7788C IIcNNNNNr, )rA r rQ )
r r r r server_version_infosupports_native_boolean#non_native_boolean_check_constraintr Mockr alembicop_proxy)
rA r rx r native_booleanr( r r r r+ rm rH r s
` @@r$
op_fixturer s D P"*=N"O"O"O
$ '))C , X . -_ ,w)++"7++ *? '' #7++ !.< + ;? 7
Y{GDDD
#_
c+z400G"7++GJ Nr, c 2 e Zd ZdZd Zd Zd Zd ZddZdS ) AlterColRoundTripFixture) alter_columnc t j | _ t j | j | _ t | j | _ t | _
d S r. ) r r/ r0 r1 r r) r r r r metadatar s r$ setUpz AlterColRoundTripFixture.setUp sK I%%'' #-di88 TX&&
r, c t d| }t d| }| j j || rJ d|d|d d S )Nqz
Type objects and didn't compare as equivalent)r r impl compare_type)r" t1t2c1c2s r$
_compare_typez&AlterColRoundTripFixture._compare_type sw
C__
C__8=--
L L LDFBB K L L
L Lr, c t d|| }t d|| }| j j |||| rJ d|d|d d S )Nr )server_defaultzserver defaults r r )r r r compare_server_default)r" r s1r s2r r s r$ _compare_server_defaultz0AlterColRoundTripFixture._compare_server_default s
CB
/
/
/
CB
/
/
/8=77 B
O O OGIrr222 N O O
O Or, c
t j | j | j 5 | j | j d d d n # 1 swxY w Y | j d S r. )r %_safe_rollback_connection_transactionr1 beginr drop_allcloser s r$ tearDownz!AlterColRoundTripFixture.tearDown s 9$)DDD
Y__
. . M " "49 - - - . . . . . . . . . . . . . . . s A A#&A#Nc t | dd | dt d | dd | dd }t d | j | }t j | j 5 | | j t | j }|
d d
}| j d |j
|j |j |j nd |j rdnd | dd | dd | dd | dd d d d n # 1 swxY w Y t | j }|
d d
}||}t# |d d|v r|d n|j
| |d | d|d t# |d | d|j | |d | d
d | d|d d|v r
|d j n|j |j j j nd d S )Nnamecolnametyper nullableTr )r r xr F)
existing_typeexisting_server_defaultexisting_nullabler r new_column_nametype_rv )r getr
r r r _ensure_scope_for_ddlr1 creater get_columnsr r r r r r r r r r
rW ) r" from_to_comparecolumntinspold_colnew_cols r$ _run_alter_colz'AlterColRoundTripFixture._run_alter_col s IIfi ( ( IIffRjj ) )YYz400 99%5t<<
#t}f - -
.ty
9
9
HHTY 49%%D&&s++A.G
G $k(4 )/(=(=*0/"D$$u T22 "ww'7 ? ? # 5 5ggfd++
!
, ty!!""3''* ?G FO % 0 0GFOOfk
FOW[[ A A
GJ Z !I!IJJJ $$ FO KK 4 ( ( KK 0 0 7**
$ % * * $0 &*//
s C F
FFr. ) r2 r3 r4 r< r r r r r r7 r, r$ r r sq %L# # # L L LO O O 7
7
7
7
7
7
r, r )r= )rv FNFN)7
__future__r configparser
contextlibr r] r typingr r rr r r r r
r r r
sqlalchemy.testingr r sqlalchemy.testing.assertionsr sqlalchemy.testing.fixturesr SQLAlchemyTablesTestr SQLAlchemyTestBaser
assertionsr environmentr migrationr
operationsr utilr util.sqla_compatr r r ConfigParsertesting_configreadr9 r5 rK rL __annotations__rb ru r r r7 r, r$ r st " " " " " " " % % % % % % % % % % % % # # # # # # - - - - - - J J J J J J F F F F F F $ $ $ $ $ $ , , , , , , ( ( ( ( ( ( # # # # # # 1 1 1 1 1 1 & & & & & & % % % % % % + *,, ZL ! ! ! ! ( / *=======* * * * * * * * *. & " 4
V V V VrY
Y
Y
Y
Y
Y
Y
Y
Y
Y
r,