관리-도구
편집 파일: assertsql.cpython-38.pyc
U -?�f�A � @ sX d dl mZ d dlZd dlZd dlZd dlZddlmZ ddlm Z ddl m Z ddl m Z G dd � d �ZG d d � d e�ZG d d � d e�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG d d � d e�ZG d d � d �ZG d d!� d!e�d!d"d#d$d%g��ZG d&d'� d'�Z ej d(d)� �Z dS )*� )� annotationsN� )�event)�url)�DefaultDialect)�BaseDDLElementc @ s( e Zd ZdZdZdZdd� Zdd� ZdS )� AssertRuleFNTc C s d S �N� ��self�execute_observedr r �L/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/assertsql.py�process_statement s z AssertRule.process_statementc C s ds t d��d S )NFz?All statements are complete, but pending assertion rules remain)�AssertionError�r r r r �no_more_statements s �z AssertRule.no_more_statements)�__name__� __module__� __qualname__� is_consumed� errormessage�consume_statementr r r r r r r s r c @ s e Zd ZdS )� SQLMatchRuleN�r r r r r r r r & s r c @ s e Zd Zddd�Zdd� ZdS )� CursorSQLNTc C s || _ || _|| _d S r )� statement�paramsr )r r r r r r r �__init__+ s zCursorSQL.__init__c C sp |j d }| j|jks,| jd k rN| j|jkrNd| _d| j| j|j|jf | _n |j �d� d| _|j sld| _d S )Nr Tz5Testing for exact SQL %s parameters %s received %s %s)� statementsr r � parametersr r �popr )r r �stmtr r r r 0 s& � ���� zCursorSQL.process_statement)NT�r r r r r r r r r r * s r c @ sF e Zd Zddd�Zdd� Zdd � Zd d � Zd d � Zdd� Zdd� Z dS )� CompiledSQLN�defaultTc C s || _ || _|| _|| _d S r )r r �dialect�enable_returning)r r r r& r' r r r r G s zCompiledSQL.__init__c C s t �dd| j�}||kS �N�[\n\t]� )�re�subr �r r �received_statementr" r r r � _compare_sqlO s zCompiledSQL._compare_sqlc C sl | j dkrTt� }d|_| jrPd |_ |_|_d|_d|_d|_ d|_ |j sPt �|S t j�| j ��� � S d S )Nr% T)r& r Zsupports_default_metavaluer' Zinsert_returningZupdate_returningZdelete_returningZuse_insertmanyvaluesZsupports_multivalues_insertZupdate_returning_multifromZdelete_returning_multifromZ insert_executemany_returningr r �URL�create� get_dialect)r r r& r r r �_compile_dialectS s � z CompiledSQL._compile_dialectc s� |j }| �|�}d}d�d|jkr.|jd }nd}t|jt�rP|jj||d�� n |jj|||jj|jj |d�� t � ddt � ��}|j }|s�� j�d�g}n� �fdd �|D �}||fS ) zxreconstruct the statement and params in terms of a target dialect, which for CompiledSQL is just DefaultDialect.N�schema_translate_map)r& r4 )� cache_keyr& � column_keys�for_executemanyr4 r) r* ��extracted_parametersc s g | ]}� j |�d ��qS )r8 )�construct_params)�.0�m��compiledr9 r r � <listcomp>� s � �z3CompiledSQL._received_statement.<locals>.<listcomp>)�contextr3 �execution_options� isinstance� clauseelementr �compiler> r6 r7 r+ r, �strr r: ) r r r@ Zcompare_dialectr5 Zmap_�_received_statementr �_received_parametersr r= r rF m s> � ���� �z CompiledSQL._received_statementc C s� |j }| �|�\}}| �|�}| �||�}|r�|d k r�t|�}t|�}|r�|r�t|�d��} tt|��D ]8\} } | D ] } | | ks�| | | | krt qhqt|| = qFqhd}q�qF|s�|r�d}|r�d| _d | _ n| � ||�||d� | _ d S )Nr FT)r. Zreceived_parameters) r@ rF � _all_paramsr/ �list�dictr! � enumerater r �_failure_message) r r r@ rF rG r Z equivalent� all_paramsZ all_received�param�idxZreceivedZ param_keyr r r r � sH � �� ���z CompiledSQL.process_statementc C s> | j r6t| j �r | � |�}n| j }t|t�s2|g}|S d S d S r )r �callablerB rI )r r@ r r r r rH � s zCompiledSQL._all_paramsc C s" d| j �dd�t|��dd�f S �NzzTesting for compiled statement %r partial params %s, received %%(received_statement)r with params %%(received_parameters)r�%�%%)r �replace�repr�r r Zexpected_paramsr r r rL � s ���z CompiledSQL._failure_message)Nr% T) r r r r r/ r3 rF r rH rL r r r r r$ F s � 53 r$ c @ s& e Zd Zd dd�Zdd� Zdd � ZdS ) �RegexSQLNr% Fc C s2 t �| � t�|�| _|| _|| _|| _|| _d S r ) r r r+ rD �regex� orig_regexr r& r' )r rX r r& r' r r r r � s zRegexSQL.__init__c C s" d| j �dd�t|��dd�f S )Nz{Testing for compiled statement ~%r partial params %s, received %%(received_statement)r with params %%(received_parameters)rrR rS )rY rT rU rV r r r rL � s ���zRegexSQL._failure_messagec C s t | j�|��S r )�boolrX �match)r r r. r r r r/ s zRegexSQL._compare_sql)Nr% F)r r r r rL r/ r r r r rW � s � rW c sD e Zd Zdd� Zdd� Z� fdd�Zdd� Zd d � Zd d � Z� Z S ) � DialectSQLc C s |j jS r )r@ r& r r r r r3 s zDialectSQL._compile_dialectc C s t �dd|�}||kS r( )r+ r, )r � real_stmt� received_stmtr" r r r �_compare_no_space s z DialectSQL._compare_no_spacec sF t � �|�\}}|jD ]}| �|j|�r q:qtd| ��||jjfS )NzICan't locate compiled statement %r in list of statements actually invoked)�superrF r r_ r r r@ Zcompiled_parameters)r r r^ Zreceived_paramsr] �� __class__r r rF s � ��z DialectSQL._received_statementc s� |j }t�dd| j�}|�dd�}|dkr:t�dd|�}n`d }|dkrLd }n@|d krZd }n2|�d �r�t�d �� |dkrzdnd�� �fdd�}t�d||�}|�dd�}|S )Nr) r* z::z!!Zpyformatz :([\w_]+)z%(\1)sZqmark�?�formatz%s�numeric� Znumeric_dollar�$�:c s �� t � �� �S r )�next)r< ��counterZnum_identifierr r �repl5 s z4DialectSQL._dialect_adjusted_statement.<locals>.repl)� paramstyler+ r, r rT � startswith� itertools�count)r r& rm r" rl r rj r �_dialect_adjusted_statement s" z&DialectSQL._dialect_adjusted_statementc C s | � |jj�}||kS r )rq r@ r& r- r r r r/ ? s �zDialectSQL._compare_sqlc C s* d| � |jj��dd�t|��dd�f S rQ )rq r@ r& rT rU rV r r r rL E s � ����zDialectSQL._failure_message) r r r r3 r_ rF rq r/ rL � __classcell__r r ra r r\ s r\ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�CountStatementsc C s || _ d| _d S �Nr )rp �_statement_count)r rp r r r r T s zCountStatements.__init__c C s | j d7 _ d S )Nrf )ru r r r r r X s z!CountStatements.process_statementc C s( | j | jkr$ds$td| j | jf ��d S )NFz,desired statement count %d does not match %d)rp ru r r r r r r [ s �z"CountStatements.no_more_statementsN)r r r r r r r r r r rs S s rs c @ s e Zd Zdd� Zdd� ZdS )�AllOfc G s t |�| _d S r )�set�rules�r rx r r r r d s zAllOf.__init__c C sj t | j�D ]H}d |_|�|� |jrB| j�|� | js<d| _ qfq |js d | _ qfq t | j�d j| _d S �NTr )rI rx r r r �discard�r r �ruler r r r g s zAllOf.process_statementNr# r r r r rv c s rv c s, e Zd Zdd� Zdd� Z� fdd�Z� ZS )�EachOfc G s t |�| _d S r )rI rx ry r r r r y s zEachOf.__init__c C sh | j sd| _d| _| j rX| j d }|�|� |jr@| j �d� n|jrN|j| _|jrqXq| j sdd| _d S )NTFr )rx r r r r! r r| r r r r | s zEachOf.process_statementc s6 | j r"| j d js"| j d �� n| j r2t� �� d S rt )rx r r r` r ra r r r � s zEachOf.no_more_statements)r r r r r r rr r r ra r r~ x s r~ c s e Zd Z� fdd�Z� ZS )� Conditionalc s" |rt � j|� n t � j|� d S r )r` r )r � conditionrx Z else_rulesra r r r � s zConditional.__init__)r r r r rr r r ra r r � s r c @ s e Zd Zdd� ZdS )�Orc C s< | j D ] }|�|� |jrd| _ q8qt| j �d j| _d S rz )rx r r rI r r| r r r r � s zOr.process_statementN)r r r r r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�SQLExecuteObservedc C s6 || _ || _|r|| _n|r&|g| _ng | _g | _d S r )r@ rC r r )r r@ rC � multiparamsr r r r r � s zSQLExecuteObserved.__init__c C s t | j�S r )rE r r r r r �__repr__� s zSQLExecuteObserved.__repr__N)r r r r r� r r r r r� � s r� c @ s e Zd ZdS )�SQLCursorExecuteObservedNr r r r r r� � s r� r r r@ � executemanyc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� SQLAsserterc C s g | _ d S r )� accumulatedr r r r r � s zSQLAsserter.__init__c C s | j | _| ` d S r )r� �_finalr r r r �_close� s zSQLAsserter._closec G sr t |� }t| j�}|rJ|�d�}|�|� |jr4qJq|jrdst|j��q|r`dsntd| ��n|jsn|�� d S )Nr Fz$Additional SQL statements remain: %s) r~ rI r� r! r r r r r )r rx r} Zobservedr r r r �assert_� s zSQLAsserter.assert_N)r r r r r� r� r r r r r� � s r� c # sr t � � g �t�| d��fdd��}t�| d�� �fdd��}z � V W 5 t�| d|� t�| d|� � �� X d S )NZbefore_executec s |||f� d d �< d S r r )�connrC r� r rA )�origr r �connection_execute� s z)assert_engine.<locals>.connection_executeZafter_cursor_executec sj |sd S � j r*� j d j|kr*� j d }n&t|�d �d �d �}� j �|� |j�t||||�� d S )N���r rf r )r� r@ r� �appendr r� )r� �cursorr r r@ r� Zobs�Zasserterr� r r �cursor_execute� s" �� ��z%assert_engine.<locals>.cursor_execute)r� r Z listens_for�remover� )�enginer� r� r r� r � assert_engine� s r� ) � __future__r � collections� contextlibro r+ r* r r� r Zengine.defaultr Zschemar r r r r$ rW r\ rs rv r~ r r� r� � namedtupler� r� �contextmanagerr� r r r r �<module> s< (L ��