관리-도구
편집 파일: test_autogen_computed.cpython-38.pyc
U 0?�f � @ s� d dl Zd dl mZ d dl mZ d dl mZ d dl mZ ddlmZ dd lm Z dd l m Z dd l m Z dd l m Z dd l m Z ddl mZ ddl mZ ddl mZ G dd� dee�ZdS )� N)�Column)�Integer)�MetaData)�Table� )�AutogenFixtureTest� )�testing)�config)�eq_)� exclusions)�is_)�is_true)�mock)�TestBasec @ s� e Zd ZdZdZdd� Zdd� Ze�dd� d d� d d� d d� �e j j d d � ��Z e�dd� dd� dd� dd� dd� e j j f�dd� �Ze j jdd� �ZdS )�AutogenerateComputedTest)Zcomputed_columnsTc C s� t � }t � }td|tdtdd�� td|tdtdd�tdtt�d��� | �||�}t|d d d� t|d d d� t|d d jd� |d d j }t t |tj�� t |j d � tt|j�d� d S ) N�user�idT�Z primary_key�foo�5r � add_column� r )r r r r �sa�Computed�_fixturer �name�computedr � isinstancer � persisted�strZsqltext��self�m1�m2�diffs�c� r' �Y/opt/hc_python/lib/python3.8/site-packages/alembic/testing/suite/test_autogen_computed.py�test_add_computed_column s" � z1AutogenerateComputedTest.test_add_computed_columnc C s t � }t � }td|tdtdd�tdtt�d��� td|tdtdd�� | �||�}t|d d d� t|d d d� |d d }t|jd� t j j j r�t t|jtj�� n t|jd � t j jj r�t t|jtj�� tt|jjj�d� n,t j j j �r t t|jtj�� n t|jd � d S ) Nr r Tr r r r Z remove_columnr r )r r r r r r r r r r � requirements�computed_reflects_normally�enabledr r r r �#computed_reflects_as_server_defaultZserver_default� DefaultClauser �arg�textr! r' r' r( �test_remove_computed_column. s. � z4AutogenerateComputedTest.test_remove_computed_columnc C s d t �d�fS �N�bar*5�r r r' r' r' r( �<lambda>P � z!AutogenerateComputedTest.<lambda>c C s t �d�d fS r2 r4 r' r' r' r( r5 Q r6 c C s t �d�t jddd�fS )Nr3 �bar * 42T)r r4 r' r' r' r( r5 R s �c C s t �d�t �d�fS )Nr3 r7 r4 r' r' r' r( r5 V r6 c C s� t j|ft� �\}}t� }t� }|d kr,g n|g}|d kr>g n|g}td|tdtdd�tdt�tdtf|�� � td|tdtdd�tdt�tdtf|�� � t�d��}| � ||�}W 5 Q R X t |j t� d�g� t t |�g � d S ) Nr r Tr �barr �alembic.util.warnz/Computed default on user.foo cannot be modified)r �resolve_lambda�localsr r r r r �patchr r � mock_calls�call�list�r" Z test_caseZ arg_beforeZ arg_afterr# r$ Z mock_warnr% r' r' r( �!test_cant_change_computed_warningO s4 � � �z:AutogenerateComputedTest.test_cant_change_computed_warningc C s dS )N)NNr' r' r' r' r( r5 | r6 c C s t �d�t �d�fS )Nr r4 r' r' r' r( r5 } r6 c C s t �d�t �d�fS r2 r4 r' r' r' r( r5 ~ r6 c C s t �d�t �d�fS )Nr3 z bar * 5r4 r' r' r' r( r5 r6 c C s t �d�d fS r2 r4 r' r' r' r( r5 � r6 c C s� t j|ft� �\}}t� }t� }|d kr,g n|g}|d kr>g n|g}td|tdtdd�tdt�tdtf|�� � td|tdtdd�tdt�tdtf|�� � t�d��}| � ||�}W 5 Q R X t |j g � t t |�g � d S )Nr r Tr r8 r r9 ) r r: r; r r r r r r< r r r= r? r@ r' r' r( �test_computed_unchanged{ s. � � z0AutogenerateComputedTest.test_computed_unchangedc C s. t � }t � }td|tdtdd�tdt�tdtt�d��� td|tdtdd�tdt�tdt�� | �||�}t|d d d d � t|d d d d� t|d d d d� |d d d }|d d d }t|d� t t |tj �� t � td��r tt|jj�d� n t � td��r*tt|jj�d� dS )z�Asserts the current behavior which is that on PG and Oracle, the GENERATED ALWAYS AS is reflected as a server default which we can't tell is actually "computed", so these come out as a modification to the server default. r r Tr r8 r zbar + 42r Zmodify_defaultr r ������N� postgresqlz (bar + 42)�oraclez"BAR"+42)r r r r r r r r r r r r. r Zagainstr r r/ r0 )r" r# r$ r% �old�newr' r' r( �(test_remove_computed_default_on_computed� s8 � � zAAutogenerateComputedTest.test_remove_computed_default_on_computedN)�__name__� __module__� __qualname__� __requires__Z __backend__r) r1 r � combinationsr r* r+ rA Z)computed_doesnt_reflect_as_server_defaultrB r- rI r' r' r' r( r s0 !� "�� r )Z sqlalchemyr r r r r Z_autogen_fixturesr � r r r r r r r r r r' r' r' r( �<module> s