관리-도구
편집 파일: assertions.cpython-38.pyc
U 0?�f� � @ sh U d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z d dl m Z d dl mZ d dl mZ d d l mZ d d l mZ d d l mZ d d l mZ d d l mZ d dlmZ ddlmZ dd� Zdd� Zdd� Zdd� Z dd� Z d5d d �Z G d d � d �Z ej d6d d!��Z!d7d#d$�Z"d8d%d&�Z#d9d'd(�Z$i Z%d)e&d*< d+d,� Z'd-d.� Z(d/d0� Z)d1d2� Z*d3d4� Z+dS ):� )� annotationsN)�Any)�Dict)�exc)�default)�_expect_warnings)�eq_)�is_)�is_false)�is_not_)�is_true)�ne_�� decorator� )� sqla_compatc C s, | j | jk r(| js(ds(td| | j f ��dS )aH assert that any exception we're catching does not have a __context__ without a __cause__, and that __suppress_context__ is never set. Python 3 will report nested as exceptions as "during the handling of error X, error Y occurred". That's not what we want to do. we want these exceptions in a cause chain. FzZException %r was correctly raised but did not set a cause, within context %r as its cause.N)� __context__� __cause__�__suppress_context__�AssertionError)� exception� r �H/opt/hc_python/lib/python3.8/site-packages/alembic/testing/assertions.py� _assert_proper_exception_context s ����r c O s t | |||dd�S )NT�� check_context��_assert_raises�� except_cls� callable_�args�kwr r r � assert_raises, s r# c O s t | |||�S �Nr r r r r �assert_raises_context_ok0 s r% c O s t | ||||dd�S )NT��msgr r �r r' r r! �kwargsr r r �assert_raises_message4 s �r* c O s t | ||||d�S )N)r' r r( r r r � assert_raises_message_context_ok: s r+ Fc C s( t | ||��}|||� W 5 Q R X |jS r$ )�_expect_raises�error)r r r! r) r' r �ecr r r r @ s r c @ s e Zd ZU dZded< dS )�_ErrorContainerNr r- )�__name__� __module__� __qualname__r- �__annotations__r r r r r/ H s r/ c c s� t � }|rt�� d }z|V d}W n| | k r� } z^||_d}|d k rnt�|t|�tj�snt|� d|� ���|r~|s~t |� t t|�� d�� W 5 d }~X Y nX |s�td��d S )Nr FTz !~ zutf-8z#Callable did not raise an exception) r/ �sys�exc_infor- �re�search�str�UNICODEr r �print�encode)r r' r r. Z are_we_already_in_a_traceback�success�errr r r r, L s &$r, Tc C s t | |d�S )Nr �r, )r r r r r � expect_raisesa s r? c C s t | ||d�S )Nr& r> )r r' r r r r �expect_raises_messagee 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)r6 �subr )�a�br' r r r �eq_ignore_whitespacei s rF zDict[Any, Any]� _dialect_modsc C sN | d ks| dkrt �� S t�| ��� � }| dkr8d|_n| dkrFd|_|S d S )Nr � postgresqlTZmssqlF)r ZDefaultDialectr Z _create_url� get_dialectZimplicit_returningZlegacy_schema_aliasing)�name�dr r r � _get_dialectu s rL c O s t t| 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. )r �Warning��messagesr" r r r �expect_warnings� s rP 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 s6 t tf� �ddi�� | ||�W 5 Q R � S Q R X d S )NZassert_F)r �DeprecationWarning)�fnr! r" �rO r r �decorate� s z2emits_python_deprecation_warning.<locals>.decorater )rO rT r rS r � emits_python_deprecation_warning� s rU c O s t tj| f|�S r$ )r �sa_excZSADeprecationWarningrN r r r � expect_sqlalchemy_deprecated� s rW c O s t tj| f|�S r$ )r rV ZRemovedIn20WarningrN r r r � expect_sqlalchemy_deprecated_20� s rX )NF)NF)T)T)N),� __future__r � contextlibr6 r4 �typingr r Z sqlalchemyr rV Zsqlalchemy.enginer Z sqlalchemy.testing.assertionsr r r r r r r Zsqlalchemy.utilr �utilr r r# r% r* r+ r r/ �contextmanagerr, r? r@ rF rG r3 rL rP rU rW rX r r r r �<module> sJ �