관리-도구
편집 파일: test_autogen_identity.cpython-38.pyc
U 0?�f� � @ s� d dl Zd dl mZ d dl mZ d dl mZ d dl mZ d dlmZ ddlm Z d d l m Z d d l m Z d d l m Z d d l mZ d dl mZ G dd� de e�ZdS )� N)�Column)�Integer)�MetaData)�Table)� sqla_compat� )�AutogenFixtureTest� )�testing)�config)�eq_)�is_true)�TestBasec @ s� e Zd ZdZdZdd� Zdd� Zdd� Zd d � Ze � d e d d �fe d d �d fe d d �e d dd�fe dd�e dd�fe ddddd�e ddddd�fe dddd�e dddd�f�e j jdd � ��Zd d � Zd d � Zd S )!�AutogenerateIdentityTest)Zidentity_columnsTc C s� t � }t � }td|tdtj�� td|tdtj�tdttjddd�dd�� | �||�}t|d d d � t|d d d� t|d d j d� |d d j }t t |tj�� t|j d� t|jd� d S ) N�user�other�id� � ��start� incrementT�Z primary_keyr � add_column� r )r r r �sa�Textr �Identity�_fixturer �name�identityr � isinstancer r )�self�m1�m2�diffs�i� r' �Y/opt/hc_python/lib/python3.8/site-packages/alembic/testing/suite/test_autogen_identity.py�test_add_identity_column s, �� z1AutogenerateIdentityTest.test_add_identity_columnc C s� t � }t � }td|tdttjddd�dd�� td|� | �||�}t|d d d � t|d d d� |d d }t|jd� t t |j tj�� t|j j d� t|j j d� d S ) Nr r r r r Tr r Z remove_column)r r r r r r r r r r r! r r r )r" r# r$ r% �cr' r' r( �test_remove_identity_column1 s* �� z4AutogenerateIdentityTest.test_remove_identity_columnc C sT t � }t � }||fD ]$}tjdd�}td|tdt|�� q| �||�}t|g � d S )Nr �r r r )r r r r r r r r )r" r# r$ �m�id_r% r' r' r( � test_no_change_identity_columnM s z7AutogenerateIdentityTest.test_no_change_identity_columnc C s� t � }t � }tjr ddd�}n ddd�}td|tdttjdd��� tjf ddi|��}td|tdt|�� | �||�}t j j d kr�t t |�d � t|d d d d � n t|g � d S ) NT)Zoracle_on_nullZ oracle_order)Zon_null�orderr r r r, r �oracler r �modify_default)r r Zidentity_has_dialect_kwargsr r r r r r r �dbr r �lenr )r" r# r$ �argsr. r% r' r' r( �test_dialect_kwargs_changesY s z4AutogenerateIdentityTest.test_dialect_kwargs_changesNr r, r r F)�alwaysr r �d )r Zminvalue�maxvalue�cycle� r i' )r r r9 i c C s |rt jf |�fnd}|r(t jf |�fnd}t� }t� }td|tdtf|�� tdt j�� td|tdtf|�� tdt j�� | �||�}tt |d �d� |d d }t|d d� t|d d� t|d d� |d }|d } d d � } | ||� | || � d S )Nr' r r r r r r2 r r r � c S sH | r8t t|tj�� | �� D ]\}}tt||�|� q n t |dk� d S )N)NF)r r! r r �itemsr �getattr)�kwZidt�k�vr' r' r( �check� s z<AutogenerateIdentityTest.test_change_identity.<locals>.check) r r r r r r r r r r4 ) r" �before�afterZ arg_beforeZ arg_afterr# r$ r% �old�newrB r' r' r( �test_change_identitym s6 � � z-AutogenerateIdentityTest.test_change_identityc C s� t � }t � }td|tdt�tdtj�� td|tdttjddd��tdtj�� | �||�}tt |d �d� |d d }t|d d � t|d d� t|d d� t|d d � |d }t t |tj�� t|j d� t|j d� d S ) Nr r r r �� �r r9 r r r2 r r r< )r r r r r r r r r r4 r r! r r9 )r" r# r$ r% �addedr' r' r( �test_add_identity_to_column� s2 � � z4AutogenerateIdentityTest.test_add_identity_to_columnc C s� t � }t � }td|tdttjddd��tdtj�� td|tdt�tdtj�� | �||�}tt |d �d� |d d }t|d d � t|d d� t|d d� t|d d � |d }t t |tj�� d S ) Nr r r rH rI r r r r2 r r< r ) r r r r r r r r r r4 r r! )r" r# r$ r% Zremovedr' r' r( � test_remove_identity_from_column� s. � � z9AutogenerateIdentityTest.test_remove_identity_from_column)�__name__� __module__� __qualname__� __requires__Z __backend__r) r+ r/ r6 r � combinations�dictr � requirementsZidentity_columns_alterrG rK rL r' r' r' r( r s, � �� * r )Z sqlalchemyr r r r r Z alembic.utilr Z_autogen_fixturesr � r r r r r r r' r' r' r( �<module> s