관리-도구
편집 파일: test_environment.cpython-38.pyc
U 0?�fe. � @ s� d dl Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lm Z dd lm Z dd lm Z dd l m Z G d d� de �ZdS )� N� )�MigrationContext)� assert_raises)�config)�eq_)�is_)�is_false)�is_not_)�is_true)�ne_)�TestBasec @ s� e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zejjd d � �Zejjd!d"� �Zd#d$� Zd%d&� Zd'd(� ZdS ))�MigrationTransactionTestTNc C s\ t j�� | _}|�dd�rFtj|j|d�| _t � � | j_ | jj _ ntj||d�| _| jS )N�as_sqlF)�dialect�opts)� connectionr ) r �db�connect�conn�getr � configurer �context�io�StringIO� output_buffer�impl)�selfr r � r �T/opt/hc_python/lib/python3.8/site-packages/alembic/testing/suite/test_environment.py�_fixture s �� �z!MigrationTransactionTest._fixturec C s | j r| j �� d S )N)r �close)r r r r �teardown_method$ s z(MigrationTransactionTest.teardown_methodc C sR | � ddd��}t| j�� � |jdd�}t| j�� � |�� t| j�� � d S �NT��transaction_per_migration�transactional_ddl�Z_per_migration)r r r �in_transaction�begin_transactionr �rollback�r r �proxyr r r � test_proxy_transaction_rollback( s � z8MigrationTransactionTest.test_proxy_transaction_rollbackc C sD | � ddd��}|jdd�}t| j�� � |�� t| j�� � d S r" �r r( r r r' �commitr r* r r r � test_proxy_transaction_commit3 s � z6MigrationTransactionTest.test_proxy_transaction_commitc C sL | � ddd��}|jdd�}t| j�� � |� W 5 Q R X t| j�� � d S r" )r r( r r r' r r* r r r �,test_proxy_transaction_contextmanager_commit< s � zEMigrationTransactionTest.test_proxy_transaction_contextmanager_commitc sR | � ddd��}|jdd�� t| j�� � � fdd�}tt|� t| j�� � d S )NTr# r& c s � � t d��W 5 Q R X d S )N�hi)� Exceptionr �r+ r r �goM s zSMigrationTransactionTest.test_proxy_transaction_contextmanager_rollback.<locals>.go)r r( r r r' r r2 r )r r r4 r r3 r �.test_proxy_transaction_contextmanager_rollbackF s � zGMigrationTransactionTest.test_proxy_transaction_contextmanager_rollbackc C sp | � ddd��}|jdd�}t| j�� � |�* t| j�� � |�� t| j�� � W 5 Q R X t| j�� � d S r" )r r( r r r' r) r r* r r r �7test_proxy_transaction_contextmanager_explicit_rollbackT s � zPMigrationTransactionTest.test_proxy_transaction_contextmanager_explicit_rollbackc C sp | � ddd��}|jdd�}t| j�� � |�* t| j�� � |�� t| j�� � W 5 Q R X t| j�� � d S r" r- r* r r r �5test_proxy_transaction_contextmanager_explicit_commitb s � zNMigrationTransactionTest.test_proxy_transaction_contextmanager_explicit_commitc C s� | � ddd��}t| j�� � |�� �H t| j�� � |jdd�� t| j�� � W 5 Q R X t| j�� � W 5 Q R X t| j�� � d S r" �r r r r' r( r �r r r r r �0test_transaction_per_migration_transactional_ddlp s � zIMigrationTransactionTest.test_transaction_per_migration_transactional_ddlc C s� | � ddd��}t| j�� � |�� �H t| j�� � |jdd�� t| j�� � W 5 Q R X t| j�� � W 5 Q R X t| j�� � d S )NTFr# r& r8 r9 r r r �4test_transaction_per_migration_non_transactional_ddl s � zMMigrationTransactionTest.test_transaction_per_migration_non_transactional_ddlc C s� | � ddi�}t| j�� � |�� �H t| j�� � |jdd�� t| j�� � W 5 Q R X t| j�� � W 5 Q R X t| j�� � d S )Nr% Tr& r8 r9 r r r �*test_transaction_per_all_transactional_ddl� s zCMigrationTransactionTest.test_transaction_per_all_transactional_ddlc C s� | � ddi�}t| j�� � |�� �H t| j�� � |jdd�� t| j�� � W 5 Q R X t| j�� � W 5 Q R X t| j�� � d S )Nr% FTr& r8 r9 r r r �.test_transaction_per_all_non_transactional_ddl� s zGMigrationTransactionTest.test_transaction_per_all_non_transactional_ddlc C s� | � ddi�}|�d� |�� �< |�d� |jdd�� |�d� W 5 Q R X |�d� W 5 Q R X |�d� |jjr�| �dd dddd d� n| �ddddd� d S ) Nr T�step 1�step 2r& �step 3�step 4�step 5�BEGIN�COMMIT�r �executer( r r% �_assert_impl_stepsr9 r r r � test_transaction_per_all_sqlmode� s2 � �z9MigrationTransactionTest.test_transaction_per_all_sqlmodec C s� | � ddd��}|�d� |�� �< |�d� |jdd�� |�d� W 5 Q R X |�d� W 5 Q R X |�d� |jjr�| �ddd dd dd� n| �ddddd� d S ) NT)r r$ r> r? r& r@ rA rB rC rD rE r9 r r r �&test_transaction_per_migration_sqlmode� s6 � � �z?MigrationTransactionTest.test_transaction_per_migration_sqlmodec C s | � ddi�}t| j�� � |�� �� t| j�� � |jdd��� t| j�� � |�� �J | jrpt|j | j� n t |j | j� t| j�� � t |j j d d� W 5 Q R X t |j j �dd �d� t| j�� � W 5 Q R X t| j�� � W 5 Q R X t| j�� � d S )Nr$ Tr& �isolation_level� AUTOCOMMIT)r r r r' r( r �autocommit_block�is_sqlalchemy_futurer r r r �_execution_optionsr r r9 r r r �test_autocommit_block� s6 �� ��z.MigrationTransactionTest.test_autocommit_blockc C s� | � ddi�}t| j�� � |�� �X t|j�� � | jrJt|j| j� n t |j| j� t| j�� � t |jj d d� W 5 Q R X t |jj � dd �d� t| j�� � d S )Nr$ TrJ rK )r r r r' rL r r rM r r r rN r r r9 r r r �$test_autocommit_block_no_transaction s" ��z=MigrationTransactionTest.test_autocommit_block_no_transactionc C s� | � dddd��}|�� �d |�d� |jdd��8 |�d� |�� � |�d� W 5 Q R X |�d� W 5 Q R X |�d� W 5 Q R X | �dd dd dd dd d� d S ) NT�r$ r% r r> r& r? r@ rA rB rC rD �r r( rF rL rG r9 r r r �/test_autocommit_block_transactional_ddl_sqlmode& s2 �� �zHMigrationTransactionTest.test_autocommit_block_transactional_ddl_sqlmodec C s� | � dddd��}|�� �d |�d� |jdd��8 |�d� |�� � |�d� W 5 Q R X |�d� W 5 Q R X |�d � W 5 Q R X | �ddddd � d S ) NTFrQ r> r& r? r@ rA rB rR r9 r r r �2test_autocommit_block_nontransactional_ddl_sqlmodeG s* �� �zKMigrationTransactionTest.test_autocommit_block_nontransactional_ddl_sqlmodec G st | j j�� }t�� | j j_}|D ]>}|dkr<| j j�� q"|dkrR| j j�� q"| j j�|� q"t ||�� � d S )NrC rD ) r r �getvaluer r r Z emit_beginZ emit_commit�_execr )r ZstepsZto_check�buf�stepr r r rG ` s z+MigrationTransactionTest._assert_impl_steps)�__name__� __module__� __qualname__Z __backend__r r r! r, r/ r0 r5 r6 r7 r: r; r< r= rH rI r � requirementsZautocommit_isolationrO rP rS rT rG r r r r r s. ' !r )r Z migrationr Ztestingr r r r r r r r Ztesting.fixturesr r r r r r �<module> s