관리-도구
편집 파일: test_rowcount.cpython-38.pyc
U -?�f� � @ 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 e � d d d d dg�e � dddg�dd� ��Z dd� Z dd� Ze � dddg�e � dd e jjfd e jjfg�dd� ��Zdd � Zd d � Zd d � Ze jjd!d"� �Ze jjd#d$� �Zd%S )&� RowCountTestztest rowcount functionality)Z sane_rowcountTc C s4 t d|tdtddd�tdtd��tdtd ��� d S ) N� employees� employee_idFT�Z autoincrementZ primary_key�name�2 � department� )r r r r )�cls�metadata� r �V/opt/hc_python/lib64/python3.8/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>< 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_basicB s ��zRowCountTest.test_basic� statement�update�deleter( r � close_firstFc C s� | j j}|jj}|jr6|�|�� �|dk�ddi�}n�|jr\|�|�� �|dk�ddi�}nx|jr�|�|�� dddd�dd dd�d d dd�g�}nB|j r�t |jj |jj��|jjdk�}|�|�}|� � n|� � |r�|� � |jd ks�t�d S ) Nr r �Z� znone 1�Xr � znone 2� znone 3)���� )r& r r- r r3 r' �wherer4 r( r r �all�fail�close�rowcount�AssertionError)r. r* r2 r5 r+ r �rr/ r r r �$test_non_rowcount_scenarios_no_raiseK sB �� �� � � z1RowCountTest.test_non_rowcount_scenarios_no_raisec C s@ | j j}|jj}|�|�� �|dk�ddi�}|jdks<t�d S )Nr r r6 r< ) r& r r- r r' r3 r= rA rB �r. r* r+ r rC r r r �test_update_rowcount1v s �z"RowCountTest.test_update_rowcount1c C s> | j j}|jj}|�|�� �|dk�ddi�}t|jd� d S )Nr r r< ) r& r r- r r' r3 r= r rA rE r r r �test_update_rowcount2� s �z"RowCountTest.test_update_rowcount2�implicit_returning�dmlc C s� |r| j j}n6tdt� tdtddd�tdtd��tdtd ��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 )rH r r6 )r r< N)r& r r r r r r r- r r3 r= �valuesZreturn_defaultsr4 r? r' r rA )r. r* rH rI r+ r �stmtrC 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'r< )Zexec_driver_sqlr rA �r. r* �resultr r r �test_raw_sql_rowcount� s �z"RowCountTest.test_raw_sql_rowcountc C s |� td��}t|jd� d S rM )r' r r rA rN r r r �test_text_rowcount� s �z RowCountTest.test_text_rowcountc C s8 | j j}|jj}|�|�� �|dk��}t|jd� d S )Nr r< ) r& r r- r r' r4 r= r rA rE r r r �test_delete_rowcount� s �z!RowCountTest.test_delete_rowcountc C sX | j j}|�� �|jjtd�k�jdd�}|�|ddiddiddig�}t |j d� d S )N�emp_namer )r r r � nonexistent� ) r& r r3 r= r- r r rJ r' r rA �r. r* r+ rK rC r r r �test_multi_update_rowcount� s ����� z'RowCountTest.test_multi_update_rowcountc C sP | j j}|�� �|jjtd�k�}|�|ddiddiddig�}t|j d� d S )NrS r r rT rU ) r& r r4 r= r- r r r' r rA rV r r r �test_multi_delete_rowcount� s ��� z'RowCountTest.test_multi_delete_rowcountN)�__name__� __module__� __qualname__�__doc__� __requires__Z __backend__� classmethodr r, r1 r Z variationrD rF rG �requiresZupdate_returningZdelete_returningrL rP rQ rR Zsane_multi_rowcountrW rX r r r r r s8 ) �� 0 r N)Z sqlalchemyr r r r r r r r r Zsqlalchemy.testingr r Z TablesTestr r r r r �<module> s