summaryrefslogtreecommitdiff
path: root/test/sql/test_insert.py
diff options
context:
space:
mode:
authorMike Bayer <mike_mp@zzzcomputing.com>2021-04-26 13:30:29 -0400
committerMike Bayer <mike_mp@zzzcomputing.com>2021-04-26 14:59:15 -0400
commit87b551c26014c2bf2406ebebb47ee9d3769da9bb (patch)
tree5cd1d3384022e6c9e99c1255806a091ec7da2c2a /test/sql/test_insert.py
parent1443945e61f1f113e46a5044315a91558d4d232a (diff)
downloadsqlalchemy-87b551c26014c2bf2406ebebb47ee9d3769da9bb.tar.gz
Handle Sequence in _process_multiparam_default_bind
Fixed issue where usage of an explicit :class:`.Sequence` would produce inconsistent "inline" behavior for an :class:`.Insert` construct that includes multiple values phrases; the first seq would be inline but subsequent ones would be "pre-execute", leading to inconsistent sequence ordering. The sequence expressions are now fully inline. Fixes: #6361 Change-Id: Ie16794ec0e19979a7e6c8d1bef5716a9fc199889
Diffstat (limited to 'test/sql/test_insert.py')
-rw-r--r--test/sql/test_insert.py69
1 files changed, 69 insertions, 0 deletions
diff --git a/test/sql/test_insert.py b/test/sql/test_insert.py
index 95a8d02a2..6c2a5d955 100644
--- a/test/sql/test_insert.py
+++ b/test/sql/test_insert.py
@@ -384,6 +384,75 @@ class InsertTest(_InsertTestBase, fixtures.TablesTest, AssertsCompiledSQL):
dialect=postgresql.dialect(),
)
+ def test_insert_seq_pk_multi_values(self):
+ """test #6361"""
+
+ m = MetaData()
+
+ t1 = Table(
+ "t",
+ m,
+ Column("id", Integer, Sequence("id_seq"), primary_key=True),
+ Column("data", String),
+ )
+
+ stmt = t1.insert().values(
+ [{"data": "d1"}, {"data": "d2"}, {"data": "d3"}]
+ )
+
+ self.assert_compile(
+ stmt,
+ "INSERT INTO t (id, data) VALUES (nextval('id_seq'), "
+ "%(data_m0)s), (nextval('id_seq'), %(data_m1)s), "
+ "(nextval('id_seq'), %(data_m2)s)",
+ dialect=postgresql.dialect(),
+ )
+
+ def test_insert_seq_non_pk_multi_values(self):
+ """test #6361"""
+
+ m = MetaData()
+
+ t1 = Table(
+ "t",
+ m,
+ Column("id", Integer, primary_key=True),
+ Column("counter", Sequence("counter_seq")),
+ Column("data", String),
+ )
+
+ stmt = t1.insert().values(
+ [{"data": "d1"}, {"data": "d2"}, {"data": "d3"}]
+ )
+
+ self.assert_compile(
+ stmt,
+ "INSERT INTO t (counter, data) VALUES (nextval('counter_seq'), "
+ "%(data_m0)s), (nextval('counter_seq'), %(data_m1)s), "
+ "(nextval('counter_seq'), %(data_m2)s)",
+ dialect=postgresql.dialect(),
+ )
+
+ def test_insert_seq_pk_multi_values_seq_not_supported(self):
+ m = MetaData()
+
+ t1 = Table(
+ "t",
+ m,
+ Column("id", Integer, Sequence("id_seq"), primary_key=True),
+ Column("data", String),
+ )
+
+ stmt = t1.insert().values(
+ [{"data": "d1"}, {"data": "d2"}, {"data": "d3"}]
+ )
+
+ self.assert_compile(
+ stmt,
+ "INSERT INTO t (data) VALUES (?), (?), (?)",
+ dialect=sqlite.dialect(),
+ )
+
def test_insert_from_select_cte_one(self):
table1 = self.tables.mytable