U
|
¸ý°d' ã @ 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 dejZG dd dejejZG dd dejZG dd dejZdS )é )Ú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 Zejddddejjdd Zejjdd ZdS )Ú SequenceTest©Ú sequencesTZeachc C sÞ t d|tdttttdddtdtd t d|tdttttdtdd ddtdtd t d
|
|tdttttd ddtdtdd d tjj j
|
rÚt d|tdttttdtj dddtdtdd 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) ú]d:\z\workplace\vscode\pyvenv\venv\Lib\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_roundtripW 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_lastrowid[ 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_directc 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_seqg 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 n 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_returningr s
|
|
ÿz8SequenceTest.test_insert_roundtrip_no_implicit_returning)T)Fr )Úargnamesc C sr t dt tdttttdddddtdtd |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_translatex 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 tdttttdtdt}| ¡ 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 ZstmtZ seq_nextvalr) r) r* Ú!test_literal_binds_inline_compile£ s. ÿú ÿ þüz6SequenceCompilerTest.test_literal_binds_inline_compileN)rS rT rU rV rW r` 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 )Nrb 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 )Nrb TZnew_seqrc F) r r rh 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 )Nrf Frg r4 r) r) r* Ú test_has_sequence_other_objectï s z.HasSequenceTest.test_has_sequence_other_objectc C s t t|jdtjdd d S )Nrb r T©r r rh 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_sequenceFrg r4 r) r) r* Útest_has_sequence_negû s z%HasSequenceTest.test_has_sequence_negc C s t t|jdtjdd d S )Nrq r Fro r4 r) r) r* Ú test_has_sequence_schemas_negþ s ÿüz-HasSequenceTest.test_has_sequence_schemas_negc C s t t|jdtjdd d S )NZother_sequencer Fro 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 )Nre Frg 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 )Nrd rb re )Ú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|jtjdg d S )Nr )r r rw 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|jtjdddg d S )Nr re rb )r Úsortedr rw 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+ ri rm rn r r" r# rp rr rs rt ru r{ r| r~ r) r) r) r* ra º s,
|
|
|
|
|
|
ra c @ s e Zd ZdZdZdd ZdS )ÚHasSequenceTestEmptyr Tc C s t t| ¡ g d S r? )r r rw r4 r) r) r* Ú#test_get_sequence_names_no_sequence4 s
|
þz8HasSequenceTestEmpty.test_get_sequence_names_no_sequenceN)rS rT rU rV rW r r) r) r) r* r 0 s r N)Ú r r Z
|
assertionsr r r Z provisionr r r r r r r r r r Z
|
TablesTestr ZAssertsCompiledSQLZTestBaserZ ra r r) r) r) r* Ú<module> s$ v
|
