summaryrefslogtreecommitdiff
path: root/test/dialect/postgresql/test_compiler.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/dialect/postgresql/test_compiler.py')
-rw-r--r--test/dialect/postgresql/test_compiler.py24
1 files changed, 24 insertions, 0 deletions
diff --git a/test/dialect/postgresql/test_compiler.py b/test/dialect/postgresql/test_compiler.py
index 0533016cd..18940ed5f 100644
--- a/test/dialect/postgresql/test_compiler.py
+++ b/test/dialect/postgresql/test_compiler.py
@@ -1396,6 +1396,30 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
"param_1": "somename",
"param_2": "unknown"})
+ def test_on_conflict_as_cte(self):
+ i = insert(
+ self.table1, values=dict(name='foo'))
+ i = i.on_conflict_do_update(
+ constraint=self.excl_constr_anon,
+ set_=dict(name=i.excluded.name),
+ where=(
+ (self.table1.c.name != i.excluded.name))
+ ).returning(literal_column("1")).cte("i_upsert")
+
+ stmt = select([i])
+
+ self.assert_compile(
+ stmt,
+ "WITH i_upsert AS "
+ "(INSERT INTO mytable (name) VALUES (%(name)s) "
+ "ON CONFLICT (name, description) "
+ "WHERE description != %(description_1)s "
+ "DO UPDATE SET name = excluded.name "
+ "WHERE mytable.name != excluded.name RETURNING 1) "
+ "SELECT i_upsert.1 "
+ "FROM i_upsert"
+ )
+
def test_quote_raw_string_col(self):
t = table('t', column("FancyName"), column("other name"))