Edit File: test_cte.cpython-37.pyc
B ��4]� � @ s� d dl mZ d dl mZ d dlmZ d dlmZ d dlmZ ddl mZ ddl m Z dd l m Z dd l m Z dd l m Z G d d � d ej �ZdS )� )�config)�fixtures)�eq_)�Column)�Table� )� ForeignKey)�Integer)�select)�String)�testingc @ s� e Zd ZdZdZdZdZedd� �Zedd� �Z dd � Z d d � Z d d � Z e jje jjdd� ��Ze jje jjdd� ��Ze jjdd� �ZdS )�CTETestT)ZctesZeachc C s\ t d|tdtdd�tdtd��tdtd��� t d |tdtdd�tdtd��tdt�� d S ) N� some_table�idT)Z primary_key�data�2 � parent_idz some_table.id�some_other_table)r r r r r )�cls�metadata� r �T/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/suite/test_cte.py� define_tables s zCTETest.define_tablesc C sL t j�| jj�� ddd d�dddd�dddd�dd dd�d d dd�g� d S ) N� �d1)r r r r �d2r �d3� �d4� �d5)r �db�execute�tablesr �insert)r r r r � insert_data& s zCTETest.insert_datac C s� | j j}tj�� �d}t|g��|jj� dddg��� d�}|� t|jjg��|jj� ddg���}t |� � dg� W d Q R X d S )Nr r r �some_cter )r )r# r r r! �connectr �where�cr �in_�cter" r �fetchall)�selfr �connr+ �resultr r r �#test_select_nonrecursive_round_trip3 s $z+CTETest.test_select_nonrecursive_round_tripc C s� | j j}tj�� ��}t|g��|jj� dddg��j ddd�}|� d�}|� � }|� t|g��|jj |jjk��}|�t|jjg��|jjdk��|jj�� ��}t|�� dd d d d d g� W d Q R X d S ) Nr r r r& T)� recursiveZc1)r )r )r )r# r r r! r' r r( r) r r* r+ �aliasZ union_allr r r" �order_by�descr r, )r- r r. r+ Z cte_aliasZst1r/ r r r � test_select_recursive_round_tripA s z(CTETest.test_select_recursive_round_tripc C s� | j j}| j j}tj�� �x}t|g��|jj � dddg��� d�}|� |� � �dddgt|g��� t|� t|g��|jj���� dd d g� W d Q R X d S ) Nr r r r& r r r )r r r )r r r )r r r )r# r r r r! r' r r( r) r r* r+ r" r$ � from_selectr r3 r r, )r- r r r. r+ r r r �"test_insert_from_select_round_trip\ s z*CTETest.test_insert_from_select_round_tripc C s� | j j}| j j}tj�� ��}|�|�� �dddgt |g��� t |g�� |j j � dddg���d�}|�|�� jdd �� |j j |j j k�� t|�t |g��|j j���� d d d d dg� W d Q R X d S )Nr r r r r r r& r )r )r r N)r r r )r r r )r r r )r r r )r# r r r r! r' r" r$ r6 r r( r) r r* r+ �update�valuesr r3 r r, )r- r r r. r+ r r r �test_update_from_round_tripr s* z#CTETest.test_update_from_round_tripc C s� | j j}| j j}tj�� ��}|�|�� �dddgt |g��� t |g�� |j j � dddg���d�}|�|�� � |j j |j j k�� t|�t |g��|j j���� dd g� W d Q R X d S ) Nr r r r r r r& )r r N)r r r )r# r r r r! r' r" r$ r6 r r( r) r r* r+ �deleter r3 r r, )r- r r r. r+ r r r �test_delete_from_round_trip� s z#CTETest.test_delete_from_round_tripc C s� | j j}| j j}tj�� ��}|�|�� �dddgt |g��� t |g�� |j j � dddg���d�}|�|�� � |j j t |j j g�� |j j|j jk�k�� t|�t |g��|j j���� dd g� W d Q R X d S ) Nr r r r r r r& )r r N)r r r )r# r r r r! r' r" r$ r6 r r( r) r r* r+ r; r r r3 r, )r- r r r. r+ r r r �"test_delete_scalar_subq_round_trip� s$ z*CTETest.test_delete_scalar_subq_round_tripN)�__name__� __module__� __qualname__Z __backend__� __requires__Z run_insertsZ run_deletes� classmethodr r% r0 r5 r7 r �requiresZctes_with_update_deleteZ update_fromr: Z delete_fromr< r= r r r r r s # r N)� r r Z assertionsr Zschemar r r r r r r Z TablesTestr r r r r �<module> s