관리-도구
편집 파일: test_results.cpython-38.pyc
U -?�fB � @ s� d dl Z d dlZddlmZ ddlmZ ddlmZ ddlmZ ddl m Z ddl m Z d d lm Z d d lm Z d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ G dd� dej�ZG dd� dej�ZG dd� dejej�ZdS )� 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|tdtdd�tdtd��� t d|tdtdd�tdt�� d S ) N�plain_pk�idT)� primary_key�data�2 � has_dates�today)r r r r r ��cls�metadata� r �U/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/suite/test_results.py� define_tables s � �zRowFetchTest.define_tablesc C s\ |� | jj�� ddd�ddd�ddd�g� |� | 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_attr> 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 �_mappingr0 r r r �test_via_stringF 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 r0 r r r � test_via_intN 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 r4 r r0 r r r �test_via_col_objectV 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)r1 r* �resultr2 r r r �test_row_with_dupe_names^ 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 r8 r& r/ r r= r) )r1 r* Z datetable�s�s2r2 r r r �test_row_w_scalar_selectj s z%RowFetchTest.test_row_w_scalar_selectN)�__name__� __module__� __qualname__� __backend__� classmethodr r+ r3 r5 r6 r7 r Z%duplicate_names_in_cursor_descriptionr; rA 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|tdt�tdt��| 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}ddd�ddd�ddd�ddd�fD ]}|�|�� |� q,| �|� d S �Nr$ r% �rH rI � � � � �r' rJ r& r( � _assert_table)r1 r* rJ �paramsr r r �test_single_roundtrip� s �z,PercentSchemaNamesTest.test_single_roundtripc C sT | j j}|�|�� ddd�� |�|�� ddd�ddd�ddd�g� | �|� d S rN rT )r1 r* rJ r r r �test_executemany_roundtrip� s ���z1PercentSchemaNamesTest.test_executemany_roundtripc C s| | j j}|�|�� ddd�� |�|�� �|jd |jd �ddd�dd d�ddd�g�}t|�� d d d g� | �|� d S ) Nr$ r% rO rH rI rP rQ rR rS �rP rQ �rR rS �rQ rR ) r' rJ r& r( Z returningr. r �allrU )r1 r* rJ r: 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 )NrH )r$ r% rY rZ r[ rI rR rS r$ r% � )r$ r^ )rP r^ )rR r^ )rQ r^ )r' rJ rM r> r �listr& r r- r. �whereZin_r/ r4 �update�values)r1 �connrJ rM rK r2 r r r rU � s^ �� ������ � ����� �z$PercentSchemaNamesTest._assert_tableN)rB rC rD �__doc__� __requires__rE rF r rW rX r Z insert_executemany_returningr] rU r r r r rG z s rG c @ s e Zd ZdZdZdd� Zdd� Zdd� Zej d dd d � dfd dd d � dfdde d�dfdddd � dfdde d�j dd�dfdde d�j dd�dfdde d�� � dfdddd � de� ddg�fddd d � dfd dd d � dfd d d!� d"d#� �Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.S )/�ServerSideCursorsTest)�server_side_cursorsTc C s� | j jjdkrt|j�S | j jjdkr<td�jj}t||�S | j jjdkrP|j S | j jjdkrttd�jj}t||�S | j jjdkr�|j S | j jjdkr�|j S | j jjd kr�t |d d �S | j jjd kr�tt |d d ��S | j jjdk� r�t |d d �S d S d S )NZpsycopg2Zpymysqlzpymysql.cursors)ZaiomysqlZasyncmyZaioodbcZmysqldbzMySQLdb.cursorsZmariadbconnector)ZasyncpgZ aiosqliteZpg8000� server_sideFZpsycopg�nameZoracledb) �engine�dialectZdriver�boolri � __import__ZcursorsZSSCursor� isinstancerh �buffered�getattr)r1 �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 )Nz�The 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.rg )�options)r Zexpect_deprecatedr Ztesting_enginerj )r1 rg r r r �_fixture s ���z ServerSideCursorsTest._fixturec C s t �dt j��dd� |�S )Nz SELECT (\d+)c S s t tt| �d����tj��S )Nr! )�strr �int�group�compiler �db)�mr r r �<lambda>- � z1ServerSideCursorsTest.stringify.<locals>.<lambda>)�rerx �I�sub)r1 Zstr_r r r � stringify+ s �z ServerSideCursorsTest.stringifyZ global_stringc C s | d�S �N�select 1r �r� r r r r{ 1 r| z ServerSideCursorsTest.<lambda>Z global_textc C s t | d��S r� r r� r r r r{ 5 r| Z global_exprr! Zglobal_off_explicitFc C s t | d��S r� r r� r r r r{ < r| Z stmt_option�Zstream_resultsZstmt_option_disabledZfor_update_exprZfor_update_stringc C s | d�S )NzSELECT 1 FOR UPDATEr r� r r r r{ P r| ZsqliteZmssqlZ text_no_ssc C s t | d��S )N� select 42r r� r r r r{ W r| Ztext_ss_optionc C s t | d��jdd�S )Nr� Tr� )r �execution_optionsr� r r r r{ ] s �Ziaaaz*engine_ss_arg, statement, cursor_ss_status)Zid_�argnamesc C st | � |�}|�� �X}t|�r,tj|| jd�}t|t�rB|�|�}n |� |�}t | � |j �|� |� � W 5 Q R X d S )Nr� )rt �begin�callabler Zresolve_lambdar� rn ru �exec_driver_sqlr& r rr rq �close)r1 Z engine_ss_argZ statementZcursor_ss_statusrj rc r: r r r �test_ss_cursor_status0 s 8 � z+ServerSideCursorsTest.test_ss_cursor_statusc C sR | � d�}|�� �6}|jdd��| �d��}| �|j�s<t�|�� W 5 Q R X d S )NFTr� r� ) rt �connectr� r� r� rr rq �AssertionErrorr� )r1 rj rc r: r r r �test_conn_optionv s ��z&ServerSideCursorsTest.test_conn_optionc C sT | � d�}td�jdd�}|�� �(}|jdd��|�}| �|j�rFt�W 5 Q R X d S )NFr! Tr� )rt r r� r� r& rr rq r� )r1 rj r? rc r: r r r �&test_stmt_enabled_conn_option_disabled� 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 �r�t �|� � W 5 Q R X d S )NF�1r< Tr� r! )rt r r Zliteral_columnr8 r� Zsubqueryr� r& rr rq r� r� � select_from)r1 rj �s1rc r: r@ r r r �test_aliases_and_ss� s �� z)ServerSideCursorsTest.test_aliases_and_ssc C s, | j }| �d�}td|tdtddd�tdtd���}|�� ��}|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 Ztest_needs_autoincrementr r �Z checkfirst�data1�r �data2)r! r� )r r� r z updated)r z data2 updated�*r )r rt r r r r r� �creater& r( �dictr r r- r. r �fetchallra r` rb r �deleteZscalarr �countr� )r1 r �mdrj r� r* r r r �test_roundtrip_fetchall� sV � � �� � ������z-ServerSideCursorsTest.test_roundtrip_fetchallc C s� | j }| �d�}td|tdtddd�tdtd���}|�� ��}|j|dd� |�|� � dd � t d d �D �� |�|� � � |j j��}t|�d �d d � t d d�D �� t|�d�dd � t dd�D �� t|�� dd � t dd �D �� W 5 Q R X d S )NTr� r r� r r r� c S s g | ]}t d | d��qS )�data%dr� )r� ��.0�ir r r � <listcomp>� s zBServerSideCursorsTest.test_roundtrip_fetchmany.<locals>.<listcomp>r! � r$ c S s g | ]}|d | f�qS �r� r r� r r r r� � s � rS c S s g | ]}|d | f�qS r� r r� r r r r� � s � c S s g | ]}|d | f�qS r� r r� r r r r� � s )r rt r r r r r� r� r& r( �ranger r- r. r r Z fetchmanyr� )r1 r r� rj r� r* r: r r r �test_roundtrip_fetchmany� s> � � ����z.ServerSideCursorsTest.test_roundtrip_fetchmanyN)rB rC rD re rE rr rt r� r � combinationsr r� Zwith_for_updateZskip_ifr� r� r� r� r� r� r r r r rf � sl ���� ����5 *rf )r) r} � r r Z assertionsr �configr Zschemar r r r r r r r r r Z TablesTestr rG ZTestBaseZAssertsExecutionResultsrf r r r r �<module> s* ^ �