U
|
¸ý°dÔ ã @ s d dl mZ d dl mZ d dl mZ d dl mZ d dl mZ d dl mZ d dl mZ d dl mZ d d l m Z d d
|
l
|
m Z d d l
|
m Z G d d d e j ZdS )é )Ú bindparam)ÚColumn)ÚInteger)ÚMetaData)Úselect)ÚString)ÚTable)Útesting)Útext)Úeq_)Úfixturesc @ s¸ e Zd ZdZdZdZedd Zedd Zdd Z d
|
d Z
|
d d Z e dddg¡e dde jjfde jjfg¡dd Zdd Zdd Zdd Ze jjdd Ze jjd d Zd S ) Ú RowCountTestztest rowcount functionality)Z sane_rowcountTc C s4 t d|tdtdddtdtdtdtd d S )
|
NÚ employeesÚ employee_idFT©Z autoincrementZ primary_keyÚnameé2 Ú
|
departmenté )r r r r )ÚclsÚmetadata© r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/testing/suite/test_rowcount.pyÚ define_tables s ü özRowCountTest.define_tablesc C sF ddddddddd g | _ }| jj}| | ¡ d
|
d t|D ¡ d S ) N)ZAngelaÚA)ZAndrewr )ZAnandr )ÚBobÚB)ZBobetter )ZBuffyr )ZCharlieÚC)ÚCynthiar )ZChrisr c S s g | ]\}\}}|||d qS ))r r r r )Ú.0ÚiÚnÚdr r r Ú
|
<listcomp>6 s
|
ÿz,RowCountTest.insert_data.<locals>.<listcomp>)ÚdataÚtablesr ÚexecuteÚinsertÚ enumerate)r Ú
|
connectionr$ Úemployees_tabler r r Ú insert_data% s" ÷
|
þþzRowCountTest.insert_datac C sB | j j}t|jj|jj |jj¡}| |¡ ¡ }t
|
|| j d S )N) r% r r Úcr r Zorder_byr r& Zfetchallr r$ )Úselfr) r* ÚsÚrowsr r r Ú
|
test_basic< s ÿþzRowCountTest.test_basicc C s@ | j j}|jj}| | ¡ |dk¡ddi¡}|jdks<td S )Nr r ÚZé ) r% r r, r r& ÚupdateÚwhereÚrowcountÚAssertionError©r- r) r* r Úrr r r Útest_update_rowcount1E s þz"RowCountTest.test_update_rowcount1c C s> | j j}|jj}| | ¡ |dk¡ddi¡}t|jd d S )Nr r r2 ) r% r r, r r& r3 r4 r r5 r7 r r r Útest_update_rowcount2P s þz"RowCountTest.test_update_rowcount2Úimplicit_returningFÚdmlr3 Údeletec C sº |r| j j}n6tdt tdtdddtdtdtdtd dd
|
}|jj}|j rz| ¡
|
|d k¡j |jjd d ¡ }n&|j r| ¡
|
|d k¡ ¡ }n| ¡ | |¡}t|jd dS )zØnote this test should succeed for all RETURNING backends
|
as of 2.0. In
|
Idf28379f8705e403a3c6a937f6a798a042ef2540 we changed rowcount to use
|
len(rows) when we have implicit returning
|
|
r r FTr r r r r )r; r r1 )r r2 N)r% r r r r r r r, r r3 r4 ÚvaluesZreturn_defaultsr= Zfailr& r r5 )r- r) r; r< r* r Ústmtr8 r r r Ú+test_update_delete_rowcount_return_defaults\ s>
|
ü õÿ
|
þÿÿÿ
|
z8RowCountTest.test_update_delete_rowcount_return_defaultsc C s | d¡}t|jd d S ©Nz8update employees set department='Z' where department='C'r2 )Zexec_driver_sqlr r5 ©r- r) Úresultr r r Útest_raw_sql_rowcount s ÿz"RowCountTest.test_raw_sql_rowcountc C s | td¡}t|jd d S rA )r& r
|
r r5 rB r r r Útest_text_rowcount s ÿz RowCountTest.test_text_rowcountc C s8 | j j}|jj}| | ¡ |dk¡¡}t|jd d S )Nr r2 ) r% r r, r r& r= r4 r r5 r7 r r r Útest_delete_rowcount¢ s ÿz!RowCountTest.test_delete_rowcountc C sX | j j}| ¡ |jjtdk¡jdd}| |ddiddiddig¡}t |j
|
d d S )NÚemp_namer )r r r Ú nonexistenté ) r% r r3 r4 r, r r r> r& r r5 ©r- r) r* r? r8 r r r Útest_multi_update_rowcount¬ s ÿþÿýþ z'RowCountTest.test_multi_update_rowcountc C sP | j j}| ¡ |jjtdk¡}| |ddiddiddig¡}t|j d d S )NrG r r rH rI )
|
r% r r= r4 r, r r r& r r5 rJ r r r Útest_multi_delete_rowcountÀ s ÿýþ z'RowCountTest.test_multi_delete_rowcountN)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__Ú __requires__Z __backend__Ú classmethodr r+ r0 r9 r: r Z variationÚrequiresZupdate_returningZdelete_returningr@ rD rE rF Zsane_multi_rowcountrK rL r r r r r s2
|
|
|
|
þþ 0
|
|
r N)Z
|
sqlalchemyr r r r r r r r r
|
Zsqlalchemy.testingr r Z
|
TablesTestr r r r r Ú<module> s
|
