관리-도구
편집 파일: test_insert.cpython-38.pyc
U -?�f�I � @ s* d dl mZ d dlZddlmZ ddlmZ ddlmZ ddlm Z dd l m Z dd l m Z d d lm Z d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ G dd� dej�ZG dd� dej�Z G d d � d ej�Z d Z dS ) � )�DecimalN� )�testing� )�fixtures)�eq_)� requirements)�Column)�Table� )�Double)�Float)�Identity)�Integer)�literal)�literal_column)�Numeric)�select)�String)� LargeBinary)�UUID)�Uuidc @ sJ e Zd ZdZdZdZedd� �Zdd� Zdd � Z d d � Z e j d d � �Z dS )� LastrowidTest�eachT)Zimplements_get_lastrowid�autoincrement_insertc C sT t d|tdtddd�tdtd��dd� t d |tdtddd �tdtd��dd� d S ) N� autoinc_pk�idT�� primary_keyZtest_needs_autoincrement�data�2 F�Zimplicit_returning� manual_pk�r Z autoincrement�r r r r ��cls�metadata� r( �T/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/suite/test_insert.py� define_tables' s& � � �zLastrowidTest.define_tablesc C s( |� |�� ��� }t||jjdf� d S �N� some data��executer �firstr �dialectZdefault_sequence_base��self�table�conn�rowr( r( r) �_assert_round_trip; s ��z LastrowidTest._assert_round_tripc C s. |� | jj�� tdd�� | �| jj|� d S �Nr, �r �r. �tablesr �insert�dictr6 �r2 � connectionr( r( r) � test_autoincrement_on_insertE s �z*LastrowidTest.test_autoincrement_on_insertc C sB |� | jj�� tdd��}|�t| jjjj��}t |j |f� d S r7 � r. r: r r; r< �scalarr �cr r �inserted_primary_key�r2 r> �r�pkr( r( r) �test_last_inserted_idK s �z#LastrowidTest.test_last_inserted_idc C sD |� | jj�� tdd��}|j}|�t| jjjj ��}t ||� d S r7 ) r. r: r r; r< � lastrowidrA r rB r r )r2 r> rE rH rF r( r( r) � test_native_lastrowid_autoincR s �z+LastrowidTest.test_native_lastrowid_autoincN)�__name__� __module__� __qualname__� run_deletes� __backend__� __requires__� classmethodr* r6 r? rG r Zdbapi_lastrowidrI r( r( r( r) r s r c @ s� e Zd ZdZdZedd� �Ze�dddg�e�ddd g�d d � ��Z e j d d � �Z e j dd� �Ze jdd� �Ze jdd� �Ze jdd� �Ze jdd� �Ze jdd� �Ze jdd� �Zd S ) �InsertBehaviorTestr Tc C s� t d|tdtddd�tdtd��� t d|tdtddd �tdtd��� t d |tdtddd�tdtd��dd � t d |tdtddd�tdtd��td tdd�tdttdtd�td� d�� d S )Nr r Tr r r r" Fr# �no_implicit_returningr! �includes_defaults�x� )�default�y�2)�type_r )r r r r r r r% r( r( r) r* ` s\ � � � � � � ��z InsertBehaviorTest.define_tables�style�plain�return_defaults� executemanyFc C sh | j j}|�� }|jr |�� }|rFddiddiddiddiddig}nddi}|�||�}|jrdt�dS )z-test another INSERT issue found during #10453r �d1�d2�d3�d4�d5N)r: rR r; r\ r. � returns_rows�AssertionError)r2 r> rZ r] r3 �stmtr rE r( r( r) �(test_no_results_for_non_returning_insert� s � z;InsertBehaviorTest.test_no_results_for_non_returning_insertc C sR |� | jj�� tdd��}|js$t�|jr.t�|js8t�|j rN|� � d ksNt�d S r7 ) r. r: r r; r< � _soft_closedrd �closed� is_insertrc �fetchone�r2 r> rE r( r( r) �test_autoclose_on_insert� s � z+InsertBehaviorTest.test_autoclose_on_insertc C sh |� | jj�� �� tdd��}|js(t�|jr2t�|j s<t�|j sFt�t |� � d � t |� � dg� d S )Nr, r8 r )r. r: r r; r\ r< rg rd rh ri rc r rj �keysrk r( r( r) �+test_autoclose_on_insert_implicit_returning� s � z>InsertBehaviorTest.test_autoclose_on_insert_implicit_returningc C s` |� | jj�� �}|js t�|jr&t�|� | jj�� �| jjj j d k��}t t |� � �d� d S )Nr �r. r: r r; rg rd rh r �whererB r r �len�allrk r( r( r) �test_empty_insert� s ��z$InsertBehaviorTest.test_empty_insertc C sh |� | jj�� i i i g�}|js$t�|jr.t�|� | jj�� �| jjj j d k��}t t |� � �d� d S )Nr ro rk r( r( r) �test_empty_insert_multiple� s ��z-InsertBehaviorTest.test_empty_insert_multiplec C s� | j j}| j j}|�|�� tddd�tddd�tddd�g� |�|�� �dt|jj �� |jj � ddg����}t |j d � |�t|jj ��|jj ��}t |�� d d g� d S ) Nr �data1�r r r �data2r �data3r8 �N�rw �rx )r: r" r r. r; r< � from_selectr rB r rp �in_r rC �order_by�fetchall�r2 r> Z src_tableZ dest_table�resultr( r( r) � test_insert_from_select_autoinc� s, �� ��� �z2InsertBehaviorTest.test_insert_from_select_autoincc C s| | j j}| j j}|�|�� �dt|jj�� |jj� ddg����}t |j d� |�t|jj�� |jj��}t |�� g � d S )Nr8 rw rx ry )r: r" r r. r; r| r rB r rp r} r rC 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}|�|�� tddd�tddd�tddd�g� |�|�� �� �dt|jj d |jj �� |jj � ddg���� t |�t|jj ��|jj ���� d d d d d g� d S ) Nr ru rv r rw r rx rU )ru rz r{ )r: r" r. r; r< �inliner| r rB r r rp r} r r~ r �r2 r> r3 r( r( r) �test_insert_from_select s, �� ��� � �z*InsertBehaviorTest.test_insert_from_selectc C s� | j j}|�|�� tddd�tddd�tddd�g� |�|�� �� �dt|jj d |jj �� |jj � ddg���� t |�t|��|jj |jj ���� d d d d d g� d S )Nr ru rv r rw r rx rU )r ru rU � )r rw rU r� )� rw rU r� )r rx rU r� )� rx rU r� )r: rS r. r; r< r� r| r rB r r rp r} r r~ r r� r( r( r) �%test_insert_from_select_with_defaults: s6 �� ��� ���z8InsertBehaviorTest.test_insert_from_select_with_defaultsN)rJ rK rL rM rN rP r* r � variationrf r r rl �insert_returningrn Z empty_insertsrs Zempty_inserts_executemanyrt Zinsert_from_selectr� r� r� r� r( r( r( r) rQ \ s. ' rQ c @ s� e Zd ZdZdZdZdd� Zedd� �Ze j dd � �Z d d � Z d d � Z dd� Ze jdd� �Zeje� ddfed�ddejjfe� ddfed�ddejjfedddd�ddejjfedddd�ed�dfdd �e�d ddg�e�d ddg�d d � ���Zejd!e dd"�e � � fd#e ddd$�e e � � �fd%e dd"�e � � ejj fd&e ddd$�e e � � �ejj fd'e!� e � � ejj fd(e"� d)fd*e"� d+fd,d-d.� e�d ddg�e�d ddg�ejj#d/d0� ����Z$d1S )2� ReturningTestr )r� r Tc C s( |� |�� ��� }t||jjdf� d S r+ r- r1 r( r( r) r6 d s ��z ReturningTest._assert_round_tripc C s( t d|tdtddd�tdtd��� d S )Nr r Tr r r r$ r% r( r( r) r* n s � �zReturningTest.define_tablesc C sT | j j}|�|�� �|jj�tdd��}|�� d }|� t |jj��}t ||� d S �Nr, r8 r � r: r r. r; � returningrB r r< r/ rA r r �r2 r> r3 rE rF Z fetched_pkr( r( r) �%test_explicit_returning_pk_autocommity s � z3ReturningTest.test_explicit_returning_pk_autocommitc C sT | j j}|�|�� �|jj�tdd��}|�� d }|� t |jj��}t ||� d S r� r� r� r( r( r) �(test_explicit_returning_pk_no_autocommit� s � z6ReturningTest.test_explicit_returning_pk_no_autocommitc C s. |� | jj�� tdd�� | �| jj|� d S r7 r9 r= r( r( r) �/test_autoincrement_on_insert_implicit_returning� s �z=ReturningTest.test_autoincrement_on_insert_implicit_returningc C sB |� | jj�� tdd��}|�t| jjjj��}t |j |f� d S r7 r@ rD r( r( r) �(test_last_inserted_id_implicit_returning� s �z6ReturningTest.test_last_inserted_id_implicit_returningc C sp |� | jj�� �| jjjj�ddiddiddiddiddig�}|�� }|� t| jjjj��}t ||�� � d S )Nr r^ r_ r` ra rb ) r. r: r r; r� rB r rr r r )r2 r> rE ZrallZpksr( r( r) � test_insertmanyvalues_returning� s ��� z-ReturningTest.test_insertmanyvalues_returninggxXM|Z!@�5 g_�Q!@r� � � F)� precision�scaleZ asdecimalz 8.5514716ztype_,value,do_rounding)�argnames�sort_by_parameter_order� multiple_rowsc s t d|tdtt� dd�td|��}|�|� |�|�� j|jj |jj t |�d�|rh� fdd�t d �D �nd� i�} |r�t d d �} n t d d �} |r�t d d� | D �� fdd�| D �� t dd� |�t|jj ��D �t� d�h� n:t t| �� fdd�| D �� t t|�t|jj ���� h� dS )zvtest #9701. this tests insertmanyvalues as well as decimal / floating point RETURNING types Zf_tr T�r �value�r� c s g | ] }d � i�qS �r� r( ��.0�ir� r( r) � <listcomp>� s z6ReturningTest.test_insert_w_floats.<locals>.<listcomp>� r � r c S s h | ]\}}|t |d �f�qS �rU ��round)r� �id_�val_r( r( r) � <setcomp> s z5ReturningTest.test_insert_w_floats.<locals>.<setcomp>c s h | ]}|t � d �f�qS r� r� �r� r� r� r( r) r� s c S s h | ]}t |d ��qS r� r� )r� r� r( r( r) r� s �rU c s h | ] }|� f�qS r( r( r� r� r( r) r� s N)r r r r �creater. r; r� rB r r� �bool�ranger �scalarsr r� �set) r2 r> r' r� rY r� Z do_roundingr� �tr� �i_ranger( r� r) �test_insert_w_floats� sN /� ��� �� ���z"ReturningTest.test_insert_w_floatsZnon_native_uuid)� native_uuidZnon_native_uuid_str)Zas_uuidr� Zgeneric_native_uuidZgeneric_native_uuid_strr Z LargeBinary1s this is binaryZ LargeBinary2s 7�z type_,valueZiaa)r� r� c s� t d|tdtt� dd�td|��}|�|� |�|�� j|jj |jj t |�d�|rh� fdd�t d �D �nd� i�}|r�t d d �} n t d d �} t t|�� fd d�| D �� t t|�t|jj ���� h� dS )a test #9739, #9808 (similar to #9701). this tests insertmanyvalues in conjunction with various datatypes. These tests are particularly for the asyncpg driver which needs most types to be explicitly cast for the new IMV format Zd_tr Tr� r� r� c s g | ] }d � i�qS r� r( r� r� r( r) r� ` s z>ReturningTest.test_imv_returning_datatypes.<locals>.<listcomp>r� r r� r c s h | ] }|� f�qS r( r( r� r� r( r) r� m s z=ReturningTest.test_imv_returning_datatypes.<locals>.<setcomp>N)r r r r r� r. r; r� rB r r� r� r� r r� r� r ) r2 r> r' r� rY r� r� r� r� r� r( r� r) � test_imv_returning_datatypes s8 5� ��� ��z*ReturningTest.test_imv_returning_datatypesN)%rJ rK rL Zrun_create_tablesrO rN r6 rP r* r Zfetch_rows_post_commitr� r� r� r� Z insert_executemany_returningr� r � combinationsr �requiresZ-float_or_double_precision_behaves_genericallyr r Zliteral_float_coercionr r� r� r �uuid�uuid4�strZuuid_data_typer r r� r� r( r( r( r) r� _ s� � � � �� O� �� �� �!r� )r rQ r� ) �decimalr r� � r r Z assertionsr �configr Zschemar r r r r r r r r r r �typesr r r Z TablesTestr rQ r� �__all__r( r( r( r) �<module> s6 <