U
|
¸ý°d/ ã @ s d dl Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
|
lm Z dd lm
|
Z
|
dd l m Z d dlm Z d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlmZ d dlmZ G dd dejZG dd dejeZG dd d ejZd ZdS ) é Né )Útestingé )Úconfig)Úfixtures)Úutil)Úeq_)Úis_false)Úis_true)Ú requirements)ÚTableé )ÚCheckConstraint)ÚColumn)ÚForeignKeyConstraint)ÚIndex)Úinspect)ÚInteger©Úschema)ÚString)ÚUniqueConstraintc @ s e Zd ZdZd ddZdd Zd ddZd d
|
Zej e
|
j d d Z ej ej e
|
j d d Zeje
|
j dd Zej e
|
j dd Zeje
|
j dd Zeje
|
j dd Zeje
|
j dd Zeje
|
j dd Zeje
|
j dd Zeje
|
j d d ZdS )!Ú TableDDLTestTNc C s* t d| jtdtdddtdtd|dS ) NÚ
|
test_tableÚidTF©Ú primary_keyZ autoincrementÚdataé2 r ©r Úmetadatar r r )Úselfr © r" úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/testing/suite/test_ddl.pyÚ_simple_fixture s ûz TableDDLTest._simple_fixturec C s& t d| jtdtdddtdtdS )NZ _test_tabler TFr Ú_datar r )r! r" r" r# Ú_underscore_fixture% s üz TableDDLTest._underscore_fixturec C s" | j |d}td|jj}||fS )Nr Ú
|
test_index)r$ r Úcr )r! r ÚtableÚidxr" r" r# Ú_table_index_fixture- s z!TableDDLTest._table_index_fixturec C sJ t j ¡ 6}| | ¡ d¡¡ | | ¡ ¡}t| ¡ d W 5 Q R X d S )N)r z some data) r ÚdbÚbeginÚexecuteÚinsertÚvaluesÚselectr Úfirst)r! r) ÚconnÚresultr" r" r# Ú_simple_roundtrip2 s z TableDDLTest._simple_roundtripc C s& | ¡ }|jtjdd | |¡ d S ©NF©Z
|
checkfirst)r$ Úcreater r, r5 ©r! r) r" r" r# Útest_create_table8 s z TableDDLTest.test_create_tablec C s, | j tjd}|jtjdd | |¡ d S )Nr Fr7 )r$ r Z test_schemar8 r, r5 r9 r" r" r# Útest_create_table_schema? s z%TableDDLTest.test_create_table_schemac C s, | ¡ }|jtjdd |jtjdd d S r6 )r$ r8 r r, Zdropr9 r" r" r# Útest_drop_tableG s z TableDDLTest.test_drop_tablec C s& | ¡ }|jtjdd | |¡ d S r6 )r& r8 r r, r5 r9 r" r" r# Útest_underscore_namesN s z"TableDDLTest.test_underscore_namesc C sH | ¡ }|j|dd d|_| t |¡¡ tt| d¡ddi d S ©NFr7 z a commentr Útext) r$ r8 Úcommentr. r ÚSetTableCommentr r Úget_table_comment©r! Ú
|
connectionr) r" r" r# Útest_add_table_commentU s þz#TableDDLTest.test_add_table_commentc C sX | ¡ }|j|dd d|_| t |¡¡ | t |¡¡ tt| d¡dd i d S r> )
|
r$ r8 r@ r. r rA ZDropTableCommentr r rB rC r" r" r# Útest_drop_table_commenta s ÿz$TableDDLTest.test_drop_table_commentc C sF | ¡ }| tj|dd¡ tt| d¡ | tj|dd¡ d S )NT©Z if_not_existsr )r$ r. r Ú CreateTabler
|
r Ú has_tablerC r" r" r# Ú test_create_table_if_not_existsm s z,TableDDLTest.test_create_table_if_not_existsc C s | ¡ \}}| tj|dd¡ tt| d¡ tddd t| d¡D k | tj |dd¡ tddd t| d¡D k | tj |dd¡ d S )NTrG r r' c S s g | ] }|d qS ©Únamer" ©Ú.0Úixr" r" r# Ú
|
<listcomp> s ÿz@TableDDLTest.test_create_index_if_not_exists.<locals>.<listcomp>c S s g | ] }|d qS rK r" rM r" r" r# rP s ÿ)
|
r+ r. r rH r
|
r rI r Ú get_indexesZ CreateIndex©r! rD r) r* r" r" r# Ú test_create_index_if_not_existsw s& þÿÿ þÿÿz,TableDDLTest.test_create_index_if_not_existsc C sb | ¡ }| |¡ tt| d¡ | tj|dd¡ tt| d¡ | tj|dd¡ d S )Nr T©Z if_exists) r$ r8 r
|
r rI r. r Z DropTabler rC r" r" r# Útest_drop_table_if_exists s
|
z&TableDDLTest.test_drop_table_if_existsc C s | ¡ \}}| |¡ tddd t| d¡D k | tj|dd¡ tddd t| d¡D k | tj|dd¡ d S )Nr' c S s g | ] }|d qS rK r" rM r" r" r# rP ª s ÿz:TableDDLTest.test_drop_index_if_exists.<locals>.<listcomp>r TrT c S s g | ] }|d qS rK r" rM r" r" r# rP ´ s ÿ) r+ r8 r
|
r rQ r. r Z DropIndexr rR r" r" r# Útest_drop_index_if_exists¡ s$
|
þÿÿ þÿÿz&TableDDLTest.test_drop_index_if_exists)N)N)Ú__name__Ú
|
__module__Ú __qualname__Ú __backend__r$ r& r+ r5 r Z create_tabler Zprovide_metadatar: Zschemasr; Z
|
drop_tabler< r= Zcomment_reflectionrE rF Ztable_ddl_if_existsrJ Zindex_ddl_if_existsrS rU rV r" r" r" r# r sH
|
|
|
|
r c @ s e Zd ZdS )ÚFutureTableDDLTestN)rW rX rY r" r" r" r# r[ ½ s r[ c @ sr e Zd ZdZdZdd Zdd Zdd Zd d
|
Zd d Z e
|
j d ddde
|
j j ¡ fde
|
j j ¡ fdddd ZdS )ÚLongNameBlowoutTestzjtest the creation of a variety of DDL structures and ensure
|
label length limits pass on backends
|
|
Tc C s° ddd dd tdD ¡ i}||_td|tdtd d
|
d d td gd g}td|td |d d |j}| |¡ t j
|
j j r¤t |}| d¡}|d d }||fS |d fS d S )NÚfkzGforeign_key_%(table_name)s_%(column_0_N_name)s_%(referred_table_name)s_Ú_c s s& | ] }d dd tdD ¡V qdS )Ú c s s | ]}t d ¡V qdS ©ZabcdefN©ÚrandomÚchoice©rN Újr" r" r# Ú <genexpr>Ð s z3LongNameBlowoutTest.fk.<locals>.<genexpr>.<genexpr>é N©ÚjoinÚrange©rN Úir" r" r# rf Ï s ÿz)LongNameBlowoutTest.fk.<locals>.<genexpr>é
|
Úa_things_with_stuffÚid_long_column_nameT©r )Z test_needs_fkZaidz'a_things_with_stuff.id_long_column_nameZb_related_things_of_valuer rL )ri rj Únaming_conventionr r r r rL Ú
|
create_allr ÚrequiresZ&foreign_key_constraint_name_reflectionZenabledr Zget_foreign_keys) r! r rD Ú
|
conventionÚconsÚ actual_nameÚinspZfksÚreflected_namer" r" r# r] É sF
|
þüÿ ü ÿÿù
|
|
|
zLongNameBlowoutTest.fkc
|
C s ddd dd tdD ¡ i}||_td|tdtd d
|
td td d
|
}|j}|j}| |¡ t |}|
|
d¡}|d } || fS ) NÚpkz.primary_key_%(table_name)s_%(column_0_N_name)sr^ c s s& | ] }d dd tdD ¡V qdS )r_ c s s | ]}t d ¡V qdS r` ra rd r" r" r# rf ý s z3LongNameBlowoutTest.pk.<locals>.<genexpr>.<genexpr>é Nrh rk r" r" r# rf ü s ÿz)LongNameBlowoutTest.pk.<locals>.<genexpr>rm rn ro Trp Úid_another_long_namerL ) ri rj rq r r r r rL rr r Zget_pk_constraint)
|
r! r rD rt Úaru rv rw ry rx r" r" r# ry ÷ s*
|
þýÿ
|
ü
|
|
zLongNameBlowoutTest.pkc
|
C s ddd dd tdD ¡ i}||_td|tdtd d
|
td t}td |jj|jj }|j
|
}| |¡ t |}| d¡}|d d } || fS )NrO z(index_%(table_name)s_%(column_0_N_name)sr^ c s s& | ] }d dd tdD ¡V qdS )r_ c s s | ]}t d ¡V qdS r` ra rd r" r" r# rf s z3LongNameBlowoutTest.ix.<locals>.<genexpr>.<genexpr>rz Nrh rk r" r" r# rf s ÿz)LongNameBlowoutTest.ix.<locals>.<genexpr>rm rn ro Trp r{ r rL )ri rj rq r r r r r( ro r{ rL rr r rQ )
|
r! r rD rt r| ru rv rw rO rx r" r" r# rO s*
|
þýÿ
|
ü
|
|
zLongNameBlowoutTest.ixc C s ddd dd tdD ¡ i}||_tdd}td |tdtd
|
d tdt| |j}| |¡ t |}|
|
d ¡}|d d }||fS )NÚuqz4unique_constraint_%(table_name)s_%(column_0_N_name)sr^ c s s& | ] }d dd tdD ¡V qdS )r_ c s s | ]}t d ¡V qdS r` ra rd r" r" r# rf 5 s z3LongNameBlowoutTest.uq.<locals>.<genexpr>.<genexpr>rz Nrh rk r" r" r# rf 4 s ÿz)LongNameBlowoutTest.uq.<locals>.<genexpr>rm ro r{ rn Trp r rL ) ri rj rq r r r r rL rr r Zget_unique_constraints) r! r rD rt ru rv rw r} rx r" r" r# r} / s,
|
þýÿ
|
|
û
|
|
zLongNameBlowoutTest.uqc C s ddd dd tdD ¡ i}||_td}td|td td
|
d td t| |j}| |¡ t |}|
|
d¡}|d d }||fS )NÚckz check_constraint_%(table_name)sr^ c s s& | ] }d dd tdD ¡V qdS )r_ c s s | ]}t d ¡V qdS r` ra rd r" r" r# rf Q s z3LongNameBlowoutTest.ck.<locals>.<genexpr>.<genexpr>rz Nrh rk r" r" r# rf P s ÿz)LongNameBlowoutTest.ck.<locals>.<genexpr>rm zsome_long_column_name > 5rn ro Trp Zsome_long_column_namer rL ) ri rj rq r r r r rL rr r Zget_check_constraints) r! r rD rt ru rv rw r~ rx r" r" r# r~ L s,
|
þþÿ û
|
|
zLongNameBlowoutTest.ck)r] )ry )rO r~ r} Útype_)Úargnamesc C s~ t | |||\}}t|dks$t|d k rz|dt|
}t|t|k rpt|dd
|dt|d
n
|
t|| d S )Néÿ r éûÿÿÿé )ÚgetattrÚlenÚAssertionErrorr )r! r r rD rv rx Úoverlapr" r" r# Útest_long_convention_nameh s ÿ$z-LongNameBlowoutTest.test_long_convention_nameN)rW rX rY Ú__doc__rZ r] ry rO r} r~ r Ú combinationsrs Zcheck_constraint_reflectionZas_skipsZ unique_constraint_reflectionr r" r" r" r# r\ Á s . úr\ )r r[ r\ )rb r_ r r r r Z
|
assertionsr r r
|
r r r r r r r r r r r ZTestBaser ZFutureEngineMixinr[ r\ Ú__all__r" r" r" r# Ú<module> s0 % ?
|
