관리-도구
편집 파일: assertions.cpython-37.pyc
B ��4]aF � @ 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 ejdJd"d#��Z!d$d%� Z"d a#d&d'� Z$dKd(d)�Z%dLd*d+�Z&dMd,d-�Z'dNd.d/�Z(dOd0d1�Z)dPd2d3�Z*dQd4d5�Z+dRd6d7�Z,dSd8d9�Z-dTd:d;�Z.dUd<d=�Z/dVd>d?�Z0d@dA� Z1dBdC� Z2G dDdE� dEe3�Z4G dFdG� dGe3�Z5G dHdI� dIe3�Z6dS )W� )�absolute_importN� )� assertsql)�config)�mock)�util)�db_spec)�fail� )�exc)�orm)�pool)�schema)�types)�default)�url)�compat)� decoratorc O s t tj| f|�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_excZ SAWarning)�messages�kw� r �P/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/assertions.py�expect_warnings s r c o sF t | �}t| tj�r&|tj�s&dV n t||�� dV W dQ R X 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 Z_currentr )�dbr r �specr r r �expect_warnings_on0 s r c s t � fdd��}|S )z�Decorator form of expect_warnings(). Note that emits_warning does **not** assert that the warnings were in fact seen. c s$ t � ddi�� | ||�S Q R X d S )N�assert_F)r )�fn�argsr )r r r �decorateI s z emits_warning.<locals>.decorate)r )r r# r )r r � emits_warningA s r$ c O s t tj| f|�S )N)r r ZSADeprecationWarning)r r r r r �expect_deprecatedQ s r% 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 s* t � f��ddi�� | ||�S Q R X d S )Nr F)r )r! r" r )r r r r r# a s z"emits_warning_on.<locals>.decorate)r )r r r# r )r r r �emits_warning_onU s r& c s t � fdd��}|S )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 s$ t � ddi�� | ||�S Q R X d S )Nr F)r% )r! r" r )r r r r# y s z!uses_deprecated.<locals>.decorate)r )r r# r )r r �uses_deprecatedi s r' TFc # s� �rdd� |D ��n|�t ���tj�� ����fdd�}t�d|�� d V W d Q R X |r�|rftjs��r�tdd��fdd ��D �� ��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 }|r8t|� �sH�| f|�|�S �sPd S xB�D ]*}�rh|�| �st�sV|| krV��|� P qVW �| f|�|� d S )Nr )r �str� issubclass�match�discard)r- �argr � exceptionZfilter_)�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)r8 r r � <genexpr>� s z#_expect_warnings.<locals>.<genexpr>) �set�warnings�warnr �patchr �py3k�AssertionError�join)r5 r r8 r Zpy2konlyr: r )r5 r6 r7 r8 r9 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 rF c C sz t �� tjr0td7 atdttj� � t �� tjr^ttj�} tj�� dat � d| � ntdkrvdsrt d��dad 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) �testutilZlazy_gcr Z_refs�_STRAY_CONNECTION_FAILURES�printr/ Z gc_collect�clearr? r@ rC )�errr r r rE � s rE c C s$ t �|| �s t|p d| |f ��d S )Nz%r !~ %r)r( r1 rC )�a�br- r r r �eq_regex� s rO c C s | |ks t |pd| |f ��dS )z.Assert a == b, with repr messaging on failure.z%r != %rN)rC )rM rN r- r r r �eq_� s rP c C s | |ks t |pd| |f ��dS )z.Assert a != b, with repr messaging on failure.z%r == %rN)rC )rM rN r- r r r �ne_� s rQ c C s | |ks t |pd| |f ��dS )z.Assert a <= b, with repr messaging on failure.z%r != %rN)rC )rM rN r- r r r �le_� s rR c C s t | d|d� d S )NT)r- )�is_)rM r- r r r �is_true� s rT c C s t | d|d� d S )NF)r- )rS )rM r- r r r �is_false� s rU c C s | |ks t |pd| |f ��dS )z.Assert a is b, with repr messaging on failure.z %r is not %rN)rC )rM rN r- r r r rS s rS c C s | |k s t |pd| |f ��dS )z2Assert a is not b, with repr messaging on failure.z%r is %rN)rC )rM rN r- r r r �is_not_ s rV c C s | |ks t |pd| |f ��dS )z.Assert a in b, with repr messaging on failure.z %r not in %rN)rC )rM rN r- r r r �in_ s rW c C s | |ks t |pd| |f ��dS )z2Assert a in not b, with repr messaging on failure.z %r is in %rN)rC )rM rN r- r r r �not_in_ s rX c C s" | � |�s t|pd| |f ��dS )z>Assert a.startswith(fragment), with repr messaging on failure.z%r does not start with %rN)� startswithrC )rM �fragmentr- r r r � startswith_ s r[ c C sX t �dd| �} t �dd| �} t �dd|�}t �dd|�}| |ksTt|pPd| |f ��d S )Nz^\s+?|\n� z {2,}� z%r != %r)r( �subrC )rM rN r- r r r �eq_ignore_whitespace s r_ c O s< y|||� d}W n | k r* d}Y nX |s8t d��d S )NFTz#Callable did not raise an exception)rC )� except_cls� callable_r" r �successr r r � assert_raises' s rc c O s| y|||� dst d��W n\ | k rv } z>t�|t�|�tj�sRt d||f ��tt�|��d�� W d d }~X Y nX d S )NFz#Callable did not raise an exceptionz%r !~ %szutf-8)rC r( �searchr � text_type�UNICODErJ �encode)r` r- ra r" �kwargs�er r r �assert_raises_message2 s rj c @ s e Zd Zddd�ZdS )�AssertsCompiledSQLNFc s0 |rt �� }np| rd }nf|d kr,t| dd �}|d kr>tjj}n@|dkrPt �� }n.|dkrbt �� }n t|tj �r~t � |�� � � }i } i } | r�| | d< |d k r�t |�| d< | r�d| d< t|tj�r�|�� }d|j_|j}nDt|tjj��rtj�|d�� }|�� |jd d d }W d Q R X | �r(| | d < |jf d |i| ��}tt|d i ��}tj �r�|� d�� dd�}t dt� |� | � d�� n t dt� |�� d� | � t!�"ddt� |��}t#||d|||f � |d k �r�t#|�$|�|� |d k �r|�$|�� t#t%� fdd�|j&D ��|� |d k �r,t#|j'|� d S )NZ __dialect__r Zdefault_enhanced�schema_translate_mapZ column_keysT� literal_bindsZ _execute_stmtr r �compile_kwargs�dialect�paramszutf-8�ascii�ignorez SQL String: z[\n\t]r\ z%r != %r on dialect %rc s g | ] }� | �qS r r )r, �x)�pr r r. � s z5AssertsCompiledSQL.assert_compile.<locals>.<listcomp>)(r ZDefaultDialect�getattrr r ro ZStrCompileDialectr r r r �URLZ get_dialect�listr ZQueryZ_compile_context� statementZ use_labelsZ persistenceZBulkUDr rA �objectZexec_Z mock_callsr) �reprrB rg �decoderJ re r( r^ rP Zconstruct_params�tupleZ positiontupZprefetch)�selfZclause�resultrp Z checkparamsro ZcheckpositionalZcheck_prefetchZuse_default_dialectZallow_dialect_selectrm rl r rn �contextZ stmt_mock�cZ param_str�ccr )rt r �assert_compile? sd z!AssertsCompiledSQL.assert_compile) NNNNNFFFN)�__name__� __module__� __qualname__r� r r r r rk > s rk c @ s e Zd Zddd�Zdd� ZdS )�ComparesTablesFc C sP t |j�t |j�kst�x�t|j|j�D ]�\}}t|j|j� ||j|j ksRt�t|j|j� t|j|j� |r�d}t|j t |j ��s�t||j |j f ��n | � ||� t|j t j �r�t|j j |j j � tdd� |jD �dd� |jD �� |jr(t|jtj�s(t�q(W t |j�t |j�k�s"t�x(|jD ] }|jj|j d k �s*t��q*W d S )Nz)Type '%s' doesn't correspond to type '%s'c S s h | ] }|j j�qS r )�column�name)r, �fr r r � <setcomp>� s z5ComparesTables.assert_tables_equal.<locals>.<setcomp>c S s h | ] }|j j�qS r )r� r� )r, r� r r r r� � s )�lenr� rC �ziprP r� Z primary_keyZnullabler �type�assert_types_base�sqltypes�String�lengthZ foreign_keysZserver_defaultr Z FetchedValue�columns)r} �tableZreflected_tableZ strict_typesr� Z reflected_cr- r r r �assert_tables_equal� s. z"ComparesTables.assert_tables_equalc C s* |j �|j �s&td|j|j |j f ��d S )Nz7On column %r, type '%s' doesn't correspond to type '%s')r� Z_compare_type_affinityrC r� )r} Zc1Zc2r 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 )N)rw rJ rz � assert_list)r} r~ �class_�objectsr r r � assert_result� s z%AssertsExecutionResults.assert_resultc C sP | � t|�t|�kd|j � x,tdt|��D ]}| �||| || � q.W d S )Nz9result list is not the same size as test list, for class r )r r� r� �range� assert_row)r} r~ r� Zlist_�ir r r r� � s z#AssertsExecutionResults.assert_listc C s� | � |j|kdt|� � x�|�� D ]�\}}t|t�r�t|d t�rb| �t||�|d |d � q�| � |d t||�|d � q$| � t||�|kd|t||�|f � q$W d S )Nzitem class is not r r z'attribute %s value %s does not match %s) r � __class__rz �itemsr r| rw r� ru r� )r} r� Zrowobj�desc�key�valuer r r r� � s z"AssertsExecutionResults.assert_rowc s� G dd� dt ��t�|�}�fdd�|D �}x4t��fdd�|�D ] }tdt|�j�jf � qBW t|�t|�kr�tdt|�t|�f � t� � � �fd d �}xF|D ]>}x8|D ]}|||�r�|� |� P q�W td �jt |�f � q�W d 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 )N)�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, ri )r� r r r� � s zBAssertsExecutionResults.assert_unordered_result.<locals>.<setcomp>c s t | � �S )N)r )�o)�clsr r �<lambda>� � zAAssertsExecutionResults.assert_unordered_result.<locals>.<lambda>z#Unexpected type "%s", expected "%s"z+Unexpected object count "%s", expected "%s"c st xn|� � D ]b\}}t|t�rXy$�jt| |�|d f|d �� W ql tk rT dS X q t| |� �|kr dS q W dS )Nr r FT)r� r r| �assert_unordered_resultru rC )�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 Z IdentitySetZitertools_filterfalser r� r� r� ry �removerz ) r} r~ r� �expected�foundZwrongr� Z expected_itemZ found_itemr )r� r� r� r} r r� � s. z/AssertsExecutionResults.assert_unordered_resultNc C s |d krddl m} t�|�S )Nr )r )r\ r r Z assert_engine)r} r r r r �sql_execution_asserter s z.AssertsExecutionResults.sql_execution_asserterc G s* | � |�� }|� }W d Q R X |j|� |S )N)r� r )r} r ra �rules�asserterr~ r r r �assert_sql_execution s z,AssertsExecutionResults.assert_sql_executionc C s\ g }xD|D ]<}t |t�r2tjdd� |�� D �� }n tj|� }|�|� q W | j||f|�� S )Nc S s g | ]\}}t �||��qS r )r � CompiledSQL)r, �k�vr r r r. ) s z6AssertsExecutionResults.assert_sql.<locals>.<listcomp>)r r� r ZAllOfr� r� �appendr� )r} r ra r� ZnewrulesZruleZnewruler r r � assert_sql# s z"AssertsExecutionResults.assert_sqlc C s | � ||t�|�� d S )N)r� r �CountStatements)r} r ra �countr r r �assert_sql_count1 s z(AssertsExecutionResults.assert_sql_countc s� � fdd�t ||�D �}g }x |D ]\}}}|�|�� � q"W z|� S x:t ||�D ],\} \}}}|�d d d � | �t�|�� qRW X d S )Nc s g | ]\}}� � |�||f�qS r )r� )r, r r� )r} r r r. 8 s zEAssertsExecutionResults.assert_multiple_sql_count.<locals>.<listcomp>)r� r� � __enter__�__exit__r r r� ) r} Zdbsra �countsZrecsZ asserters�ctxr r� r� r )r} r �assert_multiple_sql_count6 s z1AssertsExecutionResults.assert_multiple_sql_countc g s* | � |�� }d V W d Q R X |j|� d S )N)r� r )r} r r� r� r r r �assert_executionE s z(AssertsExecutionResults.assert_executionc C s | � |t�|��S )N)r� r r� )r} r r� r r r �assert_statement_countK s z.AssertsExecutionResults.assert_statement_count)N)r� r� r� r� r� r� r� r� r� r� r� r� � contextlib�contextmanagerr� r� r r r r r� � s 8 r� )TTF)N)N)N)N)N)N)N)N)N)N)N)N)7� __future__r r� r( r? r\ r r r r rH Z exclusionsr r r r r r r r r� Zenginer r r r r r� r r$ r% r& r' r rF rI rE rO rP rQ rR rT rU rS rV rW rX r[ r_ rc rj ry rk r� r� r r r r �<module> s^ , * W)