@Wa p 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 d Zd Zd Zd Z d Z d&dZ G d d Z ej d&d Z!d'd Z"d'd Z#d(d Z$i Z%d e&d < d! Z'd" Z(d# Z)d$ Z*d% Z+dS )) ) annotationsN)Any)Dict)exc)default)_expect_warnings)eq_)is_)is_false)is_not_)is_true)ne_ decorator ) sqla_compatc ` | j | j ur | j sJ d| d| j d dS 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.
Fz
Exception z> was correctly raised but did not set a cause, within context z as its cause.N) __context__ __cause____suppress_context__) exceptions l/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/alembic/testing/assertions.py _assert_proper_exception_contextr s] Y%888. 9
yy)///
1
u 9888 c * t | |||d S )NT
check_context_assert_raises
except_cls callable_argskws r
assert_raisesr% , s *ir N N NNr c & t | ||| S Nr r s r assert_raises_context_okr( 0 s *ir : ::r c , t | ||||d S )NTmsgr r r! r+ r" r# kwargss r assert_raises_messager. 4 s& ItVD r c * t | |||| S )N)r+ r r, s r assert_raises_message_context_okr0 : s *iv3 G G GGr Fc r t | || 5 } ||i | d d d n # 1 swxY w Y |j S r' )_expect_raiseserror)r! r" r# r- r+ r ecs r r r @ s
C 7 7 #2 4"6"""# # # # # # # # # # # # # # #
8Os '++c e Zd ZU dZded< dS )_ErrorContainerNr r3 )__name__
__module__ __qualname__r3 __annotations__ r r r6 r6 I s# Er r6 c # K t }|rt j d } |V d}n# | $ r}||_ d}| 2 ,S 1 1 1
c#hhoog&&''''''''( 99999799s 4 C B CC Tc $ t | | S )Nr r2 )r! r s r
expect_raisesrJ b s *M B B BBr c & t | || S )Nr* rI )r! r+ r s r expect_raises_messagerL f s *#] K K KKr c t j dd| } t j dd| } t j dd| }t j dd| }| |k sJ |p| d| d S )Nz^\s+?|\n z {2,} z != )r? sub)abr+ s r eq_ignore_whitespacerS j su
{B""A
xa A
{B""A
xa A 6663-11-66666r zDict[Any, Any]
_dialect_modsc | | dk rt j S t j | }| dk rd|_ n
| dk rd|_ |S )Nr
postgresqlTmssqlF)r DefaultDialectr _create_url get_dialectimplicit_returninglegacy_schema_aliasing)nameds r _get_dialectr_ w sr |ty((%'''
8K #D ) ) 5 5 7 7 9 9 < #'A
W__',A $r c ( t t | 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.
)r Warningmessagesr$ s r expect_warningsrd s GX 4 4 4 44r c , t fd }|S )zDecorator form of expect_warnings().
Note that emits_warning does **not** assert that the warnings
were in fact seen.
c r t t gR ddi5 | |i |cd d d S # 1 swxY w Y d S )Nassert_F)r DeprecationWarning)fnr# r$ rc s r decoratez2emits_python_deprecation_warning..decorate s
0
K(
K
K
K%
K
K # #2t?r?? # # # # # # # # # # # # # # # # # #s ,00r )rc rj s ` r emits_python_deprecation_warningrk s/ # # # # Y# Or c 2 t t j | fi |S r' )r sa_excSADeprecationWarningrb s r expect_sqlalchemy_deprecatedro s F 7 H HR H HHr c 2 t t j | fi |S r' )r rm RemovedIn20Warningrb s r expect_sqlalchemy_deprecated_20rr s F 5x F F2 F FFr )NF)Tr' ),
__future__r
contextlibr? r= typingr r
sqlalchemyr rm sqlalchemy.enginer sqlalchemy.testing.assertionsr r r
r r r
r sqlalchemy.utilr utilr r r% r( r. r0 r r6 contextmanagerr2 rJ rL rS rT r: r_ rd rk ro rr r; r r r| s " " " " " " "
$ $ $ $ $ $ % % % % % % : : : : : : - - - - - - - - - - - - 2 2 2 2 2 2 1 1 1 1 1 1 1 1 1 1 1 1 - - - - - - % % % % % %
*O O O; ; ; H H H BG : : : :(C C C CL L L L. . . . !#
" " " "
5
5
5
I I IG G G G Gr