관리-도구
편집 파일: test_autogen_fks.cpython-38.pyc
U 0?�f�� � @ s� d dl mZ d dl mZ d dl mZ d dl mZ d dl mZ d dl mZ ddlmZ d d l m Z d d l m Z d d l m Z d d l m Z d dl mZ G dd� dee�ZG dd� dee�ZG dd� dee�ZdS )� )�Column)�ForeignKeyConstraint)�Integer)�MetaData)�String)�Table� )�AutogenFixtureTest� )� combinations)�config��eq_)�mock)�TestBasec @ sd e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z e j j dd� �Zdd� Zdd� ZdS )�AutogenerateForeignKeysTestT)Z!foreign_key_constraint_reflectionc C s t � }t � }td|tdtd�dd�� td|tdtdd�tdtd �d d �td td�d d�tdtd��tdgdg�� td|tdtd�dd�� td|tdtdd�tdtd �d d �td td�d d�tdtd��� | �||�}| j|d dddgddgdd� d S )N� some_table�test� T�� primary_key�user�id�name�2 F�Znullable�a1�x�Zserver_default�test2�some_table.testr � remove_fk�servergenerated��conditional_name�r r r r r r �_fixture�_assert_fk_diff��self�m1�m2�diffs� r- �T/opt/hc_python/lib/python3.8/site-packages/alembic/testing/suite/test_autogen_fks.py�test_remove_fk sN � � � � �z*AutogenerateForeignKeysTest.test_remove_fkc C s t � }t � }td|tdtdd�tdtd��� td|tdtdd�tdtd �d d �td td�d d�tdtd��� td|tdtdd�tdtd��� td|tdtdd�tdtd �d d �td td�d d�tdtd��tdgdg�� | �||�}| �|d dddgddg� d S )Nr r Tr r r r r r Fr r r r r r r �add_fk)r r r r r r r& r' r( r- r- r. � test_add_fkC sP � � � � �z'AutogenerateForeignKeysTest.test_add_fkc C s� t � }t � }td|tdtdd�tdtd��� td|tdtdd�tdtd �d d �td td�d d�tdt�tdgdg�� td|tdtdd�tdtd��� td|tdtdd�tdtd �d d �td td�d d�tdt�tdgdg�� | �||�}t|g � d S )Nr r Tr r r r r r Fr r r r r � some_table.id)r r r r r r r& r r( r- r- r. �test_no_changen sD � � � � z*AutogenerateForeignKeysTest.test_no_changec C s6 t � }t � }td|tdtd�dd�tdtd�dd�� td|tdtdd�td td �d d �td td�dd�tdtd��tdtd��tddgddg�� td|tdtd�dd�tdtd�dd�� td|tdtdd�td td �d d �td td�dd�tdtd��tdtd��tddgddg�� | �||�}t|g � d S )Nr �id_1r Tr �id_2r r r r Fr r r r � other_id_1� other_id_2�some_table.id_1�some_table.id_2�r r r r r r r& r r( r- r- r. �test_no_change_composite_fk� sT � ��� �� z7AutogenerateForeignKeysTest.test_no_change_composite_fkc C sD t � }t � }td|tdtd�dd�� td|tdtdd�tdtd �d d �td td�d d�tdtd��tdgdgdd�� td|tdtd�dd�� td|tdtdd�tdtd �d d �td td�d d�tdtd��td gdgdd�� | �||�}| j|d dddgddgtj j j �rdnd d� | j|d ddd gddgdd� d S )Nr r r Tr r r r r Fr r r r r r ZMyFK�r Zmyfkr r! r r0 ) r r r r r r r& r' r � requirements�fk_names�enabledr( r- r- r. �1test_casing_convention_changed_so_put_drops_first� sb � � � � � �zMAutogenerateForeignKeysTest.test_casing_convention_changed_so_put_drops_firstc C sD t � }t � }td|tdtd�dd�tdtd�dd�� td|tdtdd�td td �d d �td td�dd�tdtd��tdtd��� td|tdtd�dd�tdtd�dd�� td|tdtdd�td td �d d �td td�dd�tdtd��tdtd��tddgddgdd�� | �||�}| j|d ddddgdddgdd� d S )Nr r4 r Tr r5 r r r r Fr r r r r6 r7 r8 r9 � fk_test_namer< r r0 r% r( r- r- r. � test_add_composite_fk_with_name s^ � � � �� �z;AutogenerateForeignKeysTest.test_add_composite_fk_with_namec C sD t � }t � }td|tdtd�dd�tdtd�dd�� td|tdtdd�td td �d d �td td�dd�tdtd��tdtd��tddgddgdd�� td|tdtd�dd�tdtd�dd�� td|tdtdd�td td �d d �td td�dd�tdtd��tdtd��� | �||�}| j|d ddddgdddgdd� d S )Nr r4 r Tr r5 r r r r Fr r r r r6 r7 r8 r9 rA r< r r! r# r% r( r- r- r. �test_remove_composite_fk= s^ � ��� � �z4AutogenerateForeignKeysTest.test_remove_composite_fkc C s t � }t � }td|tdtd�dd�tdtd�dd�� td|tdtdd�td td��td td��� td|tdtd�d dd �tdtd�d dd �� td|tdtdd�td td�dd�td td�dd�tddgddgdd�� | �||�}| j|d ddd d gdddgdd� d S )Nr r4 r Tr r5 r r r6 r7 �tid1��keyr �tid2�oid1�rF �oid2�some_table.tid1�some_table.tid2rA r< r r0 r% r( r- r- r. �test_add_fk_colkeysu sV � �� �� �z/AutogenerateForeignKeysTest.test_add_fk_colkeysc C s t � }t � }td|tdtd�dd�tdtd�dd�� td|tdtdd�td td��td td��td d gd d g�� td|tdtd�d dd�tdtd�ddd�� td|tdtdd�td td�dd�td td�dd�tddgddg�� | �||�}t|g � d S )Nr r4 r Tr r5 r r r6 r7 r8 r9 rD rE rG rH rI rJ rK rL r: r( r- r- r. �test_no_change_colkeys� sL � �� � �� z2AutogenerateForeignKeysTest.test_no_change_colkeysN)�__name__� __module__� __qualname__� __backend__� __requires__r/ r1 r3 r; r@ rB r r= �no_name_normalizerC rM rN r- r- r- r. r s /+*2=6 73r c @ sP e Zd ZdZdZedd�ejjdd� ��Z dd� Z edd�ejjd d � ��Z d S ) �IncludeHooksTestT)r> )�objectr< c C s t � }t � }td|tdtdd��}td|tdt�tdt��}|�t|jjg|jjgdd �� |�t|jj g|jjgd d �� td|tdtdd��}td|tdt�tdt�� |d kr�d d � }| j |||d�}n |dkr�dd� }| j |||d�}| j |d dddgddgd d� t t |�d� d S )N�refr Tr �tr �y�fk1r< �fk2rV c S s t | t�o |dko |o |dk S �N�foreign_key_constraintrZ �� isinstancer �Zobject_r �type_Z reflectedZ compare_tor- r- r. �include_object s ���zBIncludeHooksTest.test_remove_connection_fk.<locals>.include_object�Zobject_filtersr c S s< | dkr4|dkrdS t |d� t |d ddd�� dS dS d S )NrZ �indexTr] rX �Z schema_nameZ table_nameZschema_qualified_table_nameFr �r ra Z parent_namesr- r- r. � include_name s ��z@IncludeHooksTest.test_remove_connection_fk.<locals>.include_name�Z name_filtersr r! r# r �r r r r �append_constraintr �cr r rY r& r' r �len) r) � hook_typer* r+ rW �t1rb r, rg r- r- r. �test_remove_connection_fk� s\ ���� ��� z*IncludeHooksTest.test_remove_connection_fkc C s� t � }t � }td|tdtdd�� td|tdt�tdt�� td|tdtdd��}td|tdt�tdt��}|�t|jjg|jjgdd �� |�t|jj g|jjgd d �� d d � }| j |||d �}| j |d dddgddgd d � t t |�d� d S )NrW r Tr rX r rY rZ r< r[ c S s" t | t�o |dko | o |dk S r\ r^ r` r- r- r. rb K s ���z=IncludeHooksTest.test_add_metadata_fk.<locals>.include_objectrc r r0 r ri )r) r* r+ rW �t2rb r, r- r- r. �test_add_metadata_fk) sT �� ���� �z%IncludeHooksTest.test_add_metadata_fkc C s$ t � }t � }td|tdtdd��}td|tdtdd�tdtdd�� td|tdt�td t�td t��}|�t|jjg|jjgd d �� |�t|jj g|jjgd d �� td|tdtdd�� td|tdtdd�tdtdd��}td|tdt�td t�td t��}|�t|jj|jj g|jj|jj gd d �� |�t|jj |jj g|jj|jj gd d �� |dk�r|dd� }| j |||d�} n"|dk�r�dd� } | j ||| d�} |dk�r�| j | d ddd gddgd d � | j | d ddd d gdddgd d � tt| �d� n$|dk�r tdd � | D �d d d h� d S ) NZref_a�aTr Zref_b�brX r rY �zrZ r< r[ rV c S s t | t�o|dko|dk S r\ r^ r` r- r- r. rb � s ��z7IncludeHooksTest.test_change_fk.<locals>.include_objectrc r c S s< |dkr dS | dkr4t |d� t |d ddd�� dS dS d S )Nrd TrZ r] rX re Fr rf r- r- r. rg � s ��z5IncludeHooksTest.test_change_fk.<locals>.include_namerh r r! r r0 � c S s h | ]}|d |d j f�qS )r r r< )�.0�dr- r- r. � <setcomp>� s z2IncludeHooksTest.test_change_fk.<locals>.<setcomp>)r0 r[ )r0 rZ )r! r[ )r r r r rj r rk r rr rY rt rs r& r' r rl ) r) rm r* r+ Zr1arn Zr2brp rb r, rg r- r- r. �test_change_fkZ s� � ���� � �� �� �� �� �z IncludeHooksTest.test_change_fkN) rO rP rQ rR rS r r r= rT ro rq ry r- r- r- r. rU � s M1rU c @ s� e Zd ZdZdd� Zejjdd� �Zejjdd� �Z dd � Z ejj d d � �Z ejj d d � �Z ejjdd� �Zejjdd� �Zejjdd� �Zejjdd� �Zejjdd� �Zejjdd� �Zejjdd� �Zejjejj d d � ��Zejjd d � �Zejjd d!� �Z ejj ejjd"d#� ��Z ejj ejjd$d%� ��Z ejjejj d&d'� ��Z ejjejj d(d)� ��Z!ejjejj d*d+� ��Z"ejj d,d-� �Z#ejj$d.d/� �Z%d0S )1�AutogenerateFKOptionsTestTc C s� t � }t � }td|tdtdd�tdtd��� td|tdtdd�tdtd �d d �td t�td gd gf|�� td|tdtdd�tdtd��� td|tdtdd�tdtd �d d �td t�td gd gf|�� | �||�S )Nr r Tr r r r r r Fr �tidr2 )r r r r r r r& )r) Zold_optsZnew_optsr* r+ r- r- r. �_fk_opts_fixture� s> � � � � z*AutogenerateFKOptionsTest._fk_opts_fixturec C sV | � i ddi�}| j|d dddgddgd d d � | j|d d ddgddgdd � d S )N�ondelete�cascader r! r r{ r r r" �r} r$ r r0 �r} �r| r' �r) r, r- r- r. �test_add_ondelete� s( � �z+AutogenerateFKOptionsTest.test_add_ondeletec C sV | � ddii �}| j|d dddgddgdd d � | j|d d ddgddgd d � d S )Nr} �CASCADEr r! r r{ r r r" r r r0 r� r� r� r- r- r. �test_remove_ondelete s( � �z.AutogenerateFKOptionsTest.test_remove_ondeletec C s" | � ddiddi�}t|g � dS )�test case sensitivityr} �caSCAde�CasCadeN�r| r r� r- r- r. �test_nochange_ondelete& s �z0AutogenerateFKOptionsTest.test_nochange_ondeletec C sV | � i ddi�}| j|d dddgddgd d d � | j|d d ddgddgdd � d S )N�onupdater~ r r! r r{ r r r" �r� r$ r r0 �r� r� r� r- r- r. �test_add_onupdate- s( � �z+AutogenerateFKOptionsTest.test_add_onupdatec C sV | � ddii �}| j|d dddgddgdd d � | j|d d ddgddgd d � d S )Nr� r� r r! r r{ r r r" r� r r0 r� r� r� r- r- r. �test_remove_onupdateF s( � �z.AutogenerateFKOptionsTest.test_remove_onupdatec C s" | � ddiddi�}t|g � dS )r� r� r� r� Nr� r� r- r- r. �test_nochange_onupdate_ s �z0AutogenerateFKOptionsTest.test_nochange_onupdatec C s" | � ddiddi�}t|g � dS )�6test the RESTRICT option which MySQL doesn't report onr} �restrictNr� r� r- r- r. � test_nochange_ondelete_restrictg s �z9AutogenerateFKOptionsTest.test_nochange_ondelete_restrictc C s" | � ddiddi�}t|g � dS )r� r� r� Nr� r� r- r- r. � test_nochange_onupdate_restrictp s �z9AutogenerateFKOptionsTest.test_nochange_onupdate_restrictc C s" | � ddiddi�}t|g � dS )�<test the NO ACTION option which generally comes back as Noner} � no actionNr� r� r- r- r. � test_nochange_ondelete_noactiony s �z9AutogenerateFKOptionsTest.test_nochange_ondelete_noactionc C s" | � ddiddi�}t|g � dS )r� r� r� Nr� r� r- r- r. � test_nochange_onupdate_noaction� s �z9AutogenerateFKOptionsTest.test_nochange_onupdate_noactionc C s` | � ddiddi�}| j|d dddgdd gd tjd d � | j|d dddgdd gd dd� d S )r� r} r� r~ r r! r r{ r r Nr" �r� r} r$ r r0 �r� r} �r| r' r �ANYr� r- r- r. �"test_change_ondelete_from_restrict� s2 �� �z<AutogenerateFKOptionsTest.test_change_ondelete_from_restrictc C s` | � ddiddi�}| j|d dddgdd gtjd d d � | j|d dddgdd gdd d� d S )r� r� r� r~ r r! r r{ r r Nr" r� r r0 r� r� r� r- r- r. �"test_change_onupdate_from_restrict� s2 �� �z<AutogenerateFKOptionsTest.test_change_onupdate_from_restrictc C sb | � ddd�ddd��}| j|d dddgd d gddd d � | j|d dddgd d gddd� d S )Nr� zSET NULLr� ZRESTRICTr r! r r{ r r r" r� r r0 r� r� r- r- r. � test_ondelete_onupdate_combo� s2 �� �z6AutogenerateFKOptionsTest.test_ondelete_onupdate_comboc C sV | � i ddi�}| j|d dddgddgd d d � | j|d d ddgddgdd � d S )N� initially�deferredr r! r r{ r r r" �r� r$ r r0 �r� r� r� r- r- r. �test_add_initially_deferred� s( � �z5AutogenerateFKOptionsTest.test_add_initially_deferredc C sX | � ddii �}| j|d dddgddgd d d d � | j|d dddgddgd d� d S )Nr� r� r r! r r{ r r ZDEFERREDTr" �r� � deferrabler$ r r0 r� r� r� r- r- r. � test_remove_initially_deferred s* � �z8AutogenerateFKOptionsTest.test_remove_initially_deferredc C sZ | � i ddd��}| j|d dddgdd gd d d � | j|d d ddgdd gddd� d S )N� immediateT�r� r� r r! r r{ r r r" r� r r0 r� r� r- r- r. �,test_add_initially_immediate_plus_deferrable s0 �� �zFAutogenerateFKOptionsTest.test_add_initially_immediate_plus_deferrablec C s\ | � ddd�i �}| j|d dddgdd gd dd d � | j|d d ddgdd gd d d� d S )Nr� Tr� r r! r r{ r r r" r� r r0 r� r� r- r- r. �/test_remove_initially_immediate_plus_deferrable: s2 �� �zIAutogenerateFKOptionsTest.test_remove_initially_immediate_plus_deferrablec C s& | � ddd�ddd��}t|g � d S )NTr� �r� r� r� r� r- r- r. �*test_add_initially_deferrable_nochange_oneX s �zDAutogenerateFKOptionsTest.test_add_initially_deferrable_nochange_onec C s& | � ddd�ddd��}t|g � d S )NTr� r� r� r� r- r- r. �*test_add_initially_deferrable_nochange_twob s �zDAutogenerateFKOptionsTest.test_add_initially_deferrable_nochange_twoc C s& | � d dd�d dd��}t|g � d S )Nr� r� r� r� r- r- r. �,test_add_initially_deferrable_nochange_threel s �zFAutogenerateFKOptionsTest.test_add_initially_deferrable_nochange_threec C sV | � i ddi�}| j|d dddgddgd d d � | j|d d ddgddgdd � d S �Nr� Tr r! r r{ r r r" )r� r$ r r0 )r� r� r� r- r- r. �test_add_deferrablev s( � �z-AutogenerateFKOptionsTest.test_add_deferrablec C sV | � ddii �}| j|d dddgddgdd d � | j|d d ddgddgd d � d S r� r� r� r- r- r. �test_remove_deferrable� s( � �z0AutogenerateFKOptionsTest.test_remove_deferrableN)&rO rP rQ rR r| r r= Zfk_ondelete_is_reflectedr� r� r� Zfk_onupdate_is_reflectedr� r� Z fk_onupdater� Zfk_ondelete_restrictr� Zfk_onupdate_restrictr� Zfk_ondelete_noactionr� r� r� r� r� Z fk_initiallyr� r� Z fk_deferrabler� r� r� r� r� r� Zfk_deferrable_is_reflectedr� r- r- r- r. rz � sf & rz N)Z sqlalchemyr r r r r r Z_autogen_fixturesr Ztestingr r r r r r rU rz r- r- r- r. �<module> s$ I v