U
|
¸ý°d? ã @ sî d dl Z ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm
|
Z
|
d d
|
lm Z d d lm Z d d lm Z d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ G dd dejZG dd dejZG dd dejejZdS )é Né )Úengines)Úfixtures)Úeq_)Ú requirements)ÚColumn)ÚTableé )ÚDateTime)Úfunc)ÚInteger)Úselect)Úsql)ÚString)Útesting)Útextc @ s^ e Zd ZdZedd Zedd Zdd Zdd Zd
|
d Z d d Z
|
e j dd Z dd ZdS )Ú RowFetchTestTc C sD t d|tdtddtdtd t d|tdtddtdt d S ) NÚplain_pkÚidT©Z primary_keyÚdataé2 Ú has_datesÚtoday)r r r r r
|
©ÚclsÚmetadata© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/testing/suite/test_results.pyÚ define_tables s ü üzRowFetchTest.define_tablesc C s\ | | jj ¡ dddddddddg¡ | | jj ¡ dt dd d
|
d
|
d d ¡d g¡ d S ) Né Úd1)r r r Zd2r Zd3éÖ é é r )r r )ÚexecuteÚtablesr Úinsertr Údatetime)r Ú
|
connectionr r r Ú insert_data' s
|
ýþ
|
þzRowFetchTest.insert_datac C s@ | | jj ¡ | jjjj¡¡ ¡ }t|jd t|j d d S ©Nr r! )
|
r% r& r r Úorder_byÚcr Úfirstr r ©Úselfr) Úrowr r r Ú test_via_attr7 s
|
ÿ zRowFetchTest.test_via_attrc C sH | | jj ¡ | jjjj¡¡ ¡ }t|j d d t|j d d d S )Nr r r r! )
|
r% r& r r r, r- r r. r Ú_mappingr/ r r r Útest_via_string? s
|
ÿz RowFetchTest.test_via_stringc C sD | | jj ¡ | jjjj¡¡ ¡ }t|d d t|d d d S )Nr r r! ) r% r& r r r, r- r r. r r/ r r r Ú test_via_intG s
|
ÿzRowFetchTest.test_via_intc C sX | | jj ¡ | jjjj¡¡ ¡ }t|j | jjjj d t|j | jjjj
|
d d S r+ ) r% r& r r r, r- r r. r r3 r r/ r r r Útest_via_col_objectO s
|
ÿz RowFetchTest.test_via_col_objectc C s\ | t| jjjj| jjjj d¡ | jjjj¡¡}| ¡ }t
|
| ¡ ddg t
|
|d d S )Nr )r! r! ) r% r r& r r- r Úlabelr, r r. r Úkeys)r0 r) Úresultr1 r r r Útest_row_with_dupe_namesW s
|
þ
|
ýÿz%RowFetchTest.test_row_with_dupe_namesc
|
C s` | j j}t| d¡jj ¡ }t|jj| d¡}| |¡
|
¡ }t |j t dddddd¡ dS )zýtest that a scalar select as a column is returned as such
|
and that type conversion works OK.
|
|
(this is half a SQLAlchemy Core test and half to catch database
|
backends that may have unusual behavior with scalar selects.)
|
|
ÚxÚ somelabelr" r# r$ r N)r& r r Úaliasr- r Zscalar_subqueryr r7 r% r. r r< r( )r0 r) Z datetableÚsÚs2r1 r r r Útest_row_w_scalar_selectc s
|
z%RowFetchTest.test_row_w_scalar_selectN)Ú__name__Ú
|
__module__Ú __qualname__Ú __backend__Ú classmethodr r* r2 r4 r5 r6 r Z%duplicate_names_in_cursor_descriptionr: r@ r r r r r s
|
|
|
r c @ sJ e Zd ZdZdZdZedd Zdd Zdd Z e
|
j d
|
d Z d d Z dS )ÚPercentSchemaNamesTestz¡tests using percent signs, spaces in table and column names.
|
|
This didn't work for PostgreSQL / MySQL drivers for a long time
|
but is now supported.
|
|
)Zpercent_schema_namesTc C s@ t d|tdttdt| j_t dt d¡t d¡¡| j_d S )Nz percent%tableúpercent%úspaces % more spaces) r r r r& Ú percent_tabler ÚtableÚcolumnÚlightweight_percent_tabler r r r r s üýz$PercentSchemaNamesTest.define_tablesc C sP | j j}ddddddddddddfD ]}| | ¡ |¡ q,| |¡ d S ©Nr# r$ ©rG rH é é é é
|
©r& rI r% r' Ú _assert_table)r0 r) rI Úparamsr r r Útest_single_roundtrip s üz,PercentSchemaNamesTest.test_single_roundtripc C sT | j j}| | ¡ ddd¡ | | ¡ dddddddddg¡ | |¡ d S rM rS )r0 r) rI r r r Útest_executemany_roundtrip s ÿýþz1PercentSchemaNamesTest.test_executemany_roundtripc C s| | j j}| | ¡ ddd¡ | | ¡ |jd |jd ¡ddddd ddddg¡}t| ¡ d
|
d d g | |¡ d S ) Nr# r$ rN rG rH rO rP rQ rR ©rO rP ©rQ rR ©rP rQ ) r& rI r% r' Z returningr- r ÚallrT )r0 r) rI r9 r r r Ú$test_executemany_returning_roundtrip§ s" ÿþýû z;PercentSchemaNamesTest.test_executemany_returning_roundtripc C sL | j j}| j j}|| ¡ || ¡ fD ]Ø}tt| | ¡ |j d ¡¡ddddg tt| | ¡
|
|j d ddg¡¡ |j d ¡¡ddg | | ¡ |j d ¡¡ ¡ }t|j d d t|j d d
|
t|j |j d d t|j |j d d
|
q$| | ¡ |j d d i¡¡ tt| | ¡ |j d ¡¡d d ddg d S )NrG )r# r$ rX rY rZ rH rQ rR r# r$ é )r# r] )rO r] )rQ r] )rP r] )r& rI rL r= r Úlistr% r r, r- ÚwhereZin_r. r3 ÚupdateÚvalues)r0 ÚconnrI rL rJ r1 r r r rT » s^ üÿ
|
üÿþÿÿø ÿ ÿÿÿÿÿ
|
øz$PercentSchemaNamesTest._assert_tableN)rA rB rC Ú__doc__Ú __requires__rD rE r rV rW r Z insert_executemany_returningr\ rT r r r r rF s s
|
|
rF c @ s e Zd ZdZdZdd Zdd Zejddde d dfd
|
de
|
d dfd d e d d fdd e
|
d j dddfdde
|
d j d dd fdde
|
d ¡ dfdddde d¡fdd e dd fdd e dj dddfddd dd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'S )(ÚServerSideCursorsTest)Úserver_side_cursorsTc C sÞ | j jjdkrt|jS | j jjdkr<tdjj}t||S | j jjdkrP|j S | j jjdkrttdjj}t||S | j jjdkr|j
|
S | j jjdkr|j S | j jjd kr¸t |d
|
d S | j jjd krÖtt |d d S d S d S )NZpsycopg2Zpymysqlzpymysql.cursors)ZaiomysqlZasyncmyZmysqldbzMySQLdb.cursorsZmariadbconnector)ZasyncpgZ aiosqliteZpg8000Ú server_sideFZpsycopgÚname) ÚengineÚdialectZdriverÚboolrh Ú
|
__import__ZcursorsZSSCursorÚ
|
isinstancerg ZbufferedÚgetattr)r0 ÚcursorZsscursorr r r Ú_is_server_sideú s&
|
|
|
z%ServerSideCursorsTest._is_server_sidec C sF |r.t d¡ tjd|id| _W 5 Q R X ntjd|id| _| jS )NzThe create_engine.server_side_cursors parameter is deprecated and will be removed in a future release. Please use the Connection.execution_options.stream_results parameter.rf )Úoptions)r Zexpect_deprecatedr Ztesting_engineri )r0 rf r r r Ú_fixture s ÿÿÿz ServerSideCursorsTest._fixture)Z global_stringTúselect 1TZ global_textrs Z global_exprr Zglobal_off_explicitFZ stmt_option©Zstream_resultsZstmt_option_disabledZfor_update_exprZfor_update_stringzSELECT 1 FOR UPDATEZsqliteZ
|
text_no_ssz select 42Ztext_ss_optionZiaaaz*engine_ss_arg, statement, cursor_ss_status)Zid_Úargnamesc C s\ | |¡}| ¡ @}t|tr*| |¡}n
|
| |¡}t| |j¡| | ¡ W 5 Q R X d S )N)
|
rr Úbeginrm ÚstrÚexec_driver_sqlr% r rp ro Úclose)r0 Z engine_ss_argZ statementZcursor_ss_statusri rb r9 r r r Útest_ss_cursor_status# s '
|
|
|
|
z+ServerSideCursorsTest.test_ss_cursor_statusc C sL | d¡}| ¡ 0}|jdd d¡}| |j¡s6t| ¡ W 5 Q R X d S )NFTrt rs )rr ÚconnectÚexecution_optionsrx rp ro ÚAssertionErrorry )r0 ri rb r9 r r r Útest_conn_optionS s
|
|
ÿþz&ServerSideCursorsTest.test_conn_optionc C sT | d¡}tdjdd}| ¡ (}|jdd |¡}| |j¡rFtW 5 Q R X d S )NFr Trt )rr r r| r{ r% rp ro r} )r0 ri r> rb r9 r r r Ú&test_stmt_enabled_conn_option_disabledk s
|
|
|
z<ServerSideCursorsTest.test_stmt_enabled_conn_option_disabledc C s¬ | d¡}tt d¡ d¡jdd ¡ }| ¡ ,}| | ¡ ¡}| |j
|
¡rRt | ¡ W 5 Q R X td |¡}| ¡ (}| |¡}| |j
|
¡rt | ¡ W 5 Q R X d S )NFÚ1r; Trt r )rr r r Zliteral_columnr7 r| Zsubqueryrv r% rp ro r} ry Ú select_from)r0 ri Ús1rb r9 r? r r r Útest_aliases_and_ssu s
|
ÿÿ
|
|
|
z)ServerSideCursorsTest.test_aliases_and_ssc C s* | j }| d¡}td|tdtddtdtd}| ¡ æ}|j|dd | | ¡ t
|
dd ¡ | | ¡ t
|
d
|
d ¡ t | | ¡ |jj¡¡ ¡ d d g | | ¡ |jjd k¡j|jjd d ¡ t | | ¡ |jj¡¡ ¡ d dg | | ¡ ¡ t | t t d¡ |¡¡d W 5 Q R X d S )NTÚ
|
test_tabler r r r ©Z
|
checkfirstÚdata1©r Údata2)r r )r r r z updated)r z data2 updatedÚ*r )r rr r r r r rv Úcreater% r' Údictr r r, r- r Úfetchallr` r_ ra r ÚdeleteZscalarr Úcountr )r0 r Úmdri r r) r r r Útest_roundtrip_fetchall sL
|
ü
|
ÿü
|
ÿ
|
þÿÿüÿüz-ServerSideCursorsTest.test_roundtrip_fetchallc C sî | j }| d¡}td|tdtddtdtd}| ¡ ª}|j|dd | | ¡ dd t
|
d
|
d D ¡ | | ¡ |j j¡¡}t| d ¡d d t
|
d
|
dD t| d¡dd t
|
ddD t| ¡ dd t
|
dd D W 5 Q R X d S )NTr r r r r r
c S s g | ]}t d | dqS )údata%dr )r ©Ú.0Úir r r Ú
|
<listcomp>À s zBServerSideCursorsTest.test_roundtrip_fetchmany.<locals>.<listcomp>r é r# c S s g | ]}|d | fqS ©r r r r r r r É s é rR c S s g | ]}|d | fqS r r r r r r r Í s é c S s g | ]}|d | fqS r r r r r r r Ï s )r rr r r r r rv r r% r' Úranger r, r- r r Z fetchmanyr )r0 r r ri r r) r9 r r r Útest_roundtrip_fetchmany± s4
|
ü
|
þÿþþz.ServerSideCursorsTest.test_roundtrip_fetchmanyN)rA rB rC rd rD rp rr r Ú combinationsr r r| Zwith_for_updateZskip_ifrz r~ r r r r r r r r re ò sR üüûüÞ$
|
|
(re )r( Ú r r Z
|
assertionsr Úconfigr Zschemar r r
|
r r r r r r r Z
|
TablesTestr rF ZTestBaseZAssertsExecutionResultsre r r r r Ú<module> s( ^
|
ÿ
|
