관리-도구
편집 파일: sql.cpython-38.pyc
U -?�f > � @ s d dl mZ d dlZd dlZd dlZd dlZd dlZddlm Z ddl m Z ddl m Z ddl mZ dd l mZ dd lmZ dd lmZ d d l mZ d dl mZ d dlmZ d dlmZ d dlmZ G dd� de �ZG dd� d�ZG dd� d�Z G dd� de�Z G dd� d�Z d d d �Z dS ) � )� annotationsN� )�TestBase� )�config)�mock)�eq_)�ne_)�adict)� drop_all_tables_from_metadata� )�event)�util)�sort_tables_and_constraints)�visitors)� ClauseElementc @ s e Zd ZdZdZdZdZdZdZdZ dZ dZ dZ dZ ejddd�dd� �Zejdd d�d d � �Zed d � �Zedd� �Zedd� �Zedd� �Zdd� Zdd� Zdd� Zedd� �Zed d � �Z ed d � �Z ed d!� �Z ed"d#� �Z ed$d%� �Z d&d'� Z!d(d)� Z"ed*d+� �Z#dS ),� TablesTest�once�eachNT�class�Zautouse�scopec c s0 | j }|�� |�� |�� d V |�� d S �N)� __class__� _init_class�_setup_once_tables�_setup_once_inserts� _teardown_once_metadata_bind)�self�cls� r �O/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/fixtures/sql.py�_setup_tables_test_class7 s z#TablesTest._setup_tables_test_class�functionc c s" | � � | �� d V | �� d S r )�_setup_each_tables�_setup_each_inserts�_teardown_each_tables�r r r r! �_setup_tables_test_instanceD s z&TablesTest._setup_tables_test_instancec C s | j S r )�_tables_metadatar' r r r! �tables_test_metadataM s z TablesTest.tables_test_metadatac C sX | j dkr(| jdkrd| _| jdks(t�t� | _t� | _t� | _| �� | _ t � � | _ d S )Nr r )r N) �run_define_tables�run_create_tables� run_inserts�AssertionErrorr �other�tables� sequences� setup_bind�bind�saZMetaDatar) �r r r r! r Q s zTablesTest._init_classc C s6 | j dkr2| �� | j�� �}| �|� W 5 Q R X d S �Nr �r- �_load_fixturesr3 �begin� insert_data)r �connr r r! r _ s z TablesTest._setup_once_insertsc C sR | j dkrN| �| j� | jdkr.| j�| j� | j�| jj� | j�| jj � d S r6 � r+ � define_tablesr) r, Z create_allr3 r0 �updater1 Z _sequencesr5 r r r! r f s z TablesTest._setup_once_tablesc C sl | j dkrP| �| j� | jdkr.| j�| j� | j�| jj� | j�| jj � n| jdkrh| j�| j� d S �Nr r< r' r r r! r$ o s z TablesTest._setup_each_tablesc C s6 | j dkr2| �� | j�� �}| �|� W 5 Q R X d S r? r7 )r r; r r r! r% y s z TablesTest._setup_each_insertsc C sD | j dkr8| j�� | jdkr,t| j| j� | j�� n| jdkrPt| j| j� ttj dd�}|rh|j }| j dk�r@| jdk�r@| j dk�r@| j� � ��}t dd� t| jj�� �D ��D ]~}z:|r�|�� � |�|�� � W 5 Q R X n|�|�� � W q� tjjk �r2 } ztd||f tjd� W 5 d }~X Y q�X q�W 5 Q R X d S )Nr � savepointsFc S s g | ]\}}|d k r|�qS r r )�.0�t�fksr r r! � <listcomp>� s �z4TablesTest._teardown_each_tables.<locals>.<listcomp>zError emptying table %s: %r)�file)r+ r0 �clearr, r r) r3 �getattrr � requirements�enabled� run_deletesr9 �reversedr �valuesZ begin_nested�execute�deleter4 �excZ DBAPIError�print�sys�stderr)r r@ r; �table�exr r r! r& sD ��� ��� �z TablesTest._teardown_each_tablesc C sF | j rt| j| j� | jdkr*| �| j� d | j_| jd k rBd | _d S r6 )r, r r) r3 �run_dispose_bind� dispose_bind�run_setup_bindr5 r r r! r � s z'TablesTest._teardown_once_metadata_bindc C s t jS r )r �dbr5 r r r! r2 � s zTablesTest.setup_bindc C s* t |d�r|�� nt |d�r&|�� d S )N�dispose�close)�hasattrrY rZ )r r3 r r r! rV � s zTablesTest.dispose_bindc C s d S r r )r �metadatar r r! r= � s zTablesTest.define_tablesc C s i S r r r5 r r r! �fixtures� s zTablesTest.fixturesc C s d S r r )r � connectionr r r! r: � s zTablesTest.insert_datac C s | � | j||� d S r )Zassert_sql_countr3 )r �count�fnr r r! � sql_count_� s zTablesTest.sql_count_c C s | � | j||� d S r )Z assert_sqlr3 )r Z callable_Z statementsr r r! �sql_eq_� s zTablesTest.sql_eq_c s� i i � }| � � �� D ]F\�}t|�dk r,qt�t�r@| j� �|d � �< |dd� |�< qt| jj�� �D ]V\�}�dkr�qn�� kr�qn| j � � �*}|� �� � � �fdd�|� D �� W 5 Q R X qndS )z4Insert rows as represented by the fixtures() method.r r r Nc s g | ]}t t� � |���qS r )�dict�zip)rA Z column_values��headersrS r r! rD � s �z-TablesTest._load_fixtures.<locals>.<listcomp>) r] �items�len� isinstance�strr0 r r) rL r3 r9 rM �insert)r �rows�datarC r; r re r! r8 � s, � ��zTablesTest._load_fixtures)$�__name__� __module__� __qualname__rW r+ r, r- rJ rU r3 r) r0 r/ r1 r �fixturer" r( �propertyr* � classmethodr r r r$ r% r& r r2 rV r= r] r: ra rb r8 r r r r! r sT ) r c @ s" e Zd Zejddd�dd� �ZdS )�NoCacheTr# r c c s" t jj}d t j_d V |t j_d S r )r rX Z_compiled_cache)r �_cacher r r! �_disable_cache� s zNoCache._disable_cacheN)rn ro rp r rq rv r r r r! rt � s rt c @ s8 e Zd Zejdd� �Zdd� Zejddd�dd � �Z d S ) � RemovesEventsc C s t � S r )�setr' r r r! � _event_fns� s zRemovesEvents._event_fnsc K s( | j �|||f� tj|||f|� d S r )ry �addr �listen)r �target�namer` �kwr r r! � event_listen� s zRemovesEvents.event_listenTr# r c c s d V | j D ]}tj|� q d S r )ry r �remove)r �keyr r r! �_remove_events s z RemovesEvents._remove_eventsN) rn ro rp r Zmemoized_propertyry r r rq r� r r r r! rw � s rw c @ s: e Zd Zd ZZdZdZe�d�Z dd� Z e dd� �Z dS ) � ComputedReflectionFixtureTestNT)Zcomputed_columnsZtable_reflectionz[\[\]\(\)\s`'\"]*c C s | j �d|��� S )N� )�regexp�sub�lower)r �textr r r! � normalize s z'ComputedReflectionFixtureTest.normalizec C sx ddl m} ddl m} ddlm} ddlm} ddlm} |d||d|d d �|d |�|d ||d ��|d|dd�� |d||d|d d �|d |�|d||d ���}|jjj r�|d||d|d d �|d |�|d||d��t j d�}|jj j �r(|� |d||ddd��� |jjj �r(|� |d||ddd��� |jjj �rt|� |d||dd d��� |jjj �rt|� |d||d d d��� d S ) Nr )�Integer)�testing)�Column)�Computed)�TableZcomputed_default_table�idT)Z primary_key�normalZ computed_colz normal + 42Z with_defaultZ42)Zserver_defaultZcomputed_column_tableZcomputed_no_flagz normal / 42)�schemaZcomputed_virtualz normal + 2F)Z persistedz normal / 2Zcomputed_storedz normal - 42z normal * 42)r� r� r� r� r� r� r� �requiresZschemasrI r Z test_schemaZcomputed_columns_virtual� append_columnZcomputed_columns_stored) r r\ r� r� r� r� r� rB �t2r r r! r= sz � � � �� �� �� ��z+ComputedReflectionFixtureTest.define_tables) rn ro rp r- rJ Z __backend__� __requires__�re�compiler� r� rs r= r r r r! r� s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�CacheKeyFixturec C s� |� � }|� � }|d kr6|j�d�s(t�|d ks�t�nNt|j|j� tt|j�t|j�� t|j|j�D ]\}}|j ||d�sht�qh||fS )N�nocache��compare_values) �_generate_cache_key� _annotations�getr. r r� �hashrd � bindparams�compare)r �a�br� �a_key�b_key�a_param�b_paramr r r! �_compare_equalX s z CacheKeyFixture._compare_equalc C s� |� }|� }t �tt|��d�D �]�\}}||krX| �|| || |�\}}|d krVq n�|| �� }|| �� }|d ks�|d kr�|d kr�|| j�d�s�t�|d kr || j�d�s t�q |j |j k�rt |j |j �D ] \} } | j | |d�s� �qq�t |j |j � nt |j |j � t|| t�r t|| t�r g } g } t�|| �D ] } | jdk�rD| �| � �qDt�|| �D ] } | jdk�rp| �| � �qptt|j dd� d�tt�| �dd� d�� tt|j d d� d�tt�| �d d� d�� q d S ) Nr r� r� Z bindparamc S s | j S r �r� �r� r r r! �<lambda>� � z8CacheKeyFixture._run_cache_key_fixture.<locals>.<lambda>r� c S s | j S r r� r� r r r! r� � r� c S s | j S r r� r� r r r! r� � r� c S s | j S r r� r� r r r! r� � r� )� itertools� combinations_with_replacement�rangerh r� r� r� r� r. r� rd r� r� r ri r r ZiterateZ__visit_name__�appendr �sortedr Z unique_list)r rq r� �case_a�case_br� r� r� r� r� r� Zassert_a_paramsZassert_b_params�elemr r r! �_run_cache_key_fixtureh sx � � � � � �� ��z&CacheKeyFixture._run_cache_key_fixturec C sD |� }|� }t �tt|��d�D ] \}}| �|| || |� q d S )Nr )r� r� r� rh r� )r rq r� r� r� r� r� r r r! � _run_cache_key_equal_fixture� s � z,CacheKeyFixture._run_cache_key_equal_fixtureN)rn ro rp r� r� r� r r r r! r� W s Kr� Fc sJ | j }|j�| j�G dd� d�� � ���fdd����fdd�}|| _d S )Nc @ s, e Zd ZdZdd� Zedd� �Zdd� ZdS ) z.insertmanyvalues_fixture.<locals>.RandomCursor��cursorc S s || _ d S r r� )r r� r r r! �__init__� s z7insertmanyvalues_fixture.<locals>.RandomCursor.__init__c S s | j jS r )r� � descriptionr' r r r! r� � s z:insertmanyvalues_fixture.<locals>.RandomCursor.descriptionc S s | j �� }t|�}t�|� |S r )r� �fetchall�list�random�shuffle)r rl r r r! r� � s z7insertmanyvalues_fixture.<locals>.RandomCursor.fetchallN)rn ro rp � __slots__r� rr r� r� r r r r! � RandomCursor� s r� c 3 sB �r � |�}�| |||||�D ] }�r6|j r6t�d� |V q d S )Nz)Batches were downgraded for sorted INSERT)Z is_downgradedr �warn)r^ r� Z statement� parametersZgeneric_setinputsizes�context�batch)r� � orig_dialect�randomize_rows�warn_on_downgradedr r! �!_deliver_insertmanyvalues_batches� s � zCinsertmanyvalues_fixture.<locals>._deliver_insertmanyvalues_batchesc s4 t jj| d� d�� �| |�W 5 Q R � S Q R X d S )Nr� )�new)r �patch�object)�dialectr� )r� � orig_connr r! �_exec_insertmany_context� s �z:insertmanyvalues_fixture.<locals>._exec_insertmany_context)r� r� r� )r^ r� r� r� r� r )r� r� r� r� r� r� r! �insertmanyvalues_fixture� s r� )FF) � __future__r r� r� r� rQ Z sqlalchemyr4 �baser r� r r Z assertionsr r r r r r r� r �sqlr Z sql.elementsr r rt rw r� r� r� r r r r! �<module> s4 Q Pg �