?PKz V d dl mZ d dlZd dlZd dlZd dlZddlmZ ddlmZ ddlm Z ddlm
Z ddl m
Z
dd l
mZ 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 d Zej d Z d Z d Z d Z d Z!ej d9d Z"d Z#d a$d Z%d:d Z&d:d Z'd:d!Z(d:d"Z)d:d#Z*d:d$Z+d:d%Z,d:d&Z-d:d'Z.e.Z/d:d(Z0d:d)Z1e1Z2d:d*Z3d:d+Z4d, Z5d- Z6d. Z7d/ Z8d0 Z9d1 Z: d;d2Z; G d3 d4e< Z= G d5 d6e< Z> G d7 d8e< Z?dS )< )absolute_importN ) assertsql)config)mock)util)db_spec)fail )exc)pool)schema)sql)types)default)url)compat decoratorc 2 t t j | 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.
)_expect_warningssa_exc SAWarningmessageskws q/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/sqlalchemy/testing/assertions.pyexpect_warningsr ! s F ,h = =" = == c / K t | }t | t j r |t j sdV dS t
|i |5 dV ddd dS # 1 swxY w Y dS )zContext 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 _currentr )dbr r specs r expect_warnings_onr& 1 s 2;;D"d'(( fo1F1F
h
-"
-
- EEE s A%%A),A)c , t fd }|S )zDecorator form of expect_warnings().
Note that emits_warning does **not** assert that the warnings
were in fact seen.
c ` t ddi5 | |i |cd d d S # 1 swxY w Y d S Nassert_F)r fnargsr r s r decoratez emits_warning..decorateJ s
X
6U
6
6 # #2t?r?? # # # # # # # # # # # # # # # # # # #''r r r. s ` r
emits_warningr1 B s/ # # # # Y# Or c 2 t t j | fi |S N)r r SADeprecationWarningr s r expect_deprecatedr5 R s F 7 H HR H HHr c 0 t fd }|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 h t gR ddi5 | |i |cd d d S # 1 swxY w Y d S r) )r& )r, r- r r$ r s r r. z"emits_warning_on..decorateb s
=H
=
=
=E
=
= # #2t?r?? # # # # # # # # # # # # # # # # # #s '++r )r$ r r. s `` r emits_warning_onr8 V s5 # # # # # Y# Or c , t fd }|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 ` t ddi5 | |i |cd d d S # 1 swxY w Y d S r) )r5 r+ s r r. z!uses_deprecated..decoratez s
8%
8
8 # #2t?r?? # # # # # # # # # # # # # # # # # #r/ r r0 s ` r uses_deprecatedr; j s/ # # # # Y# Or TFc # d K r
d |D n|t t j fd}t j d| 5 d V d d d n # 1 swxY w Y |r<|r t
j s2r.J dd fdD z d S d S d S )Nc f g | ].}t j |t j t j z /S )recompileIS).0msgs r
z$_expect_warnings.. s, DDDC2:c24"$;//DDDr c : t | r| }t | } n
|r |d }nd }|rt | s | g|R i |S sd S D ]9}r| | ss || k r | d S : | g|R i | d S )Nr )r! str
issubclassmatchdiscard)
rD argr exceptionfilter_exc_clsfilters real_warnregexseens
r our_warnz"_expect_warnings..our_warn s c7 # # Ii..CC
AII I .
9g > > . 9S-3---"-- - F ' 'G
'--,,
%nn W%%% Ic &C & & &2 & & & & &r z
warnings.warnzWarnings were not seen: %sz, c 3 4 K | ]}d r|j n|z V dS )z%rN)pattern)rC srQ s r z#_expect_warnings.. sL B
B
34D-AIIA .B
B
B
B
B
B
r )setwarningswarnr patchr py3kjoin) rN r rQ r* py2konlyrS rO rP rR s ` ` @@@r r r sv
DD8DDD w< r r global_cleanup_assertionsra s &'''''r c t j t j rFt dz
at dt
t j z t j t j rRt
t j } t j dat j
d| z d S t dk r
J d d 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) testutillazy_gcr
_refs_STRAY_CONNECTION_FAILURESprintrG
gc_collectclearrY rZ )errs r r` r` s
z #a'"
@$*oo
z
'$*oo
%&"
IC O
$b ( (BBBBu
) (r c R t j || sJ |p| d| d S )N !~ )r? rI abrD s r eq_regexrq s5
8Aq>>553511"555>55r c 4 | |k sJ |p| d| dS )z.Assert a == b, with repr messaging on failure. != Nr> rn s r eq_rt ) 6663-11-66666r c 4 | |k sJ |p| d| dS )z.Assert a != b, with repr messaging on failure.z == Nr> rn s r ne_rw ru r c 4 | |k sJ |p| d| dS )z.Assert a <= b, with repr messaging on failure.rs Nr> rn s r le_ry ru r c H t | | sJ |p| d| d S )Nz is not an instance of r! rn s r is_instance_ofr| s6 a JJS JQQQ$JJJ JJr c * t | d| d S )NTrD is_ro rD s r is_truer s 4Sr c * t | d| d S )NFr~ r r s r is_falser s 5cr c 0 | |u sJ |p| d| dS )z.Assert a is b, with repr messaging on failure.z is not Nr> rn s r r r ) 66631AAAqq166666r c 0 | |usJ |p| d| dS )z2Assert a is not b, with repr messaging on failure.z is Nr> rn s r is_notr s) A:::s1AAAqq 1:::::r c 0 | |v sJ |p| d| dS )z.Assert a in b, with repr messaging on failure.z not in Nr> rn s r in_r r r c 0 | |vsJ |p| d| dS )z2Assert a in not b, with repr messaging on failure.z is in Nr> rn s r not_inr s) A:::s4qqq!! 4:::::r c R | | sJ |p| d| dS )z>Assert a.startswith(fragment), with repr messaging on failure.z does not start with N)
startswith)ro fragmentrD s r startswith_r # sJ << ! ! 3 $ + ! r c t j dd| } t j dd| } t j dd| }t j dd| }| |k sJ |p| d| d S )Nz^\s+?|\n z {2,} rs )r? subrn s r eq_ignore_whitespacer + su
{B""A
xa A
{B""A
xa A 6663-11-66666r c | t j sdS | 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.
NFz
Exception z> was correctly raised but did not set a cause, within context z as its cause.)r r\ __context__ __cause____suppress_context__)rL s r _assert_proper_exception_contextr 4 sn 9 Y%888. 9
yy)///
1
u 9888r c . t | |||d d S NT)
check_context_assert_raises
except_cls callable_r- r s r
assert_raisesr L s! :y$$GGGGGGr c * t | ||| d S r3 r r s r assert_raises_context_okr P s*
r c * t | |||d S r r r s r assert_raises_returnr Y s *ir N N NNr c 0 t | ||||d d S )NT)rD r r r rD r r- kwargss r assert_raises_messager ] s0 ItVD r c . t | |||| d S )Nr~ r r s r assert_raises_message_context_okr c s# :y$C@@@@@@r c d }|rt j d } ||i | d}n# | $ r} | }d}|At j |t j | t j sJ |d| |r|st | t t j | d Y d } ~ nd } ~ ww xY w|s
J d |S )Nr FTrm utf-8z#Callable did not raise an exception)
sysexc_infor? searchr text_typeUNICODEr rh encode)
r r r- r rD r ret_err are_we_already_in_a_tracebacksuccessrk s
r r r i s/ G :(+ q(9%3 4"6""" 3 3 3 ?9T^C(("*
2!> 2 ,S 1 1 1
dnS!!((1122222222 3 999997 Ns
* CBCCc ( e Zd Z ddZdS )AssertsCompiledSQLNFc |rt j }n| rd }n|t | dd }|t j j }n||dk rt j }nb|dk rt j }nHt |t j r. t j | }i } i }
| r| | d< |t | | d< |
rd|
d< dd lm} t ||j r(| }d|j _ |j }nt ||j j ret, j |d
5 }| |j d d d }d d d n # 1 swxY w Y |
r|
| d < G d
dt0 G fddt0 } || j d d|i| }t9 t |di }t j rd| d dd }tA dt j! | z |z d n:tA dt j! | d z |z tE j# ddt j! | }tI |||d|d| |#tI |% | | |C|% | tI tM fd|j' D | |tI |j( | d S d S ) N __dialect__r default_enhancedschema_translate_map column_keysT
literal_bindsr )orm
_execute_stmtr compile_kwargsc e Zd Zd ZdS )5AssertsCompiledSQL.assert_compile..DontAccessc t d )Nz=compiler accessed .statement; use compiler.current_executable)NotImplementedError)selfkeys r __getattribute__zFAssertsCompiledSQL.assert_compile..DontAccess.__getattribute__ s )2 r N)__name__
__module__ __qualname__r r> r r
DontAccessr s#
r r c 2 e Zd Zd Zd Zd Zd Z fdZdS )>AssertsCompiledSQL.assert_compile..CheckCompilerAccessc d || _ t |dd | _ | j r|j | _ t |t
j t
j t
j f r |j | _ t |t
j t
j f r |j
| _
|j | _ d S d S d S )Nsupports_executionF) test_statementgetattrr _execution_optionsr! r InsertUpdateDelete
_returninginline_return_defaults)r r s r __init__zGAssertsCompiledSQL.assert_compile..CheckCompilerAccess.__init__ s &4#*1"$8%+ +' * P.<.OD+!&SZ(L D +9*C!.3:sz2JKK P&4&; 0>0O--- P PP Pr c 4 | j S r3 )r _default_dialectr s r r zOAssertsCompiledSQL.assert_compile..CheckCompilerAccess._default_dialect s *;;===r c 6 | j j j | fd|i|S )Ndialect)r r@ __func__r r r s r r@ zFAssertsCompiledSQL.assert_compile..CheckCompilerAccess.compile s7 ;t*2; ")-/ r c 4 | j j j | |fi |S r3 )r _compilerr r s r r zHAssertsCompiledSQL.assert_compile..CheckCompilerAccess._compiler s1 =t*4=' %' r c t |d rVt j |d 5 | j j |fi |cd d d S # 1 swxY w Y d S | j j |fi |S )N statement)hasattrr r[ objectr _compiler_dispatch)r compilerr r s r r zQAssertsCompiledSQL.assert_compile..CheckCompilerAccess._compiler_dispatch s 8[11
** +zz|| Ft 2 E $ (. B4.A $* s AA "A N)r r r r r r@ r r )r s r CheckCompilerAccessr sm
P
P
P
>
>
>
r r r paramsr asciiignorez
SQL String:
z[\n\t]r rs z on dialect c g | ]
}| S r> r> )rC xps r rE z5AssertsCompiledSQL.assert_compile.. s 333qt333r r> ))r DefaultDialectr r r$ r StrCompileDialectr! r r" r URL get_dialectlist
sqlalchemyr Query_compile_contextr
use_labels persistenceBulkUDr r[ r exec_
mock_callsr@ reprr\ r decoderh r r? r rt construct_paramstuple positiontupprefetch)r clauseresultr checkparamsr checkpositionalcheck_prefetchuse_default_dialectallow_dialect_selectr r r r r context stmt_mockr c param_strccr r s @@r assert_compilez!AssertsCompiledSQL.assert_compile s2 ; ,..GG
!
;GG!$
t<< )+I%%!022 ...!355GT%677
;8#''**6688::
>)=B% & $V B} 3.2N? +"""""" fci ( ( 7 --//G+/G ( &FF
6
7
7 7""6?;;
7y "-a03A6
7
7
7
7
7
7
7
7
7
7
7
7
7
7
7 2#1B ) ) ) ) ) ) ) & ) ) ) \
0 ' ' / F F F2 F FHb1122 9 !((1188(KKI "T^A%6%66BJJ
!.##**7334
VIr4>!#4#4
5
5 BBBBHIII " ""6**K 8 8 8 &""6**A 3333Q]33344o F F F %
N + + + + + & %s $.F F"%F") NNNNNFFFN)r r r r r> r r r r sK
! " !J, J, J, J, J, J,r r c e Zd ZddZd ZdS )ComparesTablesFc t |j t |j k sJ t |j |j D ]^\ }}t |j |j ||j |j u sJ t |j |j t |j |j |rCd}t |j t |j sJ ||j |j fz n| || t |j t j r$t |j j |j j t d |j
D d |j
D |j r!t |j t j sJ `t |j t |j k sJ |j D ]}|j j |j J d S )Nz)Type '%s' doesn't correspond to type '%s'c & h | ]}|j j S r> columnnamerC fs r z5ComparesTables.assert_tables_equal..* s 7771 777r c & h | ]}|j j S r> r r s r r z5ComparesTables.assert_tables_equal..+ s AAA1 AAAr )lenr ziprt r primary_keynullabler! typeassert_types_basesqltypesStringlength foreign_keysserver_defaultr FetchedValuecolumns)r tablereflected_table strict_typesr reflected_crD s r assert_tables_equalz"ComparesTables.assert_tables_equal s 57||s?#4 5 55555 !%'?+< = = NA{ ( ) ) ) /"3AF";;;;;
{ 6 7 7 7
K 0 1 1 1
7A!+"2DLLAA 3 $FJ D A
&&{A666!&(/22
<AFM;#3#:;;; 77777AA (@AAA
! .0C 5$%%_-H)I)IIIII" K KA".6qv>JJJJ K Kr c |j |j s#J d|j d|j d|j d d S )Nz
On column z, type 'z ' doesn't correspond to type '')r" _compare_type_affinityr )r c1c2s r r# z ComparesTables.assert_types_base6 sf w-- G
GGG GGG GGG
r N)F)r r r r/ r# r> r r r r s< K K K KB
r r c l e Zd Zd Zd Zd Zd Zd
dZd Zd Z d Z
d
Z e j
d Zd ZdS )AssertsExecutionResultsc t | }t t | | ||| d S r3 )r rh r assert_list)r r class_objectss r
assert_resultz%AssertsExecutionResults.assert_resultA s? f
d6ll 11111r c | t | t | k d|j z t dt | D ]%}| ||| || &d S )Nz9result list is not the same size as test list, for class r )r* r r range
assert_row)r r r9 list_is r r8 z#AssertsExecutionResults.assert_listF s KK3u:: %
o
q#e**%% 9 9A OOFF1IuQx 8 8 8 8 9 9r c : | |j |u dt | z | D ]\ }}t |t
rt |d t r2| t || |d |d g| |d t || |d | t || |k d|dt || d| d S )Nzitem class is not r r z
attribute z value z does not match )
r* __class__r itemsr! r r r8 r r> )r r9 rowobjdescr values r r> z"AssertsExecutionResults.assert_rowP s+ &(.immutabledictc t | S r3 )idr s r __hash__zOAssertsExecutionResults.assert_unordered_result..immutabledict.__hash__i s
$xxr N)r r r rL r> r r
immutabledictrI h s#
r rM c & h | ]
} | S r> r> )rC erM s r r zBAssertsExecutionResults.assert_unordered_result..m s# 777MM!$$777r c $ t | S r3 r{ )oclss r zAAssertsExecutionResults.assert_unordered_result..p s jC(( r zUnexpected type "z
", expected ""zUnexpected object count "c | D ]o\ }}t |t r= j t | | |d g|d R E# t
$ r Y dS w xY wt | | |k r dS pdS )Nr r FT)rC r! r assert_unordered_resultr AssertionError)objr% r rF NOVALUEr s r
_compare_itemzFAssertsExecutionResults.assert_unordered_result.._compare_item s "jjll
%
%
U eU++ %%44 #C - -uQx:?( * % % % $uuu% sC11U:: $uu ;4s )A
A*)A*z Expected z instance with attributes z not found.T) dictr IdentitySetitertools_filterfalser
r" r r r remover ) r r rR expectedfoundwrongrZ
expected_item
found_itemrY rM s ` ` @@r rV z/AssertsExecutionResults.assert_unordered_resulta s D ((7777h777/ ( ( ( (%
E
D;;''' 7
u::X & & Du::::s8}}}}.
(( & M #
=];; LL,,,E |||T-%8%8%8%8: tr Nc : |ddl m} t j | S )Nr )r$ )r r$ r
assert_engine)r r$ s r sql_execution_asserterz.AssertsExecutionResults.sql_execution_asserter s*
: " " " " " "&r***r c | | 5 } | }d d d n # 1 swxY w Y |j | |S r3 rf r* )r r$ r rulesasserterr s r assert_sql_executionz,AssertsExecutionResults.assert_sql_execution s
(
(
,
, ! Y[[F ! ! ! ! ! ! ! ! ! ! ! ! ! ! !%
s -11c g }|D ]e}t |t r+t j d | D }nt j | }| | f | j ||g|R S )Nc > g | ]\ }}t j || S r> )r CompiledSQL)rC kvs r rE z6AssertsExecutionResults.assert_sql.. s) KKKdai+Aq11KKKr )r! r[ r AllOfrC rn appendrk )r r$ r ri newrulesrulenewrules r
assert_sqlz"AssertsExecutionResults.assert_sql s % %D$%%
7#/KKdjjllKKK $/6 OOG $ $ $ $(t(YBBBBBr c X | ||t j | d S r3 )rk r CountStatements)r r$ r counts r assert_sql_countz(AssertsExecutionResults.assert_sql_count s6 !! 94U;;
r c fdt || D }g }|D ]-\ }}}| | . | t || D ]G\ } \ }}}| d d d | t j | HS # t || D ]G\ } \ }}}| d d d | t j | Hw xY w)Nc F g | ] \ }} | ||f S r> )rf )rC r$ ry r s r rE zEAssertsExecutionResults.assert_multiple_sql_count.. sB
U
(
(
,
,b% 8
r )r rr __enter____exit__r* r rx )
r dbsr countsrecs assertersctxr$ ry rj s
` r assert_multiple_sql_countz1AssertsExecutionResults.assert_multiple_sql_count sL
"3 / /
" . .NCU S]]__ - - - - C 9;;.1)T.B.B
C
C**3E T4... !:5!A!ABBBB
Cc)T.B.B
C
C**3E T4... !:5!A!ABBBB
Cs
B3 3AD
c ' K | | 5 }d V d d d n # 1 swxY w Y |j | d S r3 rh )r r$ ri rj s r assert_executionz(AssertsExecutionResults.assert_execution s
(
(
,
, EEE % s )--c R | |t j | S r3 )r r rx )r r$ ry s r assert_statement_countz.AssertsExecutionResults.assert_statement_count s# $$R)B5)I)IJJJr r3 )r r r r; r8 r> rV rf rk rv rz r
contextlibcontextmanagerr r r> r r r6 r6 @ s 2 2 2
9 9 9 "6 6 6p+ + + + C C C
C
C
C ! ! !
K K K K Kr r6 )TTFr3 )NF)@
__future__r r r? r rY r r r r r rd
exclusionsr r
r r r
r r r r$ enginer r r r r r r& r1 r5 r8 r; r ra rg r` rq rt rw ry r| r r r r is_not_r r not_in_r r r r r r r r r r r r r6 r> r r r sV ' & & & & &
>
>
>
I I I ( 0 :?*
*
*
*
Z( ( ( '' '' ''T6 6 6 6. . . .
. . . .
. . . .
K K K K 2 2 2 2
2 2 2 2 2 2 2 2
5 5 5 5 . . . .
0H H H O O O A A A BG :K, K, K, K, K, K, K, K,\)
)
)
)
)
V )
)
)
XNK NK NK NK NKf NK NK NK NK NKr