Edit File: test_results.cpython-37.pyc
B ��4]�, � @ s� d dl Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd lm Z d d lm Z d d lm Z d d lm Z d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ G dd� dej�ZG dd� dej�ZG dd� dejej�ZdS )� N� )�config)�engines)�fixtures)�eq_)� requirements)�Column)�Table� )�DateTime)�func)�Integer)�select)�sql)�String)�testing)�textc @ sV e Zd ZdZedd� �Zedd� �Zdd� Zdd � Zd d � Z e j d d � �Z dd� Z dS )� 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 �X/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/suite/test_results.py� define_tables s zRowFetchTest.define_tablesc C s` t j�| jj�� ddd�ddd�ddd�g� t j�| jj�� dt�dd d d d d �d �g� d S ) N� �d1)r r r �d2r Zd3i� � � r )r r )r �db�execute�tablesr �insertr �datetime)r r r r � insert_data& s zRowFetchTest.insert_datac C sF t j�| jj�� �| jjjj��� � }t |d d� t |d d� d S )Nr r r r! ) r r% r& r' r r �order_by�cr �firstr )�self�rowr r r �test_via_string6 s z RowFetchTest.test_via_stringc C sF t j�| 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/ r r r � test_via_int> s zRowFetchTest.test_via_intc C sV t j�| jj�� �| jjjj��� � }t || jjjj d� t || jjjj d� d S )Nr r! ) r r% r& r' r r r+ r, r r- r r )r. r/ r r r �test_via_col_objectF s z RowFetchTest.test_via_col_objectc C s` t j�t| jjjj| jjjj�d�g�� | jjjj ��}|� � }t |� � ddg� t |d� d S )Nr )r! r! )r r% r& r r' r r, r �labelr+ r r- r �keys)r. �resultr/ r r r �test_row_with_dupe_namesN s z%RowFetchTest.test_row_with_dupe_namesc C sh | j j}t|�d�jjg��� }t|jj|�d�g�}t j � |�� � }t |d 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.) �xZ somelabeli� r# r$ r N)r' r r �aliasr, r Z as_scalarr r3 r r% r& r- r r) )r. Z datetable�s�s2r/ r r r �test_row_w_scalar_select\ s z%RowFetchTest.test_row_w_scalar_selectN)�__name__� __module__� __qualname__� __backend__� classmethodr r* r0 r1 r2 r Z%duplicate_names_in_cursor_descriptionr6 r; r r r r r s r c @ s< e Zd ZdZdZdZedd� �Zdd� Zdd � Z d d � Z d S ) �PercentSchemaNamesTestz�tests using percent signs, spaces in table and column names. This is a very fringe use case, doesn't work for MySQL or PostgreSQL. the requirement, "percent_schema_names", is marked "skip" by default. )Zpercent_schema_namesTc C s@ t d|tdt�tdt��| j_t�dt�d�t�d��| j_d S )Nz percent%tablezpercent%zspaces % more spaces) r r r r' � percent_tabler �table�column�lightweight_percent_table)r r r r r r y s z$PercentSchemaNamesTest.define_tablesc C sT | j j}x>ddd�ddd�ddd�ddd�gD ]}tj�|�� |� q.W | �� d S )Nr# r$ )zpercent%zspaces % more spaces� � � � )r' rB r r% r& r( � _assert_table)r. rB �paramsr r r �test_single_roundtrip� s z,PercentSchemaNamesTest.test_single_roundtripc C sV | j j}tj�|�� ddd�� tj�|�� ddd�ddd�ddd�g� | �� d S )Nr# r$ )zpercent%zspaces % more spacesrF rG rH rI )r' rB r r% r& r( rJ )r. rB r r r �test_executemany_roundtrip� s z1PercentSchemaNamesTest.test_executemany_roundtripc C sR | j j}| j j}x�||�� ||�� fD ]�}tttj�|� � � |j d ���ddddg� tttj�|� � � |j d � ddg��� |j d ���ddg� tj�|� � � |j d ���� }t|d d � t|d d � t||j d d � t||j d d � q&W tj�|�� �|j d d i�� tttj�|� � � |j d ���d d ddg� d S )Nzpercent%)r# r$ )rF rG )rH rI )rG rH zspaces % more spacesrH rI r# r$ � )r# rN )rF rN )rH rN )rG rN )r' rB rE r8 r �listr r% r& r r+ r, �whereZin_r- �update�values)r. rB rE rC r/ r r r rJ � sD z$PercentSchemaNamesTest._assert_tableN) r< r= r> �__doc__� __requires__r? r@ r rL rM rJ r r r r rA l s rA c @ s� e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zejd#d$� �Zd%S )&�ServerSideCursorsTest)�server_side_cursorsTc C sd | j jjdkr|jS | j jjdkr8td�jj}t||�S | j jjdkr\td�jj}t||�S dS d S )N�psycopg2�pymysqlzpymysql.cursorsZmysqldbzMySQLdb.cursorsF)�engineZdialectZdriver�name� __import__ZcursorsZSSCursor� isinstance)r. �cursorZsscursorr r r �_is_server_side� s z%ServerSideCursorsTest._is_server_sidec C s t jd|id�| _| jS )NrV )�options)r Ztesting_enginerY )r. rV r r r �_fixture� s z ServerSideCursorsTest._fixturec C s t j�� | j�� d S )N)r Ztesting_reaperZ close_allrY Zdispose)r. r r r �tearDown� s z ServerSideCursorsTest.tearDownc C s( | � d�}|�d�}| �|j�s$t�d S )NTzselect 1)r` r& r^ r] �AssertionError)r. rY r5 r r r �test_global_string� s z(ServerSideCursorsTest.test_global_stringc C s, | � d�}|�td��}| �|j�s(t�d S )NTzselect 1)r` r& r r^ r] rb )r. rY r5 r r r �test_global_text� s z&ServerSideCursorsTest.test_global_textc C s. | � d�}|�tdg��}| �|j�s*t�d S )NTr )r` r& r r^ r] rb )r. rY r5 r r r �test_global_expr s z&ServerSideCursorsTest.test_global_exprc C s, | � d�}|�td��}| �|j�r(t�d S )NFzselect 1)r` r& r r^ r] rb )r. rY r5 r r r �test_global_off_explicit s z.ServerSideCursorsTest.test_global_off_explicitc C s: | � d�}tdg�jdd�}|�|�}| �|j�s6t�d S )NFr T)�stream_results)r` r �execution_optionsr& r^ r] rb )r. rY r9 r5 r r r �test_stmt_option s z&ServerSideCursorsTest.test_stmt_optionc C s4 | � d�}|�� jdd��d�}| �|j�s0t�d S )NFT)rg zselect 1)r` �connectrh r& r^ r] rb )r. rY r5 r r r �test_conn_option s z&ServerSideCursorsTest.test_conn_optionc C sF | � d�}tdg�jdd�}|�� jdd��|�}| �|j�rBt�d S )NFr T)rg )r` r rh rj r& r^ r] rb )r. rY r9 r5 r r r �&test_stmt_enabled_conn_option_disabled$ s z<ServerSideCursorsTest.test_stmt_enabled_conn_option_disabledc C s: | � d�}tdg�jdd�}|�|�}| �|j�r6t�d S )NTr F)rg )r` r rh r& r^ r] rb )r. rY r9 r5 r r r �test_stmt_option_disabled/ s z/ServerSideCursorsTest.test_stmt_option_disabledc C sf | � d�}tdg�jdd��� }|�|�}| �|j�s:t�tdg|d�}|�|�}| �|j�rbt�d S )NFr T)rg )Zfrom_obj)r` r rh r8 r& r^ r] rb )r. rY �s1r5 r: r r r �test_aliases_and_ss5 s z)ServerSideCursorsTest.test_aliases_and_ssc C s6 | � d�}tdg��� }|�|�}| �|j�s2t�d S )NTr )r` r Zwith_for_updater& r^ r] rb )r. rY rn r5 r r r �test_for_update_exprA s z*ServerSideCursorsTest.test_for_update_exprc C s( | � d�}|�d�}| �|j�s$t�d S )NTzSELECT 1 FOR UPDATE)r` r& r^ r] rb )r. rY r5 r r r �test_for_update_stringG s z,ServerSideCursorsTest.test_for_update_stringc C s0 | � d�}td�}|�|�}| �|j�r,t�d S )NFz select 42)r` r r& r^ r] rb )r. rY r9 r5 r r r �test_text_no_ssL s z%ServerSideCursorsTest.test_text_no_ssc C s8 | � d�}td�jdd�}|�|�}| �|j�s4t�d S )NFz select 42T)rg )r` r rh r& r^ r] rb )r. rY r9 r5 r r r �test_text_ss_optionR s z)ServerSideCursorsTest.test_text_ss_optionc C s | j }| �d� td|tdtdd�tdtd���}|jdd� |�� jdd � |�� jd d � t |� � � |j j ��� �� d d g� |�� �|j j d k�j|j jd d ��� t |� � � |j j ��� �� d dg� |�� �� t t t�d�g��|��� d� d S )NT� test_tabler )r r r )Z checkfirst�data1)r �data2)r ru )r rv r z updated)r z data2 updated�*r )r r` r r r r �creater( r& r r r+ r, r ZfetchallrQ rP rR r �deleter �countZ select_fromZscalar)r. Zmdrt r r r �test_roundtripX s( z$ServerSideCursorsTest.test_roundtripN)r< r= r> rT r? r^ r` ra rc rd re rf ri rk rl rm ro rp rq rr rs r Zprovide_metadatar{ r r r r rU � s& rU )r) � r r r Z assertionsr r Zschemar r r r r r r r r r Z TablesTestr rA ZTestBaseZAssertsExecutionResultsrU r r r r �<module> s& Xn