U
|
¸ý°dËT ã @ sú d dl Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
|
lmZ dd lm Z dd lm
|
Z
|
dd lm Z ddl m Z ddl mZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lm Z dd l m Z dd l m Z G d d d ej!Z"G d!d" d"ej!Z#G d#d$ d$ej$Z%G d%d& d&ej!Z&G d'd( d(ej$Z'G d)d* d*ej!Z(G d+d, d,ej!Z)G d-d. d.ej*e)Z+G d/d0 d0ej!Z,G d1d2 d2ej$Z-dS )3é Né )Útestingé )Ú assert_raises)Úconfig)Úengines)Úeq_)Úfixtures)Ú is_not_none)Úis_true)Úne_)Úprovide_metadata)Ú expect_raises)Úexpect_raises_message)Ú requirements)Ú set_default_schema_on_connection)ÚColumn)ÚTableé )Ú bindparam)Údialects)Úevent)Úexc)ÚInteger)Úliteral_column)Úselect)ÚString)ÚCompiled)Úinspect_getfullargspecc @ s e Zd ZdZdd ZdS )ÚPingTestTc C s0 t j ¡ }tt jj |jj¡ W 5 Q R X d S ©N)r ÚdbÚconnectr ÚdialectZdo_pingÚ
|
connectionÚdbapi_connection)ÚselfÚconn© r( ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/testing/suite/test_dialect.pyÚ test_do_ping% s ÿzPingTest.test_do_pingN)Ú__name__Ú
|
__module__Ú __qualname__Ú __backend__r* r( r( r( r) r " s r c @ s: e Zd ZdZdd Zejee ddd Zdd Z d S )
|
ÚArgSignatureTestaË test that all visit_XYZ() in :class:`_sql.Compiler` subclasses have
|
``**kw``, for #8988.
|
|
This test uses runtime code inspection. Does not need to be a
|
``__backend__`` test as it only needs to run once provided all target
|
dialects have been imported.
|
|
For third party dialects, the suite would be run with that third
|
party as a "--dburi", which means its compiler classes will have been
|
imported by the time this test runs.
|
|
c c sR t jD ] } | d¡st d| ¡ qtg}|rN| d¡}| | ¡ ¡ |V q*d S )NÚ_zsqlalchemy.dialects.%sr ) r Ú__all__Ú
|
startswithÚ importlibÚ import_moduler ÚpopÚextendÚ__subclasses__)ÚdÚstackÚclsr( r( r) Ú_all_subclasses: s
|
|
|
z ArgSignatureTest._all_subclasses)Úparamsc c s |j V d S r )Úparam)r& Úrequestr( r( r) Úall_subclassesF s z ArgSignatureTest.all_subclassesc C sL |}|j D ]<}| d¡r
|
t||}t|}t|jd|j d| d q
|
d S )NZvisit_zCompiler visit method Ú.z:() does not accommodate for **kw in its argument signature)Ú__dict__r2 Úgetattrr r
|
Úvarkwr+ )r& r? r: ÚkÚmethZinspr( r( r) Ú test_all_visit_methods_accept_kwJ s
|
|
|
þz1ArgSignatureTest.test_all_visit_methods_accept_kwN)
|
r+ r, r- Ú__doc__r; r ÚfixtureÚlistr? rF r( r( r( r) r/ , s
|
|
r/ c @ s: e Zd ZdZdZdZedd Zej dd Z
|
dd Z d
|
S ) Ú ExceptionTestz¶Test basic exception wrapping.
|
|
DBAPIs vary a lot in exception behavior so to actually anticipate
|
specific exceptions from real round trips, we need to be conservative.
|
|
ÚeachTc C s( t d|tdtdddtdtd d S )NÚ manual_pkÚidTF©Ú primary_keyZ autoincrementÚdataé2 ©r r r r ©r: Úmetadatar( r( r) Ú define_tablese s üzExceptionTest.define_tablesc C sd t j ¡ P}| ¡ }| | jj ¡ ddd¡ tt j
|
|j| jj ¡ ddd | ¡ W 5 Q R X d S )Nr Úd1©rM rP ) r r! r" ÚbeginÚexecuteÚtablesrL Úinsertr r ZIntegrityErrorÚrollback)r& r' Útransr( r( r) Útest_integrity_errorn s
|
ÿ
|
üz"ExceptionTest.test_integrity_errorc C s t j ¡ v}z | ttd¡ ds(tW nB tjk
|
rl } z"t |}t |j
|
t |ks\tW 5 d }~X Y nX t |t s|tW 5 Q R X d S )Nu méilF) r r! r" rY r r ÚAssertionErrorr Ú
|
DBAPIErrorÚstrÚorigÚ
|
isinstance)r& r' ÚerrÚerr_strr( r( r) Ú test_exception_with_non_ascii s (z+ExceptionTest.test_exception_with_non_asciiN) r+ r, r- rG Ú run_deletesr. Ú classmethodrU r Z$duplicate_key_raises_integrity_errorr^ rf r( r( r( r) rJ Y s
|
|
rJ c @ s\ e Zd ZdZdZdd Zdd Zdd Zd d
|
Ze j
|
j d d Z e j
|
j e j
|
j d d ZdS )ÚIsolationLevelTestT©Zisolation_levelc C sF t t¡}|d }|d }t| d|g¡}|r8| ¡ S t d¡ d S )NÚdefaultÚ supportedÚ
|
AUTOCOMMITz(no non-default isolation level available)r Úget_isolation_levelsr ÚsetÚ
|
differencer5 Z skip_test)r& Úlevelsrk rl Úsr( r( r) Ú _get_non_default_isolation_level s
|
z3IsolationLevelTest._get_non_default_isolation_levelc C s t tjjjt t¡d d S )Nrk )r r r! r# Zdefault_isolation_levelr rn )r& r( r( r) Ú test_default_isolation_level¤ s þz/IsolationLevelTest.test_default_isolation_levelc C sl | ¡ }tj ¡ P}| ¡ }t|| |j|d t| ¡ | |j |j
|
j ¡ t| ¡ | W 5 Q R X d S )Nrj ) rs r r! r" Úget_isolation_levelr Úexecution_optionsr r# Úreset_isolation_levelr$ r% )r& Z non_defaultr' Úexistingr( r( r) Ú test_non_default_isolation_levelª s
|
ÿz3IsolationLevelTest.test_non_default_isolation_levelc
|
C s¢ t t¡}|d }t| dg¡D ]z}tj ¡ >}|j|d t| ¡ | |
|
¡ }| ¡ t| ¡ | W 5 Q R X tj ¡ }t| ¡ |d W 5 Q R X q"d S )Nrl rm rj rk ) r rn r ro rp r! r" rv r ru rX r\ )r& rq Z
|
all_levelsÚlevelr' r] r( r( r) Útest_all_levels¼ s
|
þz"IsolationLevelTest.test_all_levelsc
|
C sJ |}t tjdd|jjd t t¡d ¡f |j dd W 5 Q R X dS )z4test for the new get_isolation_level_values() methodúNInvalid value '%s' for isolation_level. Valid isolation levels for '%s' are %sÚFOOú, rl rj N)
|
r r Ú ArgumentErrorr# ÚnameÚjoinr rn r rv )r& Úconnection_no_transr$ r( r( r) Ú#test_invalid_level_execution_optionÒ s ÿýþþ z6IsolationLevelTest.test_invalid_level_execution_optionc
|
C sR |t ddd}ttjdd|jjd t t ¡d ¡f |
|
¡ W 5 Q R X dS )z
test for the new get_isolation_level_values() method
|
and support for the dialect-level 'isolation_level' parameter.
|
|
r} rj )Úoptionsr| r~ rl N) Údictr r r r# r r r rn r r" )r& Útesting_engineÚengr( r( r) Ú test_invalid_level_engine_paramå s ÿýþþ z2IsolationLevelTest.test_invalid_level_engine_paramN)r+ r, r- r. Ú __requires__rs rt ry r{ r ÚrequiresZget_isolation_level_valuesr Z#dialect_level_isolation_level_paramr r( r( r( r) ri s
|
ri c @ sD e Zd ZdZdZdZedd Zdd Zdd Z d
|
d Z
|
d d Z dS )ÚAutocommitIsolationTestrK )Ú
|
autocommitTc C s, t d|tdtdddtdtddd d S ) NÚ
|
some_tablerM TFrN rP rQ )Ztest_needs_acidrR rS r( r( r) rU s ûz%AutocommitIsolationTest.define_tablesc C s | ¡ }| | jj ¡ ddd¡ | ¡ t| t| jjj j
|
¡|rHdnd | ¡ | ¡ | | jj ¡ ¡ W 5 Q R X d S )Nr z some datarW ) rX rY rZ r r[ r\ r Úscalarr ÚcrM Údelete)r& r' r r] r( r( r) Ú_test_conn_autocommits s
|
ÿ
|
þ
|
z.AutocommitIsolationTest._test_conn_autocommitsc C s< |}|j dd}| |d¡ |j |jj¡ | |d¡ d S )Nrm rj TF)rv r r# rw r$ r% )r& r r' Úc2r( r( r) Útest_autocommit_on s
|
z*AutocommitIsolationTest.test_autocommit_onc C s |}| |d¡ d S )NF)r ©r& r r' r( r( r) Útest_autocommit_off( s z+AutocommitIsolationTest.test_autocommit_offc C sB |}|j dd}| |d¡ |j t t¡d d | |d¡ d S )Nrm rj Trk F)rv r r rn r r r( r( r) Ú.test_turn_autocommit_off_via_default_iso_level, s ÿÿzFAutocommitIsolationTest.test_turn_autocommit_off_via_default_iso_levelN) r+ r, r- rg r r. rh rU r r r r r( r( r( r) r ý s
|
r c @ s e Zd Zedd ZdS )Ú EscapingTestc C s | j }td|tdtd}| tj¡ tj ¡ }| | ¡ t
|
dd¡ | | ¡ t
|
dd¡ t | t |jj |jjtdk¡¡d t | t |jj |jjtdk¡¡d W 5 Q R X d S )
|
z|test that the DBAPI accommodates for escaped / nonescaped
|
percent signs in a way that matches the compiler
|
|
ÚtrP rQ z some % value)rP zsome %% other valuez'some % value'z'some %% other value'N)rT r r r Úcreater r! rX rY r[ r
r r r r rP Úwherer )r& Úmr r' r( r( r) Ú test_percent_sign_round_trip< s, ÿÿú ÿÿúz)EscapingTest.test_percent_sign_round_tripN)r+ r, r- r r r( r( r( r) r ; s r c @ s4 e Zd ZdZdZdd Zdd Zdd Zd d
|
Zd S ) Ú WeCanSetDefaultSchemaWEventsTestT)Zdefault_schema_name_switchc C s8 t jjj}t ¡ }| ¡ W 5 Q R X t|jj| d S r )r r! r# Údefault_schema_namer r r" r )r& r r r( r( r) Útest_control_casea s
|
|
|
|
z2WeCanSetDefaultSchemaWEventsTest.test_control_casec C sd t jjj}t ¡ }t |d¡dd }| ¡ }|j |¡}t
|
|t j W 5 Q R X t
|
|jj| d S )Nr" c S s t t| tj d S r ©r r Ú test_schema©r% Zconnection_recordr( r( r) Ú
|
on_connecto s
|
ÿzMWeCanSetDefaultSchemaWEventsTest.test_wont_work_wo_insert.<locals>.on_connect) r r! r# r r r r Ú listens_forr" Ú_get_default_schema_namer r¡ )r& r r r£ r' Úwhat_it_should_ber( r( r) Útest_wont_work_wo_insertj s
|
|
|
|
z9WeCanSetDefaultSchemaWEventsTest.test_wont_work_wo_insertc C s` t ¡ }tj|ddddd }| ¡ }|j |¡}t|tj W 5 Q R X t|jj
|
tj d S )Nr" T©r[ c S s t t| tj d S r r r¢ r( r( r) r£ ~ s
|
ÿzRWeCanSetDefaultSchemaWEventsTest.test_schema_change_on_connect.<locals>.on_connect) r r r r¤ r" r# r¥ r r r¡ r )r& r r£ r' r¦ r( r( r) Ú test_schema_change_on_connect{ s
|
|
z>WeCanSetDefaultSchemaWEventsTest.test_schema_change_on_connectc C s t ¡ }tj|ddddd }| ¡ F}| ¡ }|j |¡}t|t j
|
| ¡ |j |¡}t|t j
|
W 5 Q R X t|jj t j
|
d S )Nr" Tr¨ c W s t t| tj d S r r )r% Úargr( r( r) r£ s
|
ÿz\WeCanSetDefaultSchemaWEventsTest.test_schema_change_works_w_transactions.<locals>.on_connect) r r r r¤ r" rX r# r¥ r r r¡ r\ r )r& r r£ r' r] r¦ r( r( r) Ú'test_schema_change_works_w_transactions s
|
|
zHWeCanSetDefaultSchemaWEventsTest.test_schema_change_works_w_transactionsN) r+ r, r- r. r r r§ r© r« r( r( r( r) r \ s r c @ s e Zd ZdS )Ú&FutureWeCanSetDefaultSchemaWEventsTestN)r+ r, r- r( r( r( r) r¬ s r¬ c @ s e Zd ZdZejdddddddd d
|
d d d dddddddddddZeejj dd Z
|
ej dd Z ed d Z ed d Zd!S )"ÚDifficultParametersTestT)Zboring)zper cent)z
|
per % cent)z%percent)zpar(ens))zpercent%(ens)yah)zcol:ons)Z_starts_with_underscore)zdot.s)zmore :: %colons%)Ú_name)Z___name)z[BracketsAndCase])Z 42numbers)z percent%signs)z
|
has spaces)z /slashes/)z more/slashes)zq?marks)Z1param)z1col:onÚ paramname)Úargnamesc C sú |}t d|tdtddt|tddd}| |¡ | | ¡ dd|d i¡¡ t|j |
|
|j | d k¡}t | |¡d t|j |
|
|j | t |k¡}| ||d i¡ ¡ }t |j| d t|j |
|
|j | d d
|
g¡¡}| |¡ ¡ }d S ) Nr rM T)rO rQ F)Znullabler z some namezsome other_name)r r r r r rY r[ Úvaluesr r r r r r ÚfirstÚ_mappingÚin_)r& r¯ r$ rT r r ÚstmtÚrowr( r( r) Ú!test_round_trip_same_named_columnÁ s$ ü
|
"ÿz9DifficultParametersTest.test_round_trip_same_named_columnc
|
c sv t d|tdttdtdtdtd}| |¡ | | ¡ dddd d
|
d d d d ddd dddd g¡ |V d S )NÚmytableÚmyidr rQ Údescr ÚaZa_desc)r¹ r rº r ÚbZb_descr r Zc_descé r8 Zd_desc)r r r r r rY r[ )r& rT r$ r¸ r( r( r) Úmultirow_fixtureé s" û
|
|
|
|
|
üþ z(DifficultParametersTest.multirow_fixturec C sD |}t |jj |jjt|ddk¡}| ||di¡}t|d d S )NÚx©Úvaluer r )r r r¹ r r r r r ©r& r¯ r$ r¾ Ztbl1rµ Úresr( r( r) Ú test_standalone_bindparam_escape s ÿz8DifficultParametersTest.test_standalone_bindparam_escapec C s` |}t |jj |jj t|ddgd¡¡ |jj¡}| ||ddgi¡ ¡ }t
|
|ddg d S )Nr» r¼ rÀ r8 r r½ ) r r r¹ r r r´ r Zorder_byZscalarsÚallr r r( r( r) Ú*test_standalone_bindparam_escape_expanding s ÿþÿzBDifficultParametersTest.test_standalone_bindparam_escape_expandingN)r+ r, r- r. r Ú combinationsZtough_parametersr r Z unusual_column_name_charactersr· rH r¾ rÄ rÆ r( r( r( r) r ¥ sD ê &
|
|
|
r c @ s^ e Zd ZdZdZedd Zejdd Z dd Z
|
d d
|
Z d d Z d d Z dd Zdd ZdS )ÚReturningGuardsTestz=test that the various 'returning' flags are set appropriatelyTc C s( t d|tdtdddtdtd d S )Nr rM TFrN rP rQ rR rS r( r( r) rU s üz!ReturningGuardsTest.define_tablesc s | j j fdd}|S )Nc
|
s | jj¡} |r°|spttjd jj d jj d0 | |dddi|dddi|d dd
|
ig¡}W 5 Q R X n> | |dddi|dddi|d dd
|
ig¡}t
|
| ¡ d d d g nT|sàt tj | |dddi¡ W 5 Q R X n$ | |dddi¡}t
|
| ¡ d g d S )NzDialect z\+zW with current server capabilities does not support .*RETURNING when executemany is usedr rP rV r Úd2r Úd3)r )r )r )Z returningr rM r r ZStatementErrorr# r ZdriverrY r rÅ r r` )rµ Z executemanyZ id_param_nameZexpect_successÚresult©r$ r r( r) Úgo- sB þ
|
|
|
ýþ
|
|
|
ýþ
|
ÿ
|
ÿz(ReturningGuardsTest.run_stmt.<locals>.go)rZ r )r& r$ rÍ r( rÌ r) Úrun_stmt) s /z ReturningGuardsTest.run_stmtc C s& | j j}| ¡ }||dd|jj d S )NFrM )rZ r r[ r# Zinsert_returning©r& r$ rÎ r rµ r( r( r) Útest_insert_single^ s z&ReturningGuardsTest.test_insert_singlec C s& | j j}| ¡ }||dd|jj d S )NTrM )rZ r r[ r# Z insert_executemany_returningrÏ r( r( r) Útest_insert_manye s ÿz$ReturningGuardsTest.test_insert_manyc C s` | j j}| | ¡ dddddddddg¡ | ¡ |jjtdk¡}||d d|j j
|
d S ©
|
Nr rV rW r rÉ r rÊ Úb_idF) rZ r rY r[ Úupdater r rM r r# Zupdate_returningrÏ r( r( r) Útest_update_singlen s ýþ z&ReturningGuardsTest.test_update_singlec C s` | j j}| | ¡ dddddddddg¡ | ¡ |jjtdk¡}||d d|j j
|
d S ©
|
Nr rV rW r rÉ r rÊ rÓ T) rZ r rY r[ rÔ r r rM r r# Z update_executemany_returningrÏ r( r( r) Útest_update_many~ s ýþ ÿz$ReturningGuardsTest.test_update_manyc C s` | j j}| | ¡ dddddddddg¡ | ¡ |jjtdk¡}||d d|j j
|
d S rÒ ) rZ r rY r[ r r r rM r r# Zdelete_returningrÏ r( r( r) Útest_delete_single s ýþ z&ReturningGuardsTest.test_delete_singlec C s` | j j}| | ¡ dddddddddg¡ | ¡ |jjtdk¡}||d d|j j
|
d S rÖ ) rZ r rY r[ r r r rM r r# Z delete_executemany_returningrÏ r( r( r) Útest_delete_many s ýþ ÿz$ReturningGuardsTest.test_delete_manyN)r+ r, r- rG r. rh rU r rH rÎ rÐ rÑ rÕ r× rØ rÙ r( r( r( r) rÈ s
|
|
4 rÈ ).r3 Ú r r r r r r r
|
r r r Z
|
assertionsr r r Z provisionr Zschemar r r r r r r r r r Z sql.compilerr Úutilr ZTestBaser r/ Z
|
TablesTestrJ ri r r r ZFutureEngineMixinr¬ r rÈ r( r( r( r) Ú<module> sN
|
-:j>!C
|
ÿu
|
