관리-도구
편집 파일: base.cpython-38.pyc
U -?�f�/ � @ s� d dl mZ d dlZddlmZ ddlmZ ddlmZ ddlm Z dd lm Z dd lm Z dd lm Z dd lm Z dd lmZ ddlmZ ddlmZ ddlmZ e�� G dd� d��ZdaG dd� d�ZdS )� )� annotationsN� )� assertions)�config)�eq_)� drop_all_tables_from_metadata� )�Column)�func)�Integer)�select)�Table)�DeclarativeBase)�MappedAsDataclass)�registryc @ s e Zd ZdZdZdZdZdZd$dd�Ze � � dd� �Z e � � dd � �Z e � � d d � �Z e � � d d � �Ze � � dd� �Ze j dd� �Ze j dd� �Ze � � dd� �Ze � � dd� �Ze � � dd� �Ze � � dd� �Ze � � d d � �Ze j d d � d D �d!�d"d#� �ZdS )%�TestBase� NFc C s |s t |��d S �N)�AssertionError)�self�val�msgr r �P/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/fixtures/base.py�assert_0 s zTestBase.assert_c c s" t jj}d t j_d V |t j_d S r )r �dbZ_compiled_cache)r �_cacher r r �nocache3 s zTestBase.nocachec c s0 t | dd �ptj}|�� � }|V W 5 Q R X d S �N�bind)�getattrr r �connect)r �eng�connr r r �connection_no_trans: s z TestBase.connection_no_transc c sJ t | dd �ptj}|�� }|�� }|a|V d a|jr>|�� |�� d S r ) r r r r �begin� _connection_fixture_connectionZ is_active�rollback�close)r r! r" �transr r r � connectionA s zTestBase.connectionc # sj g � g �d� �fdd� }|V �D ] }z |� � W q" Y q"X q"� D ] }z |�� W qF Y qFX qFd S )NFc s � � | � |r�� | � d S r )�append)�result�consume��to_closeZ to_consumer r �go[ s z/TestBase.close_result_when_finished.<locals>.go)F)�allr' )r r/ �rr r- r �close_result_when_finishedV s z#TestBase.close_result_when_finishedc c s8 t |tt�� �t�d�ddd�id�}|V |�� d S )N�2 �mysql�mariadb�oracle)�metadata�type_annotation_map)r �str�sa�String� with_variant�dispose)r r7 �regr r r r l s ���zTestBase.registryc # s, |� G � fdd�dt �}|V |j�� d S )Nc s. e Zd Z� Zee�� �e�d�ddd�iZdS )z TestBase.decl_base.<locals>.Baser3 r4 r5 r6 N� �__name__� __module__� __qualname__r7 r9 r: r; r< r8 r �Z_mdr r �Base} s ��rD )r r r= �r r7 rD r rC r � decl_basey s zTestBase.decl_basec # s. |� G � fdd�dt t�}|V |j�� d S )Nc s, e Zd Z� Zee�� �e�d�dd�iZdS )z#TestBase.dc_decl_base.<locals>.Baser3 r4 r5 Nr? r rC r r rD � s ��rD )r r r r= rE r rC r � dc_decl_base� s zTestBase.dc_decl_basec c s |V d S r r )r � future_enginer) r r r �future_connection� s zTestBase.future_connectionc c s d V d S r r )r r r r rH � s zTestBase.future_enginec # s0 ddl m� d� fdd� }|V � j�d� d S )Nr ��enginesFc s( |d kr i }d|d< � j | ||||d�S )N�fixture�scope)�url�options�asyncio�transfer_staticpool� share_pool��testing_engine)rN rO �futurerP rQ rR rJ r r �gen_testing_engine� s �z3TestBase.testing_engine.<locals>.gen_testing_enginerL )NNNFFF)� rK Ztesting_reaperZ_drop_testing_engines)r rV r rJ r rT � s �zTestBase.testing_enginec s � fdd�}|S )Nc s d| d< � f | �S )NTrP r )�kwrS r r r/ � s z)TestBase.async_testing_engine.<locals>.gor )r rT r/ r rS r �async_testing_engine� s z TestBase.async_testing_enginec c st ddl m} |�� }||j_|V |j`trdt�� rdt�� }|�� t� � � t |t� W 5 Q R X n t |t j � dS )z>Provide bound MetaData for a single test, dropping afterwards.r )�schemaN) �sqlrZ ZMetaData�instancer7 r% Zin_transactionZget_transactionr& r$ r r r )r �requestrZ r7 r( r r r r7 � s" �� �zTestBase.metadatac C s* g | ]"}d D ]}dD ]}|||f�qq qS ))�none�executer$ �TFr )�.0r& �second_operation� begin_nestedr r r � <listcomp>� s �zTestBase.<listcomp>r` )�paramsc sP |j \��� td|tdt���t| dd �p.tj}��|� � ���fdd�}|S )N�test�datar c s� | � � ��P}� �r@tjjjs&t�d� |r4| �� }n|�� }|�� |r\| ���� ddi� n|���� ddi� �r~|� � n|� � �dk�rt � t jjd��\ �dkr�|r�| ���� ddi� q�|���� ddi� n �dkr�|r�| �� n|�� W 5 Q R X W 5 Q R X |�r(| ���� dd i� n|���� dd i� �n|�r\| ���� ddi� n|���� ddi� |�r���r�| � � n| � � n��r�|� � n|� � �dk�rVt � t jjd��� �dk�r |�r�| ���� ddi� n|���� ddi� nL�dk�r*t|d��r |� � n| � � n"�d k�rL|�rD| �� n|�� W 5 Q R X W 5 Q R X d }� �rr|d 7 }��s�|d 7 }|�r�t| �tt�� �����|� n2| �� �$}t|�tt�� �����|� W 5 Q R X d S ) Nzsavepoints not enabledrg � r^ z�Can't operate on closed transaction inside context manager. Please complete the context manager before emitting further commands.r_ � r$ � rc r � )r$ r � requirementsZ savepoints�enabledZ skip_testrc r_ �insertr& �commitr Zexpect_raises_messager: �excZInvalidRequestError�hasattrr Zscalarr r �count� select_fromr )�subjectZtrans_on_subjectZexecute_on_subjectr( Z nested_transZexpected_committedr" �rc r& rb �tr r �run_test� s� � � � � �z4TestBase.trans_ctx_manager_fixture.<locals>.run_test)�paramr r r r r r �create)r r] r7 r! rw r ru r �trans_ctx_manager_fixture� s wz"TestBase.trans_ctx_manager_fixture)N)r@ rA rB � __requires__Z__unsupported_on__Z __only_on__Z __skip_if__Z"__leave_connections_for_teardown__r r rL r r# r) r2 r rF rG rI rH rT rY r7 rz r r r r r sH �� r c @ s e Zd ZdZdS )�FutureEngineMixinz alembic's suite still using thisN)r@ rA rB �__doc__r r r r r| m s r| )� __future__r Z sqlalchemyr: rW r r r �utilr r r r r r Zormr r r Zmark_base_test_classr r% r| r r r r �<module> s& P