관리-도구
편집 파일: orm.cpython-38.pyc
U -?�f� � @ s d dl mZ d dlmZ d dlZddlmZ ddlm Z ddl m Z dd l m Z dd l m Z dd lmZ dd lmZ dd lmZ ddl mZ ddlmZ ddlmZ ddlmZ G dd� de�ZG dd� dee e j�ZG dd� de�ZG dd� d�Z e � Z dd � Z d d � Z d d � Z!d!d"� Z"dS )#� )� annotations)�AnyN� )�TestBase)� TablesTest� )� assertions)�config)�schema)� BasicEntity)�ComparableEntity)�adict� )�orm)�DeclarativeBase)�events)�registryc @ s e Zd Zejdd� �ZdS )�ORMTestc C s t � S �N)�fixture_session��self� r �O/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/fixtures/orm.pyr s zORMTest.fixture_sessionN)�__name__� __module__� __qualname__r �fixturer r r r r r s r c @ s� e Zd ZU dZdZdZded< ejddd�d d � �Z ejdd d�d d � �Z e dd� �Z e dd� �Z e dd� �Zdd� Zdd� Ze dd� �Ze dd� �Zd d � Zd d � Ze d d!� �Ze d"d#� �ZdS )$� MappedTest�once�eachNr �classesT�class)�autouse�scopec c sZ | j }|�� |jd kr t� |_|�� |�� |�� |�� d V |�� |� � d S r ) � __class__Z _init_classr! r �_setup_once_tables�_setup_once_classes�_setup_once_mappersZ_setup_once_inserts�_teardown_once_classZ _teardown_once_metadata_bind)r �clsr r r �_setup_tables_test_class* s z#MappedTest._setup_tables_test_class�functionc c sL | � � | �� | �� | �� d V tj�� | �� | �� | � � d S r ) Z_setup_each_tables�_setup_each_classes�_setup_each_mappersZ_setup_each_insertsr �session�close_all_sessions�_teardown_each_mappers�_teardown_each_classesZ_teardown_each_tablesr r r r �_setup_tables_test_instance< s z&MappedTest._setup_tables_test_instancec C s | j �� d S r )r! �clear�r* r r r r) J s z MappedTest._teardown_once_classc C s | j dkr| �| j� d S �Nr ��run_setup_classes�_with_register_classes� setup_classesr5 r r r r' N s z MappedTest._setup_once_classesc C s* | j dkr&| �� \| _| _| �| j� d S r6 )�run_setup_mappers�_generate_registry�mapper_registry�mapperr9 � setup_mappersr5 r r r r( S s z MappedTest._setup_once_mappersc C s8 | j dkr | �� \| j_| j_| j dkr4| �| j� d S )Nr r )r; r< r% r= r> r9 r? r r r r r. Y s � z MappedTest._setup_each_mappersc C s | j dkr| �| j� d S )Nr r7 r r r r r- c s z MappedTest._setup_each_classesc C s t | jd�}||jfS )N)�metadata)r �_tables_metadataZmap_imperatively)r* �declr r r r< g s z MappedTest._generate_registryc sR | j � G � fdd�d�}G dd� dt|�}G dd� dt|�}|| _|| _|� dS )z�Run a setup method, framing the operation with a Base class that will catch new subclasses to be established within the "classes" registry. c s$ e Zd Zdd�� �fdd� Z� ZS )z0MappedTest._with_register_classes.<locals>._Base�None��returnc s$ �d k s t �| �| j< t� �� d S r ��AssertionErrorr �super�__init_subclass__r5 �r% � cls_registryr r rI v s zBMappedTest._with_register_classes.<locals>._Base.__init_subclass__)r r r rI � __classcell__r �rK �r% r �_Baseu s rO c @ s e Zd ZdS )z0MappedTest._with_register_classes.<locals>.BasicN�r r r r r r r �Basic{ s rQ c @ s e Zd ZdS )z5MappedTest._with_register_classes.<locals>.ComparableNrP r r r r � Comparable~ s rR N)r! r r rQ rR )r* �fnrO rQ rR r rM r r9 l s z!MappedTest._with_register_classesc C s | j dkrt�� d S r6 )r; r � clear_mappersr r r r r1 � s z!MappedTest._teardown_each_mappersc C s | j dkr| j�� d S r6 )r8 r! r4 r r r r r2 � s z!MappedTest._teardown_each_classesc C s d S r r r5 r r r r: � s zMappedTest.setup_classesc C s d S r r r5 r r r r? � s zMappedTest.setup_mappers)r r r r8 r; r! �__annotations__r r r+ r3 � classmethodr) r' r( r. r- r< r9 r1 r2 r: r? r r r r r ! s2 r c s4 e Zd ZdZdZedd� �Ze� fdd��Z� ZS )�DeclarativeMappedTestr c C s d S r r r5 r r r r& � s z(DeclarativeMappedTest._setup_once_tablesc sN � j �G � �fdd�dt�}|� _t� �|� � jjrJ� jrJ� j�t j � d S )Nc sN e Zd ZejZ�jZee � � � e � d�ddd�iZ dd�� �fdd� Z � ZS ) z?DeclarativeMappedTest._with_register_classes.<locals>._DeclBase�2 �mysqlZmariadb�oraclerC rD c s( �d k s t �| �| j< t� jf |� d S r rF )r* �kwrJ r r rI � s zQDeclarativeMappedTest._with_register_classes.<locals>._DeclBase.__init_subclass__)r r r r �TableZ __table_cls__rA r@ �str�sa�StringZ with_variantZtype_annotation_maprI rL r �r* rK rN r � _DeclBase� s ��ra ) r! r ZDeclarativeBasicrH r9 rA ZtablesZrun_create_tablesZ create_allr �db)r* rS ra rN r` r r9 � s z,DeclarativeMappedTest._with_register_classes) r r r r8 r; rV r& r9 rL r r rN r rW � s rW c @ s e Zd Zejdd�dd� �ZdS )�RemoveORMEventsGloballyT)r# c c s< d V t j�� t j�� t j�� t j�� t j�� d S r )� orm_eventsZ MapperEvents�_clearZInstanceEventsZ SessionEventsZInstrumentationEventsZ QueryEventsr r r r �_remove_listeners� s z)RemoveORMEventsGlobally._remove_listenersN)r r r r r rf r r r r rc � s rc c K sB | � dd� | � dd� | �dtj�}tj|f| �}t�|� |S )NZ autoflushTZexpire_on_commit�bind)� setdefault�popr rb r �Session�_fixture_sessions�add)r[ rg �sessr r r r � s r c C s t �� t�� d S r )r r0 rk r4 r r r r r0 � s r0 c C s t � t�� d S r )r0 r rT r5 r r r � stop_test_class_inside_fixtures� s rn c C s t r t� d S r )rk r0 r r r r � after_test� s ro )#� __future__r �typingr Z sqlalchemyr^ �baser �sqlr � r r r �entitiesr r �utilr r r r rd r r ZAssertsExecutionResultsr rW rc �setrk r r0 rn ro r r r r �<module> s. x$