관리-도구
편집 파일: fixtures.cpython-37.pyc
B ��4]>* � @ s d dl Z d dlZddlmZ ddlmZ ddlmZ ddlmZ ddl m Z ddl m Z dd l m Z d d lmZ d d lm Z d d lmZ d dlmZ G dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� dee�ZG dd� deeej�ZG dd � d e�ZdS ) � N� )� assertions)�config)�schema)�drop_all_tables)� BasicEntity)�ComparableEntity)�adict� )�event)�util)�declarative_base)�DeclarativeMetac @ s: e Zd ZdZdZdZdZdZd dd�Zdd� Z dd� Z dS ) �TestBase� Nc C s |s t |��d S )N)�AssertionError)�self�val�msgr r �N/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/fixtures.py�assert_/ s zTestBase.assert_c C s t | d�r| �� d S )N�setUp)�hasattrr )r r r r �setup3 s zTestBase.setupc C s t | d�r| �� d S )N�tearDown)r r )r r r r �teardown7 s zTestBase.teardown)N) �__name__� __module__� __qualname__Z __whitelist__� __requires__Z__unsupported_on__Z __only_on__Z __skip_if__r r r r r r r r s r c @ s� e Zd ZdZdZdZdZdZdZdZ dZ dZ dZ e dd� �Ze dd� �Ze dd � �Ze d d � �Zd d � Zdd� Zdd� Zdd� Zdd� Ze dd� �Ze dd� �Ze dd� �Ze d d � �Ze d d � �Ze d d!� �Z e d"d#� �Z d$d%� Z d&d'� Z e d(d)� �Z dS )*� TablesTest�once�eachNc C s | � � | �� | �� d S )N)� _init_class�_setup_once_tables�_setup_once_inserts)�clsr r r � setup_classU s zTablesTest.setup_classc C sZ | j dkr(| jdkrd| _| jdks(t�t� | _t� | _| �� | _t � � | _ | j| j _d S )Nr" r! )r" N) �run_define_tables�run_create_tables� run_insertsr r �other�tables� setup_bind�bind�saZMetaData�metadata)r& r r r r# ] s zTablesTest._init_classc C s | j dkr| �� | �� d S )Nr! )r* �_load_fixtures� insert_data)r& r r r r% k s z TablesTest._setup_once_insertsc C sB | j dkr>| �| j� | jdkr.| j�| j� | j�| jj� d S )Nr! )r( � define_tablesr0 r) � create_allr. r, �update)r& r r r r$ q s z TablesTest._setup_once_tablesc C s� | j dkrl| j�� | jdkr,t| j| j� | j�� | �| j� | jdkrZ| j�| j� | j� | jj� n&| jdkr�t| j| j� | j�| j� d S )Nr" ) r( r, �clearr) r r0 r. r3 r4 r5 )r r r r �_setup_each_tablesy s z TablesTest._setup_each_tablesc C s | j dkr| �� | �� d S )Nr" )r* r1 r2 )r r r r �_setup_each_inserts� s z TablesTest._setup_each_insertsc C s� | j dkr�| jdkr�| j�� �n}xft| jj�D ]V}y|�|�� � W q. t j j k r� } z t j d||f tjd� W d d }~X Y q.X q.W W d Q R X d S )Nr" zError emptying table %s: %r)�file)r( � run_deletesr. �connect�reversedr0 � sorted_tables�execute�deleter/ �excZ DBAPIErrorr �print_�sys�stderr)r Zconn�table�exr r r �_teardown_each_tables� s z TablesTest._teardown_each_tablesc C s | � � | �� d S )N)r7 r8 )r r r r r � s zTablesTest.setupc C s | � � d S )N)rF )r r r r r � s zTablesTest.teardownc C sF | j rt| j| j� | jdkr*| �| j� d | j_| jd k rBd | _d S )Nr! )r) r r0 r. �run_dispose_bind� dispose_bind�run_setup_bind)r& r r r � _teardown_once_metadata_bind� s z'TablesTest._teardown_once_metadata_bindc C s | � � d S )N)rJ )r& r r r �teardown_class� s zTablesTest.teardown_classc C s t jS )N)r �db)r& r r r r- � s zTablesTest.setup_bindc C s* t |d�r|�� nt |d�r&|�� d S )N�dispose�close)r rM rN )r& r. r r r rH � s zTablesTest.dispose_bindc C s d S )Nr )r& r0 r r r r3 � s zTablesTest.define_tablesc C s i S )Nr )r& r r r �fixtures� s zTablesTest.fixturesc C s d S )Nr )r& r r r r2 � s zTablesTest.insert_datac C s | � | j||� d S )N)Zassert_sql_countr. )r �count�fnr r r � sql_count_� s zTablesTest.sql_count_c C s | � | j||� d S )N)Z assert_sqlr. )r Z callable_Z statementsr r r �sql_eq_� s zTablesTest.sql_eq_c s� i i � }xX| � � �� D ]H\�}t|�dk r.qt�tj�rD| j� �|d � �< |dd� |�< qW x@| jjD ]4��� kr|qn| j � �� � � �fdd�|� D �� qnW dS )z4Insert rows as represented by the fixtures() method.r r r Nc s g | ]}t t� � |���qS r )�dict�zip)�.0Z column_values)�headersrD r r � <listcomp>� s z-TablesTest._load_fixtures.<locals>.<listcomp>) rO �items�len� isinstancer � string_typesr, r0 r= r. r> �insert)r& Zrows�datar )rW rD r r1 � s zTablesTest._load_fixtures)!r r r rI r( r) r* r: rG r. r0 r, r+ � classmethodr' r# r% r$ r7 r8 rF r r rJ rK r- rH r3 rO r2 rR rS r1 r r r r r < s: r c s2 e Zd Zejdd� �Zdd� Z� fdd�Z� ZS )� RemovesEventsc C s t � S )N)�set)r r r r � _event_fns� s zRemovesEvents._event_fnsc K s( | j �|||f� tj|||f|� d S )N)rb �addr �listen)r �target�namerQ �kwr r r � event_listen� s zRemovesEvents.event_listenc s: x| j D ]}tj|� qW tt| �}t|d�r6|�� d S )Nr )rb r �remove�superr` r r )r �keyZsuper_)� __class__r r r � s zRemovesEvents.teardown) r r r r Zmemoized_propertyrb rh r � __classcell__r r )rl r r` � s r` c @ s e Zd Zedd� �ZdS )�_ORMTestc C s t jj�� t j�� d S )N)r/ �orm�session�close_all_sessions� clear_mappers)r& r r r rK � s z_ORMTest.teardown_classN)r r r r_ rK r r r r rn � s rn c @ s e Zd ZdS )�ORMTestN)r r r r r r r rs � s rs c @ s� e Zd ZdZdZdZedd� �Zedd� �Zdd � Z d d � Z ed d � �Z edd� �Z edd� �Z dd� Zdd� Zedd� �Zdd� Zdd� Zed d � �Zed d � �ZdS ) � MappedTestr! r" Nc C s> | � � | jd krt� | _| �� | �� | �� | �� d S )N)r# �classesr r$ �_setup_once_classes�_setup_once_mappersr% )r& r r r r' s zMappedTest.setup_classc C s | � � | �� d S )N)�_teardown_once_classrJ )r& r r r rK s zMappedTest.teardown_classc C s$ | � � | �� | �� | �� d S )N)r7 �_setup_each_classes�_setup_each_mappersr8 )r r r r r s zMappedTest.setupc C s( t jj�� | �� | �� | �� d S )N)r/ ro rp rq �_teardown_each_mappers�_teardown_each_classesrF )r r r r r ! s zMappedTest.teardownc C s | j �� t�� d S )N)ru r6 rn rK )r& r r r rx ' s z MappedTest._teardown_once_classc C s | j dkr| �| j� d S )Nr! )�run_setup_classes�_with_register_classes� setup_classes)r& r r r rv , s z MappedTest._setup_once_classesc C s | j dkr| �| j� d S )Nr! )�run_setup_mappersr~ � setup_mappers)r& r r r rw 1 s z MappedTest._setup_once_mappersc C s | j dkr| �| j� d S )Nr" )r� r~ r� )r r r r rz 6 s z MappedTest._setup_each_mappersc C s | j dkr| �| j� d S )Nr" )r} r~ r )r r r r ry : s z MappedTest._setup_each_classesc sl | j � G � fdd�dt�}G dd� dt�|t��}G dd� dt|�}G dd� dt|�}|| _|| _|� d S ) 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 Z� fdd�ZdS )z6MappedTest._with_register_classes.<locals>.FindFixturec s | � |< t �| |||�S )N)�type�__init__)r& � classname�bases�dict_)� cls_registryr r r� H s z?MappedTest._with_register_classes.<locals>.FindFixture.__init__N)r r r r� r )r� r r � FindFixtureG s r� c @ s e Zd ZdS )z0MappedTest._with_register_classes.<locals>._BaseN)r r r r r r r �_BaseL s r� c @ s e Zd ZdS )z0MappedTest._with_register_classes.<locals>.BasicN)r r r r r r r �BasicO s r� c @ s e Zd ZdS )z5MappedTest._with_register_classes.<locals>.ComparableN)r r r r r r r � ComparableR s r� N) ru r� r �with_metaclass�objectr r r� r� )r& rQ r� r� r� r� r )r� r r~ > s z!MappedTest._with_register_classesc C s | j dkrtj�� d S )Nr! )r� r/ ro rr )r r r r r{ Y s z!MappedTest._teardown_each_mappersc C s | j dkr| j�� d S )Nr! )r} ru r6 )r r r r r| ` s z!MappedTest._teardown_each_classesc C s d S )Nr )r& r r r r d s zMappedTest.setup_classesc C s d S )Nr )r& r r r r� h s zMappedTest.setup_mappers)r r r r} r� ru r_ r' rK r r rx rv rw rz ry r~ r{ r| r r� r r r r rt s" rt c @ s, e Zd ZdZdZedd� �Zedd� �ZdS )�DeclarativeMappedTestr! c C s d S )Nr )r& r r r r$ q s z(DeclarativeMappedTest._setup_once_tablesc sf | j � G � fdd�dt�}G dd� dt�}t| j||d�}|| _|� | jjrb| jrb| j�t j � d S )Nc s e Zd Z� fdd�ZdS )zLDeclarativeMappedTest._with_register_classes.<locals>.FindFixtureDeclarativec s | � |< t �| |||�S )N)r r� )r& r� r� r� )r� r r r� z s zUDeclarativeMappedTest._with_register_classes.<locals>.FindFixtureDeclarative.__init__N)r r r r� r )r� r r �FindFixtureDeclarativey s r� c @ s e Zd ZejZdS )zFDeclarativeMappedTest._with_register_classes.<locals>.DeclarativeBasicN)r r r r ZTableZ __table_cls__r r r r �DeclarativeBasic~ s r� )r0 � metaclassr& ) ru r r� r r0 r� r, r) r4 r rL )r& rQ r� r� Z _DeclBaser )r� r r~ u s z,DeclarativeMappedTest._with_register_classesN)r r r r} r� r_ r$ r~ r r r r r� m s r� )rB � sqlalchemyr/ � r r r Zenginesr Zentitiesr r r r r Zext.declarativer r r� r r r` rn rs ZAssertsExecutionResultsrt r� r r r r �<module> s( *l