o 8c@sbddlmZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl m Z ddl mZdd l mZdd l mZdd l mZdd l mZdd lmZddlmZddlmZddZddZddZddZddZ d3ddZGdddeZej d3d d!Z!d4d#d$Z"d4d%d&Z#d5d'd(Z$iZ%d)d*Z&d+d,Z'd-d.Z(d/d0Z)d1d2Z*dS)6)absolute_importN)exc)util)default)_expect_warnings)eq_)is_)is_false)is_not_)is_true)ne_ decorator) sqla_compat)py3kcCs6tjsdS|j|jur|jsJd||jfdSdS)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. NFzZException %r was correctly raised but did not set a cause, within context %r as its cause.)rr __context__ __cause____suppress_context__) exceptionrP/var/www/html/gps/gps/lib/python3.10/site-packages/alembic/testing/assertions.py _assert_proper_exception_contexts  rcOst||||ddS)NT check_context_assert_raises except_cls callable_argskwrrr assert_raises/sr"cOst||||SNrrrrrassert_raises_context_ok3r$cOst|||||ddS)NTmsgrrrr'rr kwargsrrrassert_raises_message7s r*cOst|||||dS)N)r'rr(rrr assert_raises_message_context_ok=sr+FcCsDt|||}||i|Wd|jS1swY|jSr#)_expect_raiseserror)rrr r)r'recrrrrCs  rc@seZdZdZdS)_ErrorContainerN)__name__ __module__ __qualname__r-rrrrr/Lsr/c cst}|r td}z|Vd}Wn@|yS}z4||_d}|dur7t|t|tjs7Jd||f|r?|s?t |t t| dWYd}~nd}~ww|sZJddS)NrFTz%r !~ %szutf-8z#Callable did not raise an exception) r/sysexc_infor-researchr text_typeUNICODErprintencode)rr'rr.are_we_already_in_a_tracebacksuccesserrrrrr,Ps,    r,TcCs t||dS)Nrr,)rrrrr expect_raisesgs r?cCst|||dS)Nr&r>)rr'rrrrexpect_raises_messagekr%r@cCsztdd|}tdd|}tdd|}tdd|}tr&tdd|}ntdd|}||ks;J|p9d||fdS)Nz^\s+?|\nz {2,} z!Uuz%r != %r)r5subr)abr'rrreq_ignore_whitespaceos rGcCsL|dus|dkr tSt|}|dkrd|_|S|dkr$d|_|S)Nr postgresqlTmssqlF)rDefaultDialectr _create_url get_dialectimplicit_returninglegacy_schema_aliasing)namedrrr _get_dialectsrQcOstt|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!rrrexpect_warningss rUcstfdd}|S)zDecorator form of expect_warnings(). Note that emits_warning does **not** assert that the warnings were in fact seen. csHttgRddi||i|WdS1swYdS)Nassert_F)rDeprecationWarning)fnr r!rTrrdecorates $z2emits_python_deprecation_warning..decorater )rTrZrrYr emits_python_deprecation_warningsr[cOttj|fi|Sr#)rsa_excSADeprecationWarningrSrrrexpect_sqlalchemy_deprecatedr_cOr\r#)rr]RemovedIn20WarningrSrrrexpect_sqlalchemy_deprecated_20r`rb)NF)Tr#)+ __future__r contextlibr5r3 sqlalchemyrr]rsqlalchemy.enginersqlalchemy.testing.assertionsrrrr r r r sqlalchemy.utilrr util.compatrrr"r$r*r+robjectr/contextmanagerr,r?r@rG _dialect_modsrQrUr[r_rbrrrrsH