관리-도구
편집 파일: test_ddl.cpython-38.pyc
U -?�f�. � @ 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� dej�ZG dd� deje�ZG dd � d ej�Zd 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 dd�Zdd� Zd dd�Zd 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| jtdtddd�tdtd��|d�S ) N� test_table�idTF�� primary_keyZ autoincrement�data�2 r �r �metadatar r r )�selfr � r" �Q/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/suite/test_ddl.py�_simple_fixture" s �z TableDDLTest._simple_fixturec C s& t d| jtdtddd�tdtd���S )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_fixture3 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_roundtrip8 s z TableDDLTest._simple_roundtripc C s& | � � }|jtjdd� | �|� d S �NF�Z checkfirst)r$ �creater r, r5 �r! r) r" r" r# �test_create_table> 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_schemaE s z%TableDDLTest.test_create_table_schemac C s, | � � }|jtjdd� |jtjdd� d S r6 )r$ r8 r r, Zdropr9 r" r" r# �test_drop_tableM s z TableDDLTest.test_drop_tablec C s& | � � }|jtjdd� | �|� d S r6 )r& r8 r r, r5 r9 r" r" r# �test_underscore_namesT 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_comment[ 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_commentg 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_existss 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_exists} 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�� fdd�dd� �ZdS )�LongNameBlowoutTestzjtest the creation of a variety of DDL structures and ensure label length limits pass on backends Tc C s� ddd� dd� td�D �� 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� td�D ��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_reflection�enabledr Zget_foreign_keys) r! r rD � convention�cons� actual_name�inspZfks�reflected_namer" r" r# r] � sF ��� � ��� zLongNameBlowoutTest.fkc C s� ddd� dd� td�D �� 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� td�D ��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 ru �arv rw rx rz ry r" r" r# rz � s* ��� � zLongNameBlowoutTest.pkc C s� ddd� dd� td�D �� 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� td�D ��V qdS )r_ c s s | ]}t �d �V qdS r` ra rd r" r" r# rf s z3LongNameBlowoutTest.ix.<locals>.<genexpr>.<genexpr>r{ 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 ru r} rv rw rx rO ry r" r" r# rO s* ��� � zLongNameBlowoutTest.ixc C s� ddd� dd� td�D �� 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� td�D ��V qdS )r_ c s s | ]}t �d �V qdS r` ra rd r" r" r# rf ; s z3LongNameBlowoutTest.uq.<locals>.<genexpr>.<genexpr>r{ Nrh rk r" r" r# rf : 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 ru rv rw rx r~ ry r" r" r# r~ 5 s, ��� � zLongNameBlowoutTest.uqc C s� ddd� dd� td�D �� 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� td�D ��V qdS )r_ c s s | ]}t �d �V qdS r` ra rd r" r" r# rf W s z3LongNameBlowoutTest.ck.<locals>.<genexpr>.<genexpr>r{ Nrh rk r" r" r# rf V 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 ru rv rw rx r ry r" r" r# r R s, ��� � zLongNameBlowoutTest.ck)r] )rz )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 rw ry Zoverlapr" r" r# �test_long_convention_namen s �$z-LongNameBlowoutTest.test_long_convention_nameN)rW rX rY �__doc__rZ r] rz 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 % ?