관리-도구
편집 파일: test_sequence.cpython-38.pyc
U -?�f�& � @ s� d dl mZ d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dl m Z d dl m Z d d l m Z d d l m Z d d l mZ d d l mZ d dl mZ d dl mZ G dd� dej�ZG dd� dejej�ZG dd� dej�ZG dd� dej�ZdS )� )�config)�fixtures)�eq_)�is_true)� requirements)�normalize_sequence)�Column)�Table� )�inspect)�Integer)�MetaData)�Sequence)�String)�testingc @ s� e Zd ZdZdZdZedd� �Zdd� Zdd � Z d d � Z e j d d � �Z dd� Zdd� Zejdddd�ejjdd� ��Zejjdd� �ZdS )� SequenceTest�� sequencesTZeachc C s� t d|tdttttd��dd�tdtd��� t d|tdttttdtdd ��dd�tdtd��� t d |tdttttd ��dd�tdtd��d d � tjj j r�t d|tdttttdtj d��dd�tdtd��d tj d� d S )N�seq_pk�idZ tab_id_seqT�Z primary_key�data�2 � seq_opt_pk)Z data_type�optional�seq_no_returningZ noret_id_seqF)�implicit_returning�seq_no_returning_sch�noret_sch_id_seq��schema�r r ) r r r r r r r r �requires�schemas�enabled� test_schema��cls�metadata� r) �V/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/suite/test_sequence.py� define_tables sr � � �� � � � ��� �zSequenceTest.define_tablesc C s. |� | jj�� tdd�� | �| jj|� d S �N� some data�r )�execute�tablesr �insert�dict�_assert_round_trip��self� connectionr) r) r* �test_insert_roundtrip] s z"SequenceTest.test_insert_roundtripc C s2 |� | jj�� tdd��}t|jtjj j f� d S r, ) r/ r0 r r1 r2 r �inserted_primary_keyr �db�dialect�default_sequence_base�r5 r6 �rr) r) r* �test_insert_lastrowida s � �z"SequenceTest.test_insert_lastrowidc C s( |� | jjjjj�}t|tjj j � d S �N) �scalarr0 r �cr �defaultr r r9 r: r; r< r) r) r* �test_nextval_directi s z SequenceTest.test_nextval_directc C s* |� | jj�� tdd��}t|jd� d S )Nr- r. )� )r/ r0 r r1 r2 r r8 r< r) r) r* �test_optional_seqm s �z SequenceTest.test_optional_seqc C s* |� |�� ��� }t|tjjjdf� d S )Nr- )r/ �select�firstr r r9 r: r; )r5 �table�conn�rowr) r) r* r3 t s z SequenceTest._assert_round_tripc C s. |� | jj�� tdd�� | �| jj|� d S r, )r/ r0 r r1 r2 r3 r4 r) r) r* �+test_insert_roundtrip_no_implicit_returningx s �z8SequenceTest.test_insert_roundtrip_no_implicit_returning)T)Fr )�argnamesc C sr t dt� tdttttddd��dd�tdtd ��|dd �}|jdtj id �}|� |� � t d d �� | � ||� d S )Nr r r � alt_schemar Tr r r r! �Zschema_translate_mapr- r. )r r r r r r r r �execution_optionsr% r/ r1 r2 r3 )r5 r6 r r r) r) r* � test_insert_roundtrip_translate~ s* �� ��z,SequenceTest.test_insert_roundtrip_translatec C sB t ttddd��}|jdtjid�}|�|�}t|tjj j � d S )Nr rM r rN ) r r r rO r% r@ r r r9 r: r; )r5 r6 �seqr= r) r) r* �$test_nextval_direct_schema_translate� s �� z1SequenceTest.test_nextval_direct_schema_translateN)�__name__� __module__� __qualname__� __requires__� __backend__Zrun_create_tables� classmethodr+ r7 r> rC r Zsequences_optionalrE r3 rK r � combinationsr" r# rP rR r) r) r) r* r s" = r c @ s e Zd ZdZdZdd� ZdS )�SequenceCompilerTestr Tc C sx t dt� tdttttd���tdt��}|�� jdd�}|j j d |j d�� tttd���}| j |d|f d |j d � d S ) N�x�yZy_seq�q� )r] )Z statementr: z#INSERT INTO x (y, q) VALUES (%s, 5)T)Z literal_bindsr: ) r r r r r r r r1 �valuesr: Zstatement_compilerZvisit_sequenceZassert_compile)r5 r6 rH �stmtZ seq_nextvalr) r) r* �!test_literal_binds_inline_compile� s. �� � ��z6SequenceCompilerTest.test_literal_binds_inline_compileN)rS rT rU rV rW ra r) r) r) r* rZ � s rZ c @ s� e Zd ZdZdZdZedd� �Zdd� Zdd � Z d d � Z e j j d d � �Zdd� Ze j j dd� �Ze j j dd� �Ze j j dd� �Zdd� Ze j j dd� �Ze j j dd� �ZdS ) �HasSequenceTestNr Tc C sx t ttd|d�� t ttd|ddd�� tjjjr^t ttdtj|d�� t ttdtj|d�� td|t d t dd �� d S ) N� user_id_seq�r( � other_seqT)r( Z nomaxvalueZ nominvalue)r r( � schema_seq� user_id_tabler r ) r r r r r" r# r$ r% r r r r& r) r) r* r+ � s@ �� �� �� �z HasSequenceTest.define_tablesc C s t t|��d�d� d S )Nrc T�r r � has_sequencer4 r) r) r* �test_has_sequence� s z!HasSequenceTest.test_has_sequencec C s� t |�}t|�d�d� tttd|d��}t|�d�d� |�|� z,t|�d�d� |�� t|�d�d� W 5 |�|� X d S )Nrc TZnew_seqrd F) r r ri r r r �createZdrop� clear_cache)r5 r6 r( Zinsp�ssr) r) r* �test_has_sequence_cache� s z'HasSequenceTest.test_has_sequence_cachec C s t t|��d�d� d S )Nrg Frh r4 r) r) r* � test_has_sequence_other_object� s z.HasSequenceTest.test_has_sequence_other_objectc C s t t|�jdtjd�d� d S )Nrc r T�r r ri r r% r4 r) r) r* �test_has_sequence_schema� s ��z(HasSequenceTest.test_has_sequence_schemac C s t t|��d�d� d S )N� some_sequenceFrh r4 r) r) r* �test_has_sequence_neg s z%HasSequenceTest.test_has_sequence_negc C s t t|�jdtjd�d� d S )Nrr r Frp r4 r) r) r* � test_has_sequence_schemas_neg s ��z-HasSequenceTest.test_has_sequence_schemas_negc C s t t|�jdtjd�d� d S )NZother_sequencer Frp r4 r) r) r* �'test_has_sequence_default_not_in_remote s ��z7HasSequenceTest.test_has_sequence_default_not_in_remotec C s t t|��d�d� d S )Nrf Frh r4 r) r) r* �'test_has_sequence_remote_not_in_default s z7HasSequenceTest.test_has_sequence_remote_not_in_defaultc C s: ddh}t t|��� �}t|�|�|k� td|k� d S )Nre rc rf )�setr �get_sequence_namesr � intersection)r5 r6 �exp�resr) r) r* �test_get_sequence_names s z'HasSequenceTest.test_get_sequence_namesc C s t t|�jtjd�g � d S )Nr )r r rx r Z test_schema_2r4 r) r) r* �*test_get_sequence_names_no_sequence_schema s ��z:HasSequenceTest.test_get_sequence_names_no_sequence_schemac C s$ t tt|�jtjd��ddg� d S )Nr rf rc )r �sortedr rx r r% r4 r) r) r* �(test_get_sequence_names_sequences_schema) s ���z8HasSequenceTest.test_get_sequence_names_sequences_schema)rS rT rU Z run_deletesrV rW rX r+ rj rn ro r r" r# rq rs rt ru rv r| r} r r) r) r) r* rb � s, rb c @ s e Zd ZdZdZdd� ZdS )�HasSequenceTestEmptyr Tc C s t t|��� g � d S r? )r r rx r4 r) r) r* �#test_get_sequence_names_no_sequence9 s �z8HasSequenceTestEmpty.test_get_sequence_names_no_sequenceN)rS rT rU rV rW r� r) r) r) r* r� 5 s r� N)� r r Z assertionsr r r Z provisionr r r r r r r r r r Z TablesTestr ZAssertsCompiledSQLZTestBaserZ rb r� r) r) r) r* �<module> s$ v