Edit File: test_insert.cpython-37.pyc
B ��4]r% � @ s� d dl mZ d dl mZ d dl mZ d dlmZ d dlmZ d dlmZ d dlm Z dd l m Z dd l m Z dd l m Z dd l m Z dd l mZ G dd� dej�ZG dd� dej�ZG dd� dej�ZdZdS )� )�config)�engines)�fixtures)�eq_)� requirements)�Column)�Table� )�Integer)�literal)�literal_column)�select)�Stringc @ sR e Zd ZdZdZdZddiZedd� �Zdd � Z d d � Z d d � Z e j dd� �ZdS )� LastrowidTest�eachT)Zimplements_get_lastrowid�autoincrement_insert�implicit_returningFc C sL t d|tdtddd�tdtd��� t d|tdtddd �tdtd��� d S ) N� autoinc_pk�idT)� primary_key�test_needs_autoincrement�data�2 � manual_pkF)r � autoincrement)r r r r )�cls�metadata� r �W/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/suite/test_insert.py� define_tables s zLastrowidTest.define_tablesc C s* |� |�� ��� }t|tjjjdf� d S )Nz some data)�executer �firstr r �db�dialect�default_sequence_base)�self�table�conn�rowr r r �_assert_round_trip* s z LastrowidTest._assert_round_tripc C s. t jj| jj�� dd� | �| jjt j� d S )Nz some data)r )r r" r �tablesr �insertr) )r% r r r � test_autoincrement_on_insert. s z*LastrowidTest.test_autoincrement_on_insertc C sD t jj| jj�� dd�}t j�t| jjjj g��}t |j |g� d S )Nz some data)r ) r r" r r* r r+ �scalarr �cr r �inserted_primary_key)r% �r�pkr r r �test_last_inserted_id3 s z#LastrowidTest.test_last_inserted_idc C sF t jj| jj�� dd�}|j}t j�t| jjj j g��}t ||� d S )Nz some data)r ) r r" r r* r r+ � lastrowidr- r r. r r )r% r0 r3 r1 r r r � test_native_lastrowid_autoinc? s z+LastrowidTest.test_native_lastrowid_autoincN)�__name__� __module__� __qualname__� run_deletes� __backend__� __requires__�__engine_options__� classmethodr r) r, r2 r Zdbapi_lastrowidr4 r r r r r s r c @ s| e Zd ZdZdZedd� �Zdd� Zej dd� �Z ej d d � �Z ej d d � �Zej d d� �Zej dd� �Zej dd� �ZdS )�InsertBehaviorTestr Tc C s� t d|tdtddd�tdtd��� t d|tdtddd �tdtd��� t d |tdtddd�tdtd��td td d �tdttdtd�td� d �� d S )Nr r T)r r r r r F)r r �includes_defaults�x� )�default�y�2)�type_r )r r r r r r )r r r r r r M s, z InsertBehaviorTest.define_tablesc C sb t jjrtjddid�}ntj}|j| jj � � dd�}|j s@t �|j rJt �|jsTt �|jr^t �d S )Nr F)�optionsz some data)r )r � returning�enabledr Ztesting_enginer r" r r* r r+ � _soft_closed�AssertionError�closed� is_insert� returns_rows)r% �enginer0 r r r �test_autoclose_on_insertl s z+InsertBehaviorTest.test_autoclose_on_insertc C sD t jj| jj�� dd�}|js"t�|jr,t�|j s6t�|j r@t�d S )Nz some data)r ) r r" r r* r r+ rH rI rJ rK rL )r% r0 r r r �+test_autoclose_on_insert_implicit_returningz s z>InsertBehaviorTest.test_autoclose_on_insert_implicit_returningc C sb t j�| jj�� �}|js t�|jr(t�t j�| jj� � � | jjj j d k��}t |�� �s^t�d S )N)r r" r r* r r+ rH rI rJ r �wherer. r �len�fetchall)r% r0 r r r �test_empty_insert� s z$InsertBehaviorTest.test_empty_insertc C s� | j j}| j j}tj�|�� tddd�tddd�tddd�g� tj�|�� �dt |j j g�� |j j � ddg����}t|jd g� tj�t |j j g��|j j ��}t|�� d d g� d S ) N� �data1)r r r �data2r �data3)r )rV )rW )r* r r r r" r r+ �dict� from_selectr r. r rP �in_r r/ �order_byrR )r% � src_table� dest_table�resultr r r � test_insert_from_select_autoinc� s z2InsertBehaviorTest.test_insert_from_select_autoincc C s� | j j}| j j}tj�|�� �dt|j j g�� |j j � ddg����}t |jd g� tj�t|j j g��|j j ��}t |�� g � d S )N)r rV rW )r* r r r r" r r+ rY r r. r rP rZ r r/ r[ rR )r% r\ r] r^ r r r �'test_insert_from_select_autoinc_no_rows� s z:InsertBehaviorTest.test_insert_from_select_autoinc_no_rowsc C s� | j j}tj�|�� tddd�tddd�tddd�g� tj�|jdd ��dt|j j d |j j g�� |j j � ddg���� ttj�t|j j g��|j j ���� d d d d d g� d S )NrT rU )r r r rV r rW T)�inliner@ )rU )rV )rW )r* r r r" r r+ rX rY r r. r r rP rZ r r[ rR )r% r& r r r �test_insert_from_select� s z*InsertBehaviorTest.test_insert_from_selectc C s� | j j}tj�|�� tddd�tddd�tddd�g� tj�|jdd ��dt|j j d |j j g�� |j j � ddg���� ttj�t|g��|j j |j j ���� d d d ddg� d S )NrT rU )r r r rV r rW T)ra r@ )rT rU r@ � )r rV r@ rc )� rV r@ rc )r rW r@ rc )� rW r@ rc )r* r> r r" r r+ rX rY r r. r r rP rZ r r[ rR )r% r& r r r �%test_insert_from_select_with_defaults� s& z8InsertBehaviorTest.test_insert_from_select_with_defaultsN)r5 r6 r7 r8 r9 r< r rN r rF rO Z empty_insertsrS Zinsert_from_selectr_ r` rb rf r r r r r= I s r= c @ sZ e Zd ZdZdZdZddiZdd� Zedd� �Z e j d d � �Z d d � Z d d� Zdd� ZdS )� ReturningTestr )rF r Tr c C s* |� |�� ��� }t|tjjjdf� d S )Nz some data)r r r! r r r" r# r$ )r% r& r' r( r r r r) s z ReturningTest._assert_round_tripc C s( t d|tdtddd�tdtd��� d S )Nr r T)r r r r )r r r r )r r r r r r s zReturningTest.define_tablesc C sZ t j}| jj}|j|�� �|jj�dd�}|� � d }t j� t |jjg��}t ||� d S )Nz some data)r � ) r r" r* r r r+ rF r. r r! r- r r )r% rM r& r0 r1 � fetched_pkr r r �%test_explicit_returning_pk_autocommit s z3ReturningTest.test_explicit_returning_pk_autocommitc C sn t j}| jj}|�� �.}|j|�� �|jj �dd�}|� � d }W d Q R X t j� t |jj g��}t ||� d S )Nz some data)r rh )r r" r* r �beginr r+ rF r. r r! r- r r )r% rM r& r' r0 r1 ri r r r �(test_explicit_returning_pk_no_autocommit$ s z6ReturningTest.test_explicit_returning_pk_no_autocommitc C s. t jj| jj�� dd� | �| jjt j� d S )Nz some data)r )r r" r r* r r+ r) )r% r r r �/test_autoincrement_on_insert_implicit_returning/ s z=ReturningTest.test_autoincrement_on_insert_implicit_returningc C sD t jj| jj�� dd�}t j�t| jjjj g��}t |j |g� d S )Nz some data)r ) r r" r r* r r+ r- r r. r r r/ )r% r0 r1 r r r �(test_last_inserted_id_implicit_returning4 s z6ReturningTest.test_last_inserted_id_implicit_returningN)r5 r6 r7 Zrun_create_tablesr: r9 r; r) r< r r Zfetch_rows_post_commitrj rl rm rn r r r r rg s rg )r r= rg N)� r r r Z assertionsr r Zschemar r r r r r r Z TablesTestr r= rg �__all__r r r r �<module> s : ;: