Dashboard
PHP:
8.2.30
OS:
Linux
User:
dh_nv8b2m
/
/
usr
/
local
/
lib
/
python3.10
/
dist-packages
/
sqlalchemy
/
testing
/
__pycache__
📤 Upload
📝 New File
📁 New Folder
Close
Editing: assertsql.cpython-310.pyc
o [�h5 � @ sN 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 g d!���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 �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 ZdZddd�Zdd� ZdS )� CursorSQLFNc C s || _ || _d S r )� statement�params)r r r r r r �__init__+ � zCursorSQL.__init__c C sp |j d }| j|jks| jd ur%| j|jkr%d| j| j|j|jf | _d S |j �d� d| _|j s6d| _d S 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_statementr )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 r )r r �dialect)r r r r' r r r r E s zCompiledSQL.__init__c C s t �dd| j�}||kS �N�[\n\t]� )�re�subr )r r �received_statementr$ r r r � _compare_sqlJ s zCompiledSQL._compare_sqlc C sB | j dkrt� S | j dkrddi}ni }t�| j ��� di |��S )Nr& � postgresql�implicit_returningTr )r' r r �URL� 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}|sH� �� g}||fS � fdd�|D �}||fS )zxreconstruct the statement and params in terms of a target dialect, which for CompiledSQL is just DefaultDialect.�schema_translate_map)r' r4 )r' � column_keys�inliner4 r) r* c s g | ]}� � |��qS r )�construct_params)�.0�m��compiledr r � <listcomp>u s �z3CompiledSQL._received_statement.<locals>.<listcomp>)�contextr3 � isinstancer; r r �compile�execution_options�getr5 r6 r+ r, r � text_typer" r7 )r r r= �compare_dialect�_received_statementr" �_received_parametersr r: r rD Y s4 ���� ��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) r= rD � _all_paramsr. �list�dictr# � enumerater r �_failure_message) r r r= rD rE r � equivalent� all_params� all_received�param�idx�received� param_keyr r r r { sB � �� � �z CompiledSQL.process_statementc C s< | j r t�| j �r| � |�}n| j }t|t�s|g}|S d S r )r r �callabler> rH )r r= r r r r rG � 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 �expected_paramsr r r rK � � ���z CompiledSQL._failure_message�Nr& ) r r r r r. r3 rD r rG rK 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 r\ )r r r+ r? �regex� orig_regexr r' )r r^ r r r r r � s zRegexSQL.__init__c C rT )Nz{Testing for compiled statement ~%r partial params %s, received %%(received_statement)r with params %%(received_parameters)rrU rV )r_ rW rX rY r r r rK � r[ zRegexSQL._failure_messagec C s t | j�|��S r )�boolr^ �match)r r r- r r r r. � s zRegexSQL._compare_sqlr )r r r r rK r. r r r r r] � s r] c s4 e Zd Zdd� Zdd� Z� fdd�Zdd� Z� ZS ) � 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 sJ t t| ��|�\}}|jD ] }| �|j|�r nq td| ��||jjfS )NzICan't locate compiled statement %r in list of statements actually invoked) �superrb rD r! re r �AssertionErrorr= �compiled_parameters)r r rd �received_paramsrc �� __class__r r rD � s �� ��� z DialectSQL._received_statementc C sz t �dd| j�}|jjj}|dkr t �dd|�}||kS d }|dkr%d}n |dkr,d }n|d kr2d }t �d||�}||kS ) Nr) r* �pyformatz :([\w_]+)z%(\1)s�qmark�?�formatz%s�numeric)r+ r, r r= r' � paramstyle)r r r- r$ rq �replr r r r. � s �zDialectSQL._compare_sql)r r r r3 re rD r. � __classcell__r r rj r rb � s rb c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�CountStatementsc C s || _ d| _d S �Nr ��count�_statement_count)r rx r r r r r zCountStatements.__init__c C s | j d7 _ d S )N� )ry r r r r r s z!CountStatements.process_statementc C s$ | j | jkrJ d| j | jf ��d S )NFz,desired statement count %d does not match %drw 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 � t |�| _d S r )�set�rules�r r~ r r r r � 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 )rH r~ r r r �discard�r r �ruler r r r s �zAllOf.process_statementN)r r r r r r r r r r{ s r{ c s, e Zd Zdd� Zdd� Z� fdd�Z� ZS )�EachOfc G r| r )rH r~ r r r r r / r� zEachOf.__init__c C s^ | j r%| j d }|�|� |jr| j �d� n|jr |j| _|jr"n| j s| j s-d| _d S d S )Nr T)r~ r r r# r r r� r r r r 2 s � �zEachOf.process_statementc s@ | j r| j d js| j d �� d S | j r tt| ��� d S d S rv )r~ r r rf r� r rj r r r @ s �zEachOf.no_more_statements)r r r r r r rs r r rj r r� . s r� c @ � e Zd Zdd� ZdS )�Orc C s> | j D ]}|�|� |jrd| _ d S qt| j �d j| _d S r� )r~ r r rH r r� r r r r H s �zOr.process_statementN)r r r r r r r r r� G � r� c @ r� )�SQLExecuteObservedc C s" || _ || _t||�| _g | _d S r )r= � clauseelementr r" r! )r r= r� � multiparamsr r r r r S s zSQLExecuteObserved.__init__N)r r r r r r r r r� R r� r� c @ r )�SQLCursorExecuteObservedNr r r r r r� Z s r� )r r" r= � executemanyc @ rt )� SQLAsserterc C s g | _ d S r )� accumulatedr r r r r d s zSQLAsserter.__init__c C s | j | _| ` d S r )r� �_finalr r r r �_closeg s zSQLAsserter._closec G sh t |� }t| j�}|r#|�d�}|�|� |jrn |jr!J |j��|s |r)J d��|js2|�� d S d S )Nr Fz Additional SQL statements remain)r� rH r� r# r r r r )r r~ r� �observedr r r r �assert_k s � �zSQLAsserter.assert_N)r r r r r� r� r r r r r� c 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 )�connr� r� r )�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 rz r )r� r= r� �appendr! r� )r� �cursorr r" r= 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� ) � collections� contextlibr+ r* r r r� r �engine.defaultr � engine.utilr �schemar �objectr r r r% r] rb ru r{ r� r� r� � namedtupler� r� �contextmanagerr� r r r r �<module> s< . ��
Save
Cancel