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: assertions.cpython-310.pyc
o [�h�G � @ s d dl mZ d dlZd dlZd dlZddlmZ ddlmZ ddlmZ ddlm Z ddl m Z dd l m Z d d lmZ d d lmZ d d lmZ d dlmZ d dlmZ d dlm Z d dlmZ d dlmZ d dl mZ d dl mZ dd� Zejdd� �Z dd� Z dd� Z d d � Z d d � Z ej !dLd"d#��Z!d$d%� Z"d a#d&d'� Z$dMd(d)�Z%dMd*d+�Z&dMd,d-�Z'dMd.d/�Z(dMd0d1�Z)dMd2d3�Z*dMd4d5�Z+dMd6d7�Z,dMd8d9�Z-dMd:d;�Z.dMd<d=�Z/dMd>d?�Z0d@dA� Z1dBdC� Z2dDdE� Z3G dFdG� dGe4�Z5G dHdI� dIe4�Z6G dJdK� dKe4�Z7dS )N� )�absolute_importN� )� assertsql)�config)�mock)�util)�db_spec)�fail� )�exc)�orm)�pool)�schema)�types)�default)�url)�compat�� decoratorc O � t tj| fi |��S )a� Context manager which expects one or more warnings. With no arguments, squelches all SAWarnings emitted via sqlalchemy.util.warn and sqlalchemy.util.warn_limited. Otherwise pass string expressions that will match selected warnings via regex; all non-matching warnings are sent through. The expect version **asserts** that the warnings were in fact seen. Note that the test suite sets SAWarning warnings to raise exceptions. )�_expect_warnings�sa_exc� SAWarning��messages�kw� r �H/usr/local/lib/python3.10/dist-packages/sqlalchemy/testing/assertions.py�expect_warnings s r c o sd � t | �}t| tj�r|tj�sdV dS t|i |��� dV W d � dS 1 s+w Y dS )z�Context manager which expects one or more warnings on specific dialects. The expect version **asserts** that the warnings were in fact seen. N)r � isinstancer � string_typesr �_currentr )�dbr r �specr r r �expect_warnings_on0 s � "�r$ c � t � fdd��}|S )z�Decorator form of expect_warnings(). Note that emits_warning does **not** assert that the warnings were in fact seen. c �@ t � ddi�� | |i |��W d � S 1 sw Y d S �N�assert_F)r ��fn�argsr �r r r �decorateI � $�z emits_warning.<locals>.decorater �r r- r r, r � emits_warningA s r0 c O r �N)r r �SADeprecationWarningr r r r �expect_deprecatedQ s r3 c s t � �fdd��}|S )aV Mark a test as emitting a warning on a specific dialect. With no arguments, squelches all SAWarning failures. Or pass one or more strings; these will be matched to the root of the warning description by warnings.filterwarnings(). Note that emits_warning_on does **not** assert that the warnings were in fact seen. c sH t � g��R ddi�� | |i |��W d � S 1 s w Y d S r' )r$ r) �r" r r r r- a s $�z"emits_warning_on.<locals>.decorater )r" r r- r r4 r �emits_warning_onU s r5 c r% )a+ Mark a test as immune from fatal deprecation warnings. With no arguments, squelches all SADeprecationWarning failures. Or pass one or more strings; these will be matched to the root of the warning description by warnings.filterwarnings(). As a special case, you may pass a function name prefixed with // and it will be re-written as needed to match the standard warning verbiage emitted by the sqlalchemy.util.deprecated decorator. Note that uses_deprecated does **not** assert that the warnings were in fact seen. c r& r' )r3 r) r, r r r- y r. z!uses_deprecated.<locals>.decorater r/ r r, r �uses_deprecatedi s r6 TFc # s� � �r dd� |D ��n|�t ���tj�� ����fdd�}t�d|�� d V W d � n1 s2w Y |rP|r>tjsT�rRJ dd��fdd ��D �� ��d S d S d S ) Nc S s g | ] }t �|t jt jB ��qS r )�re�compile�I�S)�.0�msgr r r � <listcomp>� s z$_expect_warnings.<locals>.<listcomp>c s� t | � �r | }t|�} n |r|d }nd }|r t|� �s'�| g|�R i |��S �s+d S �D ]}�r6|�| �s<�sD|| krD��|� d S q-�| g|�R i |�� d S )Nr )r �str� issubclass�match�discard)r< �argr � exception�filter_��exc_cls�filters� real_warn�regex�seenr r �our_warn� s& � �z"_expect_warnings.<locals>.our_warnz warnings.warnzWarnings were not seen: %sz, c 3 s" � | ] }d � r |j n| V qdS )z%rN)�pattern)r; �s)rI r r � <genexpr>� s � �z#_expect_warnings.<locals>.<genexpr>)�set�warnings�warnr �patchr �py3k�join)rF r rI r( �py2konlyrK r rE r r � s" � � ���r c C s t � dS )z�Check things that have to be finalized at the end of a test suite. Hardcoded at the moment, a modular system can be built here to support things like PG prepared transactions, tables all dropped, etc. N)�!_assert_no_stray_pool_connectionsr r r r �global_cleanup_assertions� s rW c C st t �� tjrtd7 atdttj� � t �� tjr0ttj�} tj�� dat � d| � d S tdkr8J d��d S )Nr z2Encountered a stray connection in test cleanup: %sr z8Stray connection refused to leave after gc.collect(): %s� Fz*Encountered more than 10 stray connections) �testutil�lazy_gcr �_refs�_STRAY_CONNECTION_FAILURES�printr> � gc_collect�clearrP rQ )�errr r r rV � s& �� ��rV c C s$ t �|| �sJ |pd| |f ��d S )Nz%r !~ %r)r7 r@ ��a�br< r r r �eq_regex� s $rd c C s | |ksJ |p d| |f ��dS )z.Assert a == b, with repr messaging on failure.�%r != %rNr ra r r r �eq_� � rf c C s | |ksJ |p d| |f ��dS )z.Assert a != b, with repr messaging on failure.z%r == %rNr ra r r r �ne_� rg rh c C s | |ksJ |p d| |f ��dS )z.Assert a <= b, with repr messaging on failure.re Nr ra r r r �le_� rg ri c C � t | d|d� d S )NT�r< ��is_�rb r< r r r �is_true� � ro c C rj )NFrk rl rn r r r �is_false� rp rq c C s | |u sJ |p d| |f ��dS )z.Assert a is b, with repr messaging on failure.z %r is not %rNr ra r r r rm rg rm c C s | |usJ |p d| |f ��dS )z2Assert a is not b, with repr messaging on failure.z%r is %rNr ra r r r �is_not_ rg rr c C s | |v sJ |p d| |f ��dS )z.Assert a in b, with repr messaging on failure.z %r not in %rNr ra r r r �in_ rg rs c C s | |vsJ |p d| |f ��dS )z2Assert a in not b, with repr messaging on failure.z %r is in %rNr ra r r r �not_in_ rg rt c C s" | � |�sJ |p d| |f ��dS )z>Assert a.startswith(fragment), with repr messaging on failure.z%r does not start with %rN)� startswith)rb �fragmentr< r r r � startswith_ s �rw c C sX t �dd| �} t �dd| �} t �dd|�}t �dd|�}| |ks*J |p(d| |f ��d S )Nz^\s+?|\n� z {2,}� re )r7 �subra r r r �eq_ignore_whitespace s r{ c O s> z ||i |�� d}W n | y d}Y nw |s J d��d S �NFT�#Callable did not raise an exceptionr )� except_cls� callable_r+ r �successr r r � assert_raises' s �r� c O sX d }z ||i |�� d}W n | y# } z d}|}W Y d }~nd }~ww |s*J d��|S r| r )r~ r r+ r �ret_errr� r` r r r �assert_raises_return2 s �� r� c O sx z ||i |�� J d�� | y; } z$t �|t�|�t j�s&J d||f ��tt�|��d�� W Y d }~d S d }~ww )NFr} z%r !~ %s�utf-8)r7 �searchr � text_type�UNICODEr] �encode)r~ r< r r+ �kwargs�er r r �assert_raises_message@ s �"��r� c @ s( e Zd Z ddd�ZdS )�AssertsCompiledSQLNFc s@ |rt �� }n8| r d }n3|d u rt| dd �}|d u r tjj}n |dkr(t �� }n|dkr1t �� }nt|tj �r?t � |�� � � }i } i } | rI| | d< |d urSt |�| d< | rYd| d< t|tj�rk|�� }d|j_|j}n+t|tjj�r�tj�|d��}|�� |jd d d }W d � n1 s�w Y | r�| | d < |jdd |i| ��}tt|d i ��}tj r�|� d�� dd�}t dt� |� | � d�� nt dt� |�� d� | � t!�"ddt� |��}t#||d|||f � |d ur�t#|�$|�|� |d u�r|�$|�� t#t%� fdd�|j&D ��|� |d u�r t#|j'|� d S d S )N� __dialect__r �default_enhanced�schema_translate_map� column_keysT� literal_binds� _execute_stmtr r �compile_kwargs�dialect�paramsr� �ascii�ignorez SQL String: z[\n\t]rx z%r != %r on dialect %rc s g | ]}� | �qS r r )r; �x��pr r r= � � z5AssertsCompiledSQL.assert_compile.<locals>.<listcomp>r )(r �DefaultDialect�getattrr r" r� �StrCompileDialectr r r r �URL� get_dialect�listr �Query�_compile_context� statement� use_labels� persistence�BulkUDr rR �object�exec_� mock_callsr8 �reprrS r� �decoder] r� r7 rz rf �construct_params�tuple� positiontup�prefetch)�self�clause�resultr� � checkparamsr� �checkpositional�check_prefetch�use_default_dialect�allow_dialect_selectr� r� r r� �context� stmt_mock�c� param_str�ccr r� r �assert_compileM sv ������ �z!AssertsCompiledSQL.assert_compile) NNNNNFFFN)�__name__� __module__� __qualname__r� r r r r r� L s �r� c @ s e Zd Zddd�Zdd� ZdS )�ComparesTablesFc C sB t |j�t |j�ks J �t|j|j�D ]n\}}t|j|j� ||j|j u s(J �t|j|j� t|j|j� |rNd}t|jt|j��sMJ ||j|jf ��n| � ||� t|jt j �rdt|jj |jj � tdd� |j D �dd� |j D �� |jr�t|jtj�s�J �qt |j�t |j�ks�J �|jD ] }|jj|j d us�J �q�d S )Nz)Type '%s' doesn't correspond to type '%s'c S � h | ]}|j j�qS r ��column�name�r; �fr r r � <setcomp>� r� z5ComparesTables.assert_tables_equal.<locals>.<setcomp>c S r� r r� r� r r r r� � r� )�lenr� �ziprf r� � primary_key�nullabler �type�assert_types_base�sqltypes�String�length� foreign_keys�server_defaultr � FetchedValue�columns)r� �table�reflected_table� strict_typesr� � reflected_cr< r r r �assert_tables_equal� s8 � ��� �z"ComparesTables.assert_tables_equalc C s* |j �|j �sJ d|j|j |j f ��d S )Nz7On column %r, type '%s' doesn't correspond to type '%s')r� �_compare_type_affinityr� )r� �c1�c2r r r r� � s ��z ComparesTables.assert_types_baseN)F)r� r� r� r� r� r r r r r� � s !r� c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zdd d �Zd d � Zdd� Z dd� Z dd� Z e j dd� �Zdd� Zd S )�AssertsExecutionResultsc G s& t |�}tt|�� | �|||� d S r1 )r� r] r� � assert_list)r� r� �class_�objectsr r r � assert_result� s z%AssertsExecutionResults.assert_resultc C sL | � t|�t|�kd|j � tdt|��D ] }| �||| || � qd S )Nz9result list is not the same size as test list, for class r )r( r� r� �range� assert_row)r� r� r� �list_�ir r r r� � s ���z#AssertsExecutionResults.assert_listc C s� | � |j|u dt|� � |�� D ]A\}}t|t�r?t|d t�r0| �t||�|d |d � q| � |d t||�|d � q| � t||�|kd|t||�|f � qd S )Nzitem class is not r r z'attribute %s value %s does not match %s) r( � __class__r� �itemsr r� r� r� r� r� )r� r� �rowobj�desc�key�valuer r r r� � s � ���z"AssertsExecutionResults.assert_rowc s� G dd� dt ��t�|�}�fdd�|D �}t��fdd�|�D ]}tdt|�j�jf � q t|�t|�krCtdt|�t|�f � t� � � �fd d �}|D ] }|D ]}|||�ra|� |� n qStd �jt |�f � qOd S ) z�As assert_result, but the order of objects is not considered. The algorithm is very expensive but not a big deal for the small numbers of rows that the test suite manipulates. c @ s e Zd Zdd� ZdS )zFAssertsExecutionResults.assert_unordered_result.<locals>.immutabledictc S s t | �S r1 )�id�r� r r r �__hash__� s zOAssertsExecutionResults.assert_unordered_result.<locals>.immutabledict.__hash__N)r� r� r� r� r r r r � immutabledict� s r� c s h | ]}� |��qS r r )r; r� )r� r r r� � r� zBAssertsExecutionResults.assert_unordered_result.<locals>.<setcomp>c s t | � �S r1 )r )�o)�clsr r �<lambda>� s zAAssertsExecutionResults.assert_unordered_result.<locals>.<lambda>z#Unexpected type "%s", expected "%s"z+Unexpected object count "%s", expected "%s"c st |� � D ]3\}}t|t�r,z�jt| |�|d g|d �R � W q ty+ Y dS w t| |� �|kr7 dS qdS )Nr r FT)r� r r� �assert_unordered_resultr� �AssertionError)�objr# r� r� )�NOVALUEr� r r � _compare_item s � � ��zFAssertsExecutionResults.assert_unordered_result.<locals>._compare_itemz2Expected %s instance with attributes %s not found.T) �dictr � IdentitySet�itertools_filterfalser r� r� r� r� �remover� ) r� r� r� �expected�found�wrongr� � expected_item� found_itemr )r� r� r� r� r r� � s@ ����� � ���z/AssertsExecutionResults.assert_unordered_resultNc C s |d u r ddl m} t�|�S )Nr )r" )rx r" r � assert_engine)r� r" r r r �sql_execution_asserter% s z.AssertsExecutionResults.sql_execution_asserterc G s> | � |�� }|� }W d � n1 sw Y |j|� |S r1 �r r( )r� r" r �rules�asserterr� r r r �assert_sql_execution+ s � z,AssertsExecutionResults.assert_sql_executionc C sZ g }|D ] }t |t�rtjdd� |�� D �� }ntj|� }|�|� q| j||g|�R � S )Nc S s g | ] \}}t �||��qS r )r � CompiledSQL)r; �k�vr r r r= 7 s z6AssertsExecutionResults.assert_sql.<locals>.<listcomp>)r r� r �AllOfr� r �appendr )r� r" r r �newrules�rule�newruler r r � assert_sql1 s � z"AssertsExecutionResults.assert_sqlc C s | � ||t�|�� d S r1 )r r �CountStatements)r� r" r �countr r r �assert_sql_count? s �z(AssertsExecutionResults.assert_sql_countc s� � fdd�t ||�D �}g }|D ] \}}}|�|�� � qz |� W t ||�D ]\} \}}}|�d d d � | �t�|�� q&S t ||�D ]\} \}}}|�d d d � | �t�|�� qCw )Nc s g | ] \}}� � |�||f�qS r )r )r; r" r r� r r r= E s ��zEAssertsExecutionResults.assert_multiple_sql_count.<locals>.<listcomp>)r� r � __enter__�__exit__r( r r ) r� �dbsr �counts�recs� asserters�ctxr" r r r r� r �assert_multiple_sql_countD s ���z1AssertsExecutionResults.assert_multiple_sql_countc g s@ � | � |�� }d V W d � n1 sw Y |j|� d S r1 r )r� r" r r r r r �assert_executionS s � �z(AssertsExecutionResults.assert_executionc C s | � |t�|��S r1 )r r r )r� r" r r r r �assert_statement_countY rp z.AssertsExecutionResults.assert_statement_countr1 )r� r� r� r� r� r� r� r r r r r � contextlib�contextmanagerr r r r r r r� � s 8 r� )TTFr1 )8� __future__r r r7 rP rx r r r r rY � exclusionsr r r r r r r r r� �enginer r r r r r r$ r0 r3 r5 r6 r rW r\ rV rd rf rh ri ro rq rm rr rs rt rw r{ r� r� r� r� r� r� r� r r r r �<module> sf �- * W)
Save
Cancel