Edit File: test_select.cpython-37.pyc
B ��4]N � @ s` d dl mZ d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ ddl m Z dd l m Z dd l m Z dd l m Z dd l m Z dd l mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ G dd� dej�ZG dd� dej�ZG dd� dej�ZG d d � d ej�ZG d d � d ej�Z G d d!� d!ej�Z d"S )#� )�config)�fixtures)�eq_)�in_)�Column)�Table� )� bindparam)�case)�false)�func)�Integer)�literal_column)�null)�select)�String)�testing)�true)�tuple_)�union)�utilc @ s@ e Zd ZdZedd� �Zedd� �Zdd� Zej j dd � �Z d S ) � CollateTestTc C s& t d|tdtdd�tdtd��� d S )N� some_table�idT)� primary_key�data�d )r r r r )�cls�metadata� r �W/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/suite/test_select.py� define_tables s zCollateTest.define_tablesc C s* t j�| jj�� ddd�ddd�g� d S )N� z collate data1)r r r z collate data2)r �db�execute�tablesr �insert)r r r r � insert_data# s zCollateTest.insert_datac C s t tj�|��� |� d S )N)r r r# r$ �fetchall)�selfr �resultr r r �_assert_result- s zCollateTest._assert_resultc C sD t j�t j�}| �t| jjg��| jjj j � |�� � �ddg� d S )N)r" z collate data1)r z collate data2) r �requiresZget_order_by_collationr r+ r r% r �order_by�cr �collateZasc)r) Z collationr r r �test_collate_order_by0 s z!CollateTest.test_collate_order_byN) �__name__� __module__� __qualname__� __backend__� classmethodr! r' r+ r r, Zorder_by_collationr0 r r r r r s r c @ sl e Zd ZdZdZedd� �Zedd� �Zdd� Zd d � Z d d � Z d d� Z dd� Z dd� Z ejjdd� �ZdS )�OrderByLabelTestz�Test the dialect sends appropriate ORDER BY expressions when labels are used. This essentially exercises the "supports_simple_order_by_label" setting. Tc C sB t d|tdtdd�tdt�tdt�tdtd��td td��� d S ) Nr r T)r �x�y�q�2 �p)r r r r )r r r r r r! G s z OrderByLabelTest.define_tablesc C sD t j�| jj�� dddddd�dddddd�ddd d d d�g� d S ) Nr" r Zq1Zp3)r r7 r8 r9 r; r Zq2�p2� Zq3�p1)r r# r$ r% r r& )r r r r r' S s z OrderByLabelTest.insert_datac C s t tj�|��� |� d S )N)r r r# r$ r( )r) r r* r r r r+ ^ s z OrderByLabelTest._assert_resultc C s8 | j j}|jj�d�}| �t|g��|�dddg� d S )N�lx)r" )r )r )r% r r. r7 �labelr+ r r- )r) �tabler? r r r � test_plaina s zOrderByLabelTest.test_plainc C s@ | j j}|jj|jj �d�}| �t|g��|�dddg� d S )Nr? )r )� )� ) r% r r. r7 r8 r@ r+ r r- )r) rA r? r r r �test_composed_intf s z"OrderByLabelTest.test_composed_intc C s� | j j}|jj|jj �d�}t�|jj�|jj �d�}| � t ||g�� ||� � �dt�d�fdt�d�fdt�d�fg� d S ) Nr? �lyr Zq1p3rC Zq2p2rD Zq3p1)r% r r. r7 r8 r@ r �lowerr9 r; r+ r r- �descr �u)r) rA r? rF r r r �test_composed_multiplek s z'OrderByLabelTest.test_composed_multiplec C s< | j j}|jj�d�}| �t|g��|�� �dddg� d S )Nr? )r )r )r" ) r% r r. r7 r@ r+ r r- rH )r) rA r? r r r �test_plain_desct s z OrderByLabelTest.test_plain_descc C sD | j j}|jj|jj �d�}| �t|g��|� � �dddg� d S )Nr? )rD )rC )r ) r% r r. r7 r8 r@ r+ r r- rH )r) rA r? r r r �test_composed_int_desc{ s z'OrderByLabelTest.test_composed_int_descc C sV | j j}|jj|jj �d�}tt�|jj �|g�� |�� |�}| � |dddg� d S )Nr? )r" r )r" rC )r" rD ) r% r r. r7 r8 r@ r r �countr Zgroup_byr- r+ )r) rA �expr�stmtr r r �test_group_by_composed� s z'OrderByLabelTest.test_group_by_composedN)r1 r2 r3 �__doc__r4 r5 r! r' r+ rB rE rJ rK rL r r, Zgroup_by_complex_expressionrP r r r r r6 < s r6 c @ s� e Zd ZdZedd� �Zedd� �Zddd�Zd d � Ze j j d d � �Z e j j d d� �Z e j j dd� �Ze j jdd� �Ze j jdd� �Ze j jdd� �ZdS )�LimitOffsetTestTc C s* t d|tdtdd�tdt�tdt�� d S )Nr r T)r r7 r8 )r r r )r r r r r r! � s z LimitOffsetTest.define_tablesc C sB t j�| jj�� dddd�dddd�dddd�dddd�g� d S )Nr" r )r r7 r8 r r= rC )r r# r$ r% r r& )r r r r r' � s zLimitOffsetTest.insert_datar c C s t tj�||��� |� d S )N)r r r# r$ r( )r) r r* �paramsr r r r+ � s z LimitOffsetTest._assert_resultc C s2 | j j}| �t|g��|jj��d�ddg� d S )Nr )r" r" r )r r r )r% r r+ r r- r. r �limit)r) rA r r r �test_simple_limit� s z!LimitOffsetTest.test_simple_limitc C s2 | j j}| �t|g��|jj��d�ddg� d S )Nr )r r r= )r= r= rC )r% r r+ r r- r. r �offset)r) rA r r r �test_simple_offset� s z"LimitOffsetTest.test_simple_offsetc C s8 | j j}| �t|g��|jj��d��d�ddg� d S )Nr r" )r r r )r r r= ) r% r r+ r r- r. r rT rV )r) rA r r r �test_simple_limit_offset� s z(LimitOffsetTest.test_simple_limit_offsetc C sZ | j j}t|g��|jj��d��d�}|jt j j ddid�}t |�}| � |ddg� dS ) z7test that 'literal binds' mode works - no bound params.r r" Z literal_bindsT)�dialectZcompile_kwargs)r r r )r r r= N)r% r r r- r. r rT rV �compiler r# rY �strr+ )r) rA rO Zsqlr r r �test_limit_offset_nobinds� s z)LimitOffsetTest.test_limit_offset_nobindsc C s> | j j}| jt|g��|jj��td��ddgddid� d S )N�l)r" r" r )r r r r )rS ) r% r r+ r r- r. r rT r )r) rA r r r �test_bound_limit� s z LimitOffsetTest.test_bound_limitc C s> | j j}| jt|g��|jj��td��ddgddid� d S )N�o)r r r= )r= r= rC r )rS ) r% r r+ r r- r. r rV r )r) rA r r r �test_bound_offset� s z!LimitOffsetTest.test_bound_offsetc C sJ | j j}| jt|g��|jj��td��� td��ddgddd�d� d S ) Nr] r_ )r r r )r r r= r r" )r] r_ )rS ) r% r r+ r r- r. r rT r rV )r) rA r r r �test_bound_limit_offset� s z'LimitOffsetTest.test_bound_limit_offsetN)r )r1 r2 r3 r4 r5 r! r' r+ rU r r, rV rW rX r\ Zbound_limit_offsetr^ r` ra r r r r rR � s rR c @ s� e Zd ZdZedd� �Zedd� �Zddd�Zd d � Zd d � Z e j j e j j d d� ��Ze j jdd� �Zdd� Ze j j dd� �Zdd� ZdS )�CompoundSelectTestTc C s* t d|tdtdd�tdt�tdt�� d S )Nr r T)r r7 r8 )r r r )r r r r r r! � s z CompoundSelectTest.define_tablesc C sB t j�| jj�� dddd�dddd�dddd�dddd�g� d S )Nr" r )r r7 r8 r r= rC )r r# r$ r% r r& )r r r r r' � s z CompoundSelectTest.insert_datar c C s t tj�||��� |� d S )N)r r r# r$ r( )r) r r* rS r r r r+ s z!CompoundSelectTest._assert_resultc C s` | j j}t|g��|jjdk�}t|g��|jjdk�}t||�}| �|�|jj�ddg� d S )Nr r )r r r )r r r= ) r% r r �wherer. r r r+ r- )r) rA �s1�s2�u1r r r �test_plain_union s z#CompoundSelectTest.test_plain_unionc C sh | j j}t|g��|jjdk�}t|g��|jjdk�}t||��� �� }| �|� |jj�ddg� d S )Nr r )r r r )r r r= ) r% r r rc r. r r �aliasr+ r- )r) rA rd re rf r r r � test_select_from_plain_union s z/CompoundSelectTest.test_select_from_plain_unionc C s� | j j}t|g��|jjdk��d��|jj�}t|g��|jjdk��d��|jj�}t||��d�}| � |�|jj�ddg� d S )Nr r" r )r r r )r r r= ) r% r r rc r. r rT r- r r+ )r) rA rd re rf r r r �&test_limit_offset_selectable_in_unions s z9CompoundSelectTest.test_limit_offset_selectable_in_unionsc C sz | j j}t|g��|jjdk��|jj�}t|g��|jjdk��|jj�}t||��d�}| � |�|jj�ddg� d S )Nr r )r r r )r r r= ) r% r r rc r. r r- r rT r+ )r) rA rd re rf r r r �"test_order_by_selectable_in_unions- s ""z5CompoundSelectTest.test_order_by_selectable_in_unionsc C sn | j j}t|g��|jjdk��� }t|g��|jjdk��� }t||��d�}| � |� |jj�ddg� d S )Nr r )r r r )r r r= ) r% r r rc r. r Zdistinctr rT r+ r- )r) rA rd re rf r r r �"test_distinct_selectable_in_unions6 s z5CompoundSelectTest.test_distinct_selectable_in_unionsc C s� | j j}t|g��|jjdk��d��|jj�}t|g��|jjdk��d��|jj�}t||�� � }| � |�� �d��|jj�ddg� d S )Nr r" r )r r r )r r r= ) r% r r rc r. r rT r- r rh r+ )r) rA rd re rf r r r �&test_limit_offset_in_unions_from_alias> s z9CompoundSelectTest.test_limit_offset_in_unions_from_aliasc C s� | j j}t|g��|jjdk��d��|jj��� �� }t|g��|jjdk��d��|jj��� �� }t ||��d�}| � |�|jj�ddg� d S )Nr r" r )r r r )r r r= ) r% r r rc r. r rT r- rh r r+ )r) rA rd re rf r r r �.test_limit_offset_aliased_selectable_in_unionsT s zACompoundSelectTest.test_limit_offset_aliased_selectable_in_unionsN)r )r1 r2 r3 r4 r5 r! r' r+ rg ri r r, Zorder_by_col_from_unionZ/parens_in_union_contained_select_w_limit_offsetrj Z0parens_in_union_contained_select_wo_limit_offsetrk rl rm rn r r r r rb � s rb c @ s� e Zd ZdZedd� �Zedd� �Zd dd�Zd d � Ze j j d d � �Z e j j d d� �Z dd� Ze j j dd� �Ze j j dd� �Zdd� Zdd� Zdd� Zdd � Zd d � Zd S )!�ExpandingBoundInTestTc C s6 t d|tdtdd�tdt�tdt�tdtd��� d S ) Nr r T)r r7 r8 �zr: )r r r r )r r r r r r! n s z"ExpandingBoundInTest.define_tablesc C sJ t j�| jj�� ddddd�ddddd�ddddd�ddd d d�g� d S ) Nr" r Zz1)r r7 r8 rp r �z2r= �z3rC �z4)r r# r$ r% r r& )r r r r r' y s z ExpandingBoundInTest.insert_datar c C s t tj�||��� |� d S )N)r r r# r$ r( )r) r r* rS r r r r+ � s z#ExpandingBoundInTest._assert_resultc C sj | j j}t|jjg��|jj�tddd����|jj �tddd���� |jj�}| j |g g g d�d� d S )Nr9 T)� expandingr; )r9 r; )rS ) r% r r r. r rc r7 r r r8 r- r+ )r) rA rO r r r �test_multiple_empty_sets� s z-ExpandingBoundInTest.test_multiple_empty_setsc C sZ | j j}t|jjg��t|jj|jj�� t ddd���� |jj�}| j |g dg id� d S )Nr9 T)rt )rS ) r% r r r. r rc r r7 rp r r r- r+ )r) rA rO r r r � test_empty_heterogeneous_tuples� s z4ExpandingBoundInTest.test_empty_heterogeneous_tuplesc C sZ | j j}t|jjg��t|jj|jj�� t ddd���� |jj�}| j |g dg id� d S )Nr9 T)rt )rS ) r% r r r. r rc r r7 r8 r r r- r+ )r) rA rO r r r � test_empty_homogeneous_tuples� s z2ExpandingBoundInTest.test_empty_homogeneous_tuplesc C s\ | j j}t|jjg��|jj�tddd���� |jj�}| j |dddgdddd gid � d S ) Nr9 T)rt )r )r )r= r r r= )rS ) r% r r r. r rc r7 r r r- r+ )r) rA rO r r r �test_bound_in_scalar� s z)ExpandingBoundInTest.test_bound_in_scalarc C sf | j j}t|jjg��t|jj|jj�� t ddd���� |jj�}| j |dddgdddd gid � d S ) Nr9 T)rt )r )r )r= )r r )r r= )r= rC )rS ) r% r r r. r rc r r7 r8 r r r- r+ )r) rA rO r r r �test_bound_in_two_tuple� s z,ExpandingBoundInTest.test_bound_in_two_tuplec C sf | j j}t|jjg��t|jj|jj�� t ddd���� |jj�}| j |dddgdddd gid � d S ) Nr9 T)rt )r )r )r= )r rq )r rr )r= rs )rS ) r% r r r. r rc r r7 rp r r r- r+ )r) rA rO r r r �%test_bound_in_heterogeneous_two_tuple� s z:ExpandingBoundInTest.test_bound_in_heterogeneous_two_tuplec C sP | j j}t|jjg��|jj�tddd���� |jj�}| j |g dg id� d S )Nr9 T)rt )rS ) r% r r r. r rc r7 r r r- r+ )r) rA rO r r r � test_empty_set_against_integer� s z3ExpandingBoundInTest.test_empty_set_against_integerc C sX | j j}t|jjg��|jj�tddd���� |jj�}| j |ddddgdg id� d S ) Nr9 T)rt )r" )r )r )r= )rS ) r% r r r. r rc r7 �notin_r r- r+ )r) rA rO r r r �'test_empty_set_against_integer_negation� s z<ExpandingBoundInTest.test_empty_set_against_integer_negationc C sP | j j}t|jjg��|jj�tddd���� |jj�}| j |g dg id� d S )Nr9 T)rt )rS ) r% r r r. r rc rp r r r- r+ )r) rA rO r r r � test_empty_set_against_string� s z2ExpandingBoundInTest.test_empty_set_against_stringc C sX | j j}t|jjg��|jj�tddd���� |jj�}| j |ddddgdg id� d S ) Nr9 T)rt )r" )r )r )r= )rS ) r% r r r. r rc rp r| r r- r+ )r) rA rO r r r �&test_empty_set_against_string_negation s z;ExpandingBoundInTest.test_empty_set_against_string_negationc C sL t tt� �tdddd��t� fgt� d�g�}ttj� |�� � d d� d S )NZfoor T)�valuert )Zelse_� )Fr� ) r r r r r r r r r# r$ Zfetchone)r) rO r r r � test_null_in_empty_set_is_false s z4ExpandingBoundInTest.test_null_in_empty_set_is_falseN)r )r1 r2 r3 r4 r5 r! r' r+ ru r r, Ztuple_inrv rw rx ry rz r{ r} r~ r r� r r r r ro k s ro c @ s� e Zd ZdZdZdZedd� �Zedd� �Zdd � Z d d � Z d d � Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� ZdS )&�LikeFunctionsTestT�onceNc C s& t d|tdtdd�tdtd��� d S )Nr r T)r r r: )r r r r )r r r r r r! , s z LikeFunctionsTest.define_tablesc C sj t j�| jj�� ddd�ddd�ddd�dd d�d d d�d d d�ddd�ddd�ddd�ddd�g � d S )Nr" Zabcdefg)r r r zab/cdefgr zab%cdefgr= Zab_cdefgrC zabcde/fg� zabcde%fgrD zab#cdefg� Zab9cdefg� zabcde#fg� Zabcd9fg)r r# r$ r% r r& )r r r r r' 5 s z LikeFunctionsTest.insert_datac C sP | j j}tj�� �*}dd� |�t|jjg�� |��D �}W d Q R X t ||� d S )Nc S s h | ] \}|�qS r r )�.0r� r r r � <setcomp>K s z*LikeFunctionsTest._test.<locals>.<setcomp>) r% r r r# �connectr$ r r. r rc r )r) rN �expectedr ZconnZrowsr r r �_testG s $zLikeFunctionsTest._testc C s6 | j jjj}| �|�d�dddddddd d d h � d S ) Nzab%cr" r r r= rC r� rD r� r� r� )r% r r. r r� � startswith)r) �colr r r �test_startswith_unescapedT s z+LikeFunctionsTest.test_startswith_unescapedc C s( | j jjj}| �|jddd�dh� d S )Nzab%cT)� autoescaper )r% r r. r r� r� )r) r� r r r �test_startswith_autoescapeX s z,LikeFunctionsTest.test_startswith_autoescapec C s: | j jjj}| �|�td��dddddddd d d h � d S ) Nz'ab%c'r" r r r= rC r� rD r� r� r� )r% r r. r r� r� r )r) r� r r r �test_startswith_sqlexpr\ s z)LikeFunctionsTest.test_startswith_sqlexprc C s( | j jjj}| �|jddd�dh� d S )Nzab##c�#)�escaperD )r% r r. r r� r� )r) r� r r r �test_startswith_escapec s z(LikeFunctionsTest.test_startswith_escapec C sD | j jjj}| �|jdddd�dh� | �|jdddd�dh� d S )Nzab%cTr� )r� r� r zab#crD )r% r r. r r� r� )r) r� r r r �!test_startswith_autoescape_escapeg s z3LikeFunctionsTest.test_startswith_autoescape_escapec C s4 | j jjj}| �|�d�dddddddd d h � d S ) Nze%fgr" r r r= rC r� rD r� r� )r% r r. r r� �endswith)r) r� r r r �test_endswith_unescapedl s z)LikeFunctionsTest.test_endswith_unescapedc C s8 | j jjj}| �|�td��dddddddd d h � d S ) Nz'e%fg'r" r r r= rC r� rD r� r� )r% r r. r r� r� r )r) r� r r r �test_endswith_sqlexprp s z'LikeFunctionsTest.test_endswith_sqlexprc C s( | j jjj}| �|jddd�dh� d S )Nze%fgT)r� r� )r% r r. r r� r� )r) r� r r r �test_endswith_autoescapev s z*LikeFunctionsTest.test_endswith_autoescapec C s( | j jjj}| �|jddd�dh� d S )Nze##fgr� )r� r� )r% r r. r r� r� )r) r� r r r �test_endswith_escapez s z&LikeFunctionsTest.test_endswith_escapec C sD | j jjj}| �|jdddd�dh� | �|jdddd�dh� d S )Nze%fgTr� )r� r� r� ze#fgr� )r% r r. r r� r� )r) r� r r r � test_endswith_autoescape_escape~ s z1LikeFunctionsTest.test_endswith_autoescape_escapec C s4 | j jjj}| �|�d�dddddddd d h � d S ) Nzb%cder" r r r= rC r� rD r� r� )r% r r. r r� �contains)r) r� r r r �test_contains_unescaped� s z)LikeFunctionsTest.test_contains_unescapedc C s( | j jjj}| �|jddd�dh� d S )Nzb%cdeT)r� r )r% r r. r r� r� )r) r� r r r �test_contains_autoescape� s z*LikeFunctionsTest.test_contains_autoescapec C s( | j jjj}| �|jddd�dh� d S )Nzb##cder� )r� rD )r% r r. r r� r� )r) r� r r r �test_contains_escape� s z&LikeFunctionsTest.test_contains_escapec C sD | j jjj}| �|jdddd�dh� | �|jdddd�dh� d S )Nzb%cdTr� )r� r� r zb#cdrD )r% r r. r r� r� )r) r� r r r � test_contains_autoescape_escape� s z1LikeFunctionsTest.test_contains_autoescape_escape)r1 r2 r3 r4 Z run_insertsZ run_deletesr5 r! r' r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� & s( r� N) � r r Z assertionsr r Zschemar r r r r r r r r r r r r r r r Z TablesTestr r6 rR rb ro r� r r r r �<module> s4 %R_~ <