Edit File: assertsql.cpython-37.pyc
B ��4]V4 � @ sR d dl Z d dlZd dlZddlmZ ddlmZ ddlmZ ddlm Z ddl m Z ddl m Z G d d � d e�ZG d d � d e�ZG d d� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd � d e�ZG d d � d e�ZG d d � d e �d d!d"d#d$g��ZG d%d&� d&e�Z ej d'd(� �Z dS ))� N� )�event)�util)�url)�DefaultDialect)�_distill_params)� _DDLCompilesc @ s( e Zd ZdZdZdZdd� Zdd� ZdS )� AssertRuleFNTc C s d S )N� )�self�execute_observedr r �O/opt/alt/python37/lib64/python3.7/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 ZdZddd�Zdd� ZdS )� CursorSQLFNc C s || _ || _d S )N)� statement�params)r r r r r r �__init__+ s zCursorSQL.__init__c C sj |j d }| j|jks,| jd k rH| j|jkrHd| j| j|j|jf | _n |j �d� d| _|j sfd| _d S )Nr z5Testing for exact SQL %s parameters %s received %s %sT)� statementsr r � parametersr �popr r )r r �stmtr r r r / s zCursorSQL.process_statement)N)r r r r r r r r r r r ( s r c @ sF e Zd Zddd�Zdd� Zdd� Zd d � Zd d � Zd d� Zdd� Z dS )� CompiledSQLN�defaultc C s || _ || _|| _d S )N)r r �dialect)r r r r" r r r r E s zCompiledSQL.__init__c C s t �dd| j�}||kS )Nz[\n\t]� )�re�subr )r r �received_statementr r r r � _compare_sqlJ s zCompiledSQL._compare_sqlc C sB | j dkrt� S | j dkr$ddi}ni }t�| j ��� f |�S d S )Nr! � postgresqlZimplicit_returningT)r" r r �URLZ get_dialect)r r r r r r �_compile_dialectN s z CompiledSQL._compile_dialectc s� |j }| �|�}t|jjt�r:|jjj||j�d�d�� n&|jjj||jj |jj |j�d�d�� t � ddt �� ��}|j}|s�� �� g}n� fdd�|D �}||fS )zxreconstruct the statement and params in terms of a target dialect, which for CompiledSQL is just DefaultDialect.�schema_translate_map)r" r+ )r" � column_keys�inliner+ z[\n\t]r# c s g | ]}� � |��qS r )�construct_params)�.0�m)�compiledr r � <listcomp>v s z3CompiledSQL._received_statement.<locals>.<listcomp>)�contextr* � isinstancer1 r r �compileZexecution_options�getr, r- r$ r% r Z text_typer r. )r r r3 Zcompare_dialect�_received_statementr �_received_parametersr )r1 r r7 Y s& z CompiledSQL._received_statementc C s� |j }| �|�\}}| �|�}| �||�}|r�|d k r�t|�}t|�}xj|r�|r�t|�d��} xNtt|��D ]8\} } x.| D ] } | | ks�| | | | krzP qzW || = P qlW d}P qHW |s�|r�d}|r�d| _d | _ n| � |�||d� | _ d S )Nr FT)r& Zreceived_parameters) r3 r7 � _all_paramsr' �list�dictr � enumerater r �_failure_message) r r r3 r7 r8 r Z equivalent� all_paramsZ all_received�param�idxZreceivedZ param_keyr r r r { s8 z CompiledSQL.process_statementc C s@ | j r8t�| j �r | � |�}n| j }t|t�s4|g}|S d S d S )N)r r �callabler4 r: )r r3 r r r r r9 � s zCompiledSQL._all_paramsc C s d| j �dd�|f S )NzzTesting for compiled statement %r partial params %r, received %%(received_statement)r with params %%(received_parameters)r�%z%%)r �replace)r �expected_paramsr r r r= � s z CompiledSQL._failure_message)Nr! ) r r r r r' r* r7 r r9 r= r r r r r D s "1 r c @ s& e Zd Zddd�Zdd� Zdd� ZdS ) �RegexSQLNc C s, t �| � t�|�| _|| _|| _d| _d S )Nr! )r r r$ r5 �regex� orig_regexr r" )r rF r r r r r � s zRegexSQL.__init__c C s d| j |f S )Nz{Testing for compiled statement ~%r partial params %r, received %%(received_statement)r with params %%(received_parameters)r)rG )r rD r r r r= � s zRegexSQL._failure_messagec C s t | j�|��S )N)�boolrF �match)r r r& r r r r' � s zRegexSQL._compare_sql)N)r r r r r= r' r r r r rE � s rE c s4 e Zd Zdd� Zdd� Z� fdd�Zdd� Z� ZS ) � DialectSQLc C s |j jS )N)r3 r" )r r r r r r* � s zDialectSQL._compile_dialectc C s t �dd|�}||kS )Nz[\n\t]r# )r$ r% )r � real_stmt� received_stmtr r r r �_compare_no_space� s z DialectSQL._compare_no_spacec sL t t| ��|�\}}x*|jD ]}| �|j|�r P q W td| ��||jjfS )NzICan't locate compiled statement %r in list of statements actually invoked) �superrJ r7 r rM r r r3 Zcompiled_parameters)r r rL Zreceived_paramsrK )� __class__r r r7 � s z DialectSQL._received_statementc C st t �dd| j�}|jjj}|dkr2t �dd|�}n:d }|dkrDd}n|dkrRd }n |d kr^d }t �d||�}||kS ) Nz[\n\t]r# Zpyformatz :([\w_]+)z%(\1)sZqmark�?�formatz%sZnumeric)r$ r% r r3 r" � paramstyle)r r r& r rR �replr r r r' � s zDialectSQL._compare_sql)r r r r* rM r7 r' � __classcell__r r )rO r rJ � s rJ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�CountStatementsc C s || _ d| _d S )Nr )�count�_statement_count)r rV r r r r s zCountStatements.__init__c C s | j d7 _ d S )N� )rW )r r r r r r s z!CountStatements.process_statementc C s( | j | jkr$ds$td| j | jf ��d S )NFz,desired statement count %d does not match %d)rV rW r )r r r r r s z"CountStatements.no_more_statementsN)r r r r r r r r r r rU s rU c @ s e Zd Zdd� Zdd� ZdS )�AllOfc G s t |�| _d S )N)�set�rules)r r[ r r r r s zAllOf.__init__c C sj xdt | j�D ]D}d |_|�|� |jrB| j�|� | js>d| _P q |js d | _P q W t | j�d j| _d S )NTr )r: r[ r r r �discard)r r �ruler r r r s zAllOf.process_statementN)r r r r r r r r r rY s rY c s, e Zd Zdd� Zdd� Z� fdd�Z� ZS )�EachOfc G s t |�| _d S )N)r: r[ )r r[ r r r r ( s zEachOf.__init__c C sZ xH| j rH| j d }|�|� |jr0| j �d� n|jr>|j| _|jrP qW | j sVd| _d S )Nr T)r[ r r r r r )r r r] r r r r + s zEachOf.process_statementc s: | j r"| j d js"| j d �� n| j r6tt| ��� d S )Nr )r[ r r rN r^ )r )rO r r r 9 s zEachOf.no_more_statements)r r r r r r rT r r )rO r r^ ' s r^ c @ s e Zd Zdd� ZdS )�Orc C s> x8| j D ] }|�|� |jrd| _P qW t| j �d j| _d S )NTr )r[ r r r: r )r r r] r r r r A s zOr.process_statementN)r r r r r r r r r_ @ s r_ c @ s e Zd Zdd� ZdS )�SQLExecuteObservedc C s" || _ || _t||�| _g | _d S )N)r3 � clauseelementr r r )r r3 ra � multiparamsr r r r r L s zSQLExecuteObserved.__init__N)r r r r r r r r r` K s r` c @ s e Zd ZdS )�SQLCursorExecuteObservedN)r r r r r r r rc S s rc r r r3 � executemanyc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� SQLAsserterc C s g | _ d S )N)� accumulated)r r r r r ] s zSQLAsserter.__init__c C s | j | _| ` d S )N)rf �_final)r r r r �_close` s zSQLAsserter._closec G sr t |� }t| j�}x:|rL|�d�}|�|� |jr6P q|jrdst|j��qW |r`dsntd��n|jsn|�� d S )Nr Fz Additional SQL statements remain) r^ r: rg r r r r r r )r r[ r] Zobservedr r r r �assert_d s zSQLAsserter.assert_N)r r r r rh ri r r r r re \ s re c # sr t � � g �t�| d��fdd��}t�| d�� �fdd��}z � V W d t�| d|� t�| d|� � �� X d S )NZbefore_executec s |||f� d d �< d S )Nr )�connra rb r )�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 rX r )rf r3 r` �appendr rc )rj Zcursorr r r3 rd Zobs)�asserterrk r r �cursor_execute� s z%assert_engine.<locals>.cursor_execute)re r Z listens_for�removerh )�enginerl rp r )ro rk r � assert_engineu s rs ) � collections� contextlibr$ r# r r rr r Zengine.defaultr Z engine.utilr Zschemar �objectr r r r rE rJ rU rY r^ r_ r` � namedtuplerc re �contextmanagerrs r r r r �<module> s2 }.