jIhUddlmZddlZddlZddlZddlmZddlmZ ddl m Z ddl m Z ddl mZddl mZdd l mZdd l mZdd l mZdd l mZdd l mZddlmZdZdZdZdZdZ d$dZGddZej d%dZd&dZ d'dZ!d(dZ"iZ#de$d<dZ%d Z&d!Z'd"Z(d#Z)dS))) annotationsN)Any)exc)default)URL)_expect_warnings)eq_)is_)is_false)is_not_)is_true)ne_ decoratorc`|j|jur|jsJd|d|jddSdS)aHassert 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 }/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib/python3.11/site-packages/alembic/testing/assertions.py _assert_proper_exception_contextrs` Y%888. 9  yy)/// 1    9888c*t||||dS)NT check_context_assert_raises except_cls callable_argskws r assert_raisesr#*s *ir N N NNrc&t||||SNrrs rassert_raises_context_okr&.s *ir : ::rc,t|||||dS)NT)msgrrrr(r r!kwargss rassert_raises_messager+2s& ItVD   rc*t|||||S)N)r(rr)s r assert_raises_message_context_okr-8s *iv3 G G GGrFcrt|||5}||i|dddn #1swxYwY|jSr%)_expect_raiseserror)rr r!r*r(recs rrr>s  C 7 7#2 4"6"""############### 8Os '++ceZdZUdZded<dS)_ErrorContainerNrr0)__name__ __module__ __qualname__r0__annotations__rrr3r3Fs#Err3c#Kt}|rtjd} |Vd}n#|$r}||_d}|a|r#t ||ksJ|d|n 2 ,S 1 1 1 c#hhoog&&'''''''' ( 99999999s4C1B.C,,C1Tc$t||S)Nrr/)rrs r expect_raisesrIds *M B B BBrc(t||||S)N)r(rrCrH)rr(rrCs rexpect_raises_messagerKhs$ =Z   rctjdd|}tjdd|}tjdd|}tjdd|}||ksJ|p|d|dS)Nz^\s+?|\nz {2,} r:)r>sub)abr(s req_ignore_whitespacerRpsu {B""A xa  A {B""A xa  A 6663-11-66666rzdict[Any, Any] _dialect_modsc||dkrtjStj|}|dkrd|_n|dkrd|_d|_|S)Nr postgresqlTmssqlFdbo)rDefaultDialectrcreate get_dialectimplicit_returninglegacy_schema_aliasingdefault_schema_name)nameds r _get_dialectr`|sw |ty((%''' *CJt   ( ( * * , , <  #'A W__',A $$)A !rc(tt|fi|S)aContext 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. )rWarningmessagesr"s rexpect_warningsres GX 4 4 4 44rc,tfd}|S)zDecorator form of expect_warnings(). Note that emits_warning does **not** assert that the warnings were in fact seen. crttgRddi5||i|cdddS#1swxYwYdS)Nassert_FrDeprecationWarning)fnr!r"rds rdecoratez2emits_python_deprecation_warning..decorates 0 K( K K K% K K # #2t?r?? # # # # # # # # # # # # # # # # # #s ,00r)rdrls` r emits_python_deprecation_warningrms/####Y# Orc(tt|fi|Sr%rircs rexpect_deprecatedros . ? ?B ? ??rc2ttj|fi|Sr%)rsa_excSADeprecationWarningrcs rexpect_sqlalchemy_deprecatedrss F7 H HR H HHr)NF)NFF)T)TFr%)* __future__r contextlibr>r;typingr sqlalchemyrrqsqlalchemy.enginerrsqlalchemy.testing.assertionsrr r r r r rsqlalchemy.utilrrr#r&r+r-rr3contextmanagerr/rIrKrRrSr7r`rermrorsr8rrr|s""""""" $$$$$$%%%%%%!!!!!!::::::------------222222111111111111------%%%%%%   *OOO;;; HHHBG :?::::2CCCC 5:....!# """"    5 5 5    @@@IIIIIr