File: //usr/local/lib/python3.10/dist-packages/sqlalchemy/testing/__pycache__/assertsql.cpython-310.pyc
o
���g�A � @ sT 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!g d"���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
�G/usr/local/lib/python3.10/dist-packages/sqlalchemy/testing/assertsql.py�process_statement s z AssertRule.process_statementc C s J d��)NFz?All statements are complete, but pending assertion rules remainr
�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 @ � 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 sv |j d }| j|jks| jd ur(| j|jkr(d| _d| j| j|j|jf | _d S |j �d� d| _|j s9d| _d S 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 @ sH e Zd Z ddd�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 sh | j dkr*t� }d|_| jr(d |_ |_|_d|_d|_d|_ d|_
|j s(J �|S t j
�| j ��� � S �Nr% T)r& r �supports_default_metavaluer' �insert_returning�update_returning�delete_returning�use_insertmanyvalues�supports_multivalues_insert�update_returning_multifrom�delete_returning_multifrom� insert_executemany_returningr �URL�create� get_dialect)r r
r& r
r
r �_compile_dialectS s
�
z CompiledSQL._compile_dialectc s� |j }| �|�}d}d�d|jv r|jd }nd}t|jt�r(|jj||d�� n|jj|||jj|jj |d�� t
� ddt � ��}|j
}|sP� j�d�g}||fS � �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& r> )� cache_keyr& � column_keys�for_executemanyr> r) r* ��extracted_parametersc s g | ] }� j |�d ��qS )rB )�construct_params)�.0�m��compiledrC r
r �
<listcomp>� s ���z3CompiledSQL._received_statement.<locals>.<listcomp>)�contextr= �execution_options�
isinstance�
clauseelementr �compilerH r@ rA r+ r, �strr rD ) r r
rJ �compare_dialectr? �map_�_received_statementr �_received_parametersr
rG r rR m s@
�
����
��z CompiledSQL._received_statementc
C s� |j }| �|�\}}| �|�}| �||�}|r^|d ur^t|�}t|�}|rX|rXt|�d��} tt|��D ] \}
} | D ]} | | vsH| | | | krJ nq:||
= nq4d}n|rX|s'|s\|r^d}|rhd| _d | _ d S | �
||�||d� | _ d S )Nr FT)r. �received_parameters) rJ rR � _all_paramsr/ �list�dictr! � enumerater r �_failure_message)
r r
rJ rR rS r �
equivalent�
all_params� all_received�param�idx�received� param_keyr
r
r r � sH �
�� �
��
�z CompiledSQL.process_statementc C s: | j rt| j �r| � |�}n| j }t|t�s|g}|S d S r )r �callablerL rV )r rJ r r
r
r rU � s
zCompiledSQL._all_paramsc C �" 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
�expected_paramsr
r
r rY � � ���z CompiledSQL._failure_messager0 )
r r r r r/ r= rR r rU rY r
r
r
r r$ F s
�53 r$ c @ s( e Zd Z d
dd�Zdd� Zdd � ZdS ) �RegexSQLNr% Fc C s2 t �| � t�|�| _|| _|| _|| _|| _d S r ) r r r+ rN �regex�
orig_regexr r& r' )r rl r r& r' r
r
r r � s
zRegexSQL.__init__c C rb )Nz{Testing for compiled statement ~%r partial params %s, received %%(received_statement)r with params %%(received_parameters)rrd re )rm rf rg rh r
r
r rY � rj zRegexSQL._failure_messagec C s t | j�|��S r )�boolrl �match)r r
r. r
r
r r/ � zRegexSQL._compare_sql)Nr% F)r r r r rY r/ r
r
r
r rk � s
�
rk c sD e Zd Zdd� Zdd� Z� fdd�Zdd� Zd d
� Zd d � Z� Z S )
�
DialectSQLc C s |j jS r )rJ r& r r
r
r r= 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 nq td| ��||jjfS )NzICan't locate compiled statement %r in list of statements actually invoked)�superrR r rt r �AssertionErrorrJ �compiled_parameters)r r
rs �received_paramsrr �� __class__r
r rR s �
��� z DialectSQL._received_statementc s� |j }t�dd| j�}|�dd�}|dkr t�dd|�}n0d }|dkr&d }n |d
kr-d }n|�d �rFt�d
�� |dkr=dnd�� �fdd�}t�d||�}|�dd�}|S )Nr) r* z::z!!�pyformatz :([\w_]+)z%(\1)s�qmark�?�formatz%s�numeric� �numeric_dollar�$�:c s �� t � �� �S r )�next)rF ��counter�num_identifierr
r �repl5 rp z4DialectSQL._dialect_adjusted_statement.<locals>.repl)�
paramstyler+ r, r rf �
startswith� itertools�count)r r& r� r" r� r
r� r �_dialect_adjusted_statement s"
z&DialectSQL._dialect_adjusted_statementc C s | � |jj�}||kS r )r� rJ r& r- r
r
r r/ ? s �zDialectSQL._compare_sqlc C s* d| � |jj��dd�t|��dd�f S rc )r� rJ r& rf rg rh r
r
r rY E s ����zDialectSQL._failure_message)
r r r r= rt rR r� r/ rY �
__classcell__r
r
ry r rq s rq c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�CountStatementsc C s || _ d| _d S �Nr �r� �_statement_count)r r� r
r
r r T s
zCountStatements.__init__c C s | j d7 _ d S )Nr� )r� r r
r
r r X s z!CountStatements.process_statementc C s$ | j | jkrJ d| j | jf ��d S )NFz,desired statement count %d does not match %dr� r r
r
r r [ s ��z"CountStatements.no_more_statementsN)r r r r r r r
r
r
r r� S s r� c @ � e Zd Zdd� Zdd� ZdS )�AllOfc G � t |�| _d S r )�set�rules�r r� r
r
r r d � zAllOf.__init__c C sl t | j�D ]%}d |_|�|� |jr!| j�|� | js d| _ d S |js*d | _ d S qt | j�d j| _d S �NTr )rV r� r r r �discard�r r
�ruler
r
r r g s
�zAllOf.process_statementNr# r
r
r
r r� c s r� c s, e Zd Zdd� Zdd� Z� fdd�Z� ZS )�EachOfc G r� r )rV r� r� r
r
r r y r� zEachOf.__init__c C sp | j s d| _d| _| j r.| j d }|�|� |jr | j �d� n|jr'|j| _|jr+n| j s | j s6d| _d S d S )NTFr )r� r r r r! r r� r
r
r r | s
�
�zEachOf.process_statementc s< | j r| j d js| j d �� d S | j r t� �� d S d S r� )r� r r ru r ry r
r r � s
�zEachOf.no_more_statements)r r r r r r r� r
r
ry r r� x s r� c s e Zd Z� fdd�Z� ZS )� Conditionalc s$ |r
t � j|� d S t � j|� d S r )ru r )r � conditionr� �
else_rulesry r
r r � s zConditional.__init__)r r r r r� r
r
ry r r� � s r� c @ s e Zd Zdd� ZdS )�Orc C s> | j D ]}|�|� |jrd| _ d S qt| j �d j| _d S r� )r� r r rV r r� r
r
r r � s
�zOr.process_statementN)r r r r r
r
r
r r� � s r� c @ r� )�SQLExecuteObservedc C s6 || _ || _|r || _n
|r|g| _ng | _g | _d S r )rJ rM r r )r rJ rM � multiparamsr r
r
r r � s
zSQLExecuteObserved.__init__c C s
t | j�S r )rO r r r
r
r �__repr__� �
zSQLExecuteObserved.__repr__N)r r r r r� r
r
r
r r� � s r� c @ r )�SQLCursorExecuteObservedNr r
r
r
r r� � s r� )r r rJ � executemanyc @ r� )� SQLAsserterc C s
g | _ d S r )� accumulatedr r
r
r r � r� zSQLAsserter.__init__c C s | j | _| ` d S r )r� �_finalr r
r
r �_close� s zSQLAsserter._closec G sl t |� }t| j�}|r#|�d�}|�|� |jrn
|jr!J |j��|s |r+J d| ��|js4|�� d S d S )Nr Fz$Additional SQL statements remain:
%s)r� rV r� r! r r r r )r r� r� �observedr r
r
r �assert_� s
� �zSQLAsserter.assert_N)r r r r r� r� r
r
r
r r� � s r� c # s� � t � � g �t�| d��fdd��}t�| d�� �fdd��}z� V W t�| d|� t�| d|� � �� d S t�| d|� t�| d|� � �� w )N�before_executec s |||f� d d �< d S r r
)�connrM r� r rK )�origr
r �connection_execute� s z)assert_engine.<locals>.connection_execute�after_cursor_executec sj |sd S � j r� j d j|u r� j d }nt|�d �d �d �}� j �|� |j�t||||�� d S )N���r r� r )r� rJ r� �appendr r� )r� �cursorr r rJ r� �obs��asserterr� r
r �cursor_execute� s � ��z%assert_engine.<locals>.cursor_execute)r� r � listens_for�remover� )�enginer� r� r
r� r �
assert_engine� s �
�
r� ) �
__future__r � collections�
contextlibr� r+ r* r r� r �engine.defaultr �schemar r r r r$ rk rq r� r� r� r� r� r� �
namedtupler� r� �contextmanagerr� r
r
r
r �<module> s> (L
��