summaryrefslogtreecommitdiff
path: root/test/dialect/postgresql/test_compiler.py
diff options
context:
space:
mode:
authorGord Thompson <gord@gordthompson.com>2021-07-26 12:52:23 -0600
committerGord Thompson <gord@gordthompson.com>2021-07-26 14:00:53 -0600
commit2c4a497a0c7f3981be38340089cbb5f32e376b03 (patch)
tree84ff588f3f48afc4db2b4a401293c4332f7898de /test/dialect/postgresql/test_compiler.py
parentc6b1d24fe71c22e4d2117d084f06df3597671985 (diff)
downloadsqlalchemy-2c4a497a0c7f3981be38340089cbb5f32e376b03.tar.gz
Modernize tests - dml_values
Change-Id: Ida86ed40c43d91813151621b847376976773a5f9
Diffstat (limited to 'test/dialect/postgresql/test_compiler.py')
-rw-r--r--test/dialect/postgresql/test_compiler.py83
1 files changed, 47 insertions, 36 deletions
diff --git a/test/dialect/postgresql/test_compiler.py b/test/dialect/postgresql/test_compiler.py
index cd360c2d4..d2c81e63f 100644
--- a/test/dialect/postgresql/test_compiler.py
+++ b/test/dialect/postgresql/test_compiler.py
@@ -129,8 +129,10 @@ class CompileTest(fixtures.TestBase, AssertsCompiledSQL):
column("name", String(128)),
column("description", String(128)),
)
- u = update(table1, values=dict(name="foo")).returning(
- table1.c.myid, table1.c.name
+ u = (
+ update(table1)
+ .values(dict(name="foo"))
+ .returning(table1.c.myid, table1.c.name)
)
self.assert_compile(
u,
@@ -138,7 +140,7 @@ class CompileTest(fixtures.TestBase, AssertsCompiledSQL):
"RETURNING mytable.myid, mytable.name",
dialect=dialect,
)
- u = update(table1, values=dict(name="foo")).returning(table1)
+ u = update(table1).values(dict(name="foo")).returning(table1)
self.assert_compile(
u,
"UPDATE mytable SET name=%(name)s "
@@ -146,8 +148,10 @@ class CompileTest(fixtures.TestBase, AssertsCompiledSQL):
"mytable.description",
dialect=dialect,
)
- u = update(table1, values=dict(name="foo")).returning(
- func.length(table1.c.name)
+ u = (
+ update(table1)
+ .values(dict(name="foo"))
+ .returning(func.length(table1.c.name))
)
self.assert_compile(
u,
@@ -165,8 +169,10 @@ class CompileTest(fixtures.TestBase, AssertsCompiledSQL):
column("description", String(128)),
)
- i = insert(table1, values=dict(name="foo")).returning(
- table1.c.myid, table1.c.name
+ i = (
+ insert(table1)
+ .values(dict(name="foo"))
+ .returning(table1.c.myid, table1.c.name)
)
self.assert_compile(
i,
@@ -175,7 +181,7 @@ class CompileTest(fixtures.TestBase, AssertsCompiledSQL):
"mytable.name",
dialect=dialect,
)
- i = insert(table1, values=dict(name="foo")).returning(table1)
+ i = insert(table1).values(dict(name="foo")).returning(table1)
self.assert_compile(
i,
"INSERT INTO mytable (name) VALUES "
@@ -183,8 +189,10 @@ class CompileTest(fixtures.TestBase, AssertsCompiledSQL):
"mytable.name, mytable.description",
dialect=dialect,
)
- i = insert(table1, values=dict(name="foo")).returning(
- func.length(table1.c.name)
+ i = (
+ insert(table1)
+ .values(dict(name="foo"))
+ .returning(func.length(table1.c.name))
)
self.assert_compile(
i,
@@ -1742,7 +1750,7 @@ class CompileTest(fixtures.TestBase, AssertsCompiledSQL):
)
self.assert_compile(stmt, expected)
- stmt = update(tbl1, values=dict(id=1))
+ stmt = update(tbl1).values(dict(id=1))
stmt = stmt.with_hint("ONLY", dialect_name="postgresql")
expected = "UPDATE ONLY testtbl1 SET id=%(id)s"
self.assert_compile(stmt, expected)
@@ -2240,9 +2248,11 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
def test_do_nothing_no_target(self):
- i = insert(
- self.table1, values=dict(name="foo")
- ).on_conflict_do_nothing()
+ i = (
+ insert(self.table1)
+ .values(dict(name="foo"))
+ .on_conflict_do_nothing()
+ )
self.assert_compile(
i,
"INSERT INTO mytable (name) VALUES "
@@ -2251,9 +2261,11 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
def test_do_nothing_index_elements_target(self):
- i = insert(
- self.table1, values=dict(name="foo")
- ).on_conflict_do_nothing(index_elements=["myid"])
+ i = (
+ insert(self.table1)
+ .values(dict(name="foo"))
+ .on_conflict_do_nothing(index_elements=["myid"])
+ )
self.assert_compile(
i,
"INSERT INTO mytable (name) VALUES "
@@ -2342,7 +2354,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
)
def test_do_update_str_index_elements_target_two(self):
- i = insert(self.table1, values=dict(name="foo"))
+ i = insert(self.table1).values(dict(name="foo"))
i = i.on_conflict_do_update(
index_elements=["myid"], set_=dict(name=i.excluded.name)
)
@@ -2354,7 +2366,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
)
def test_do_update_col_index_elements_target(self):
- i = insert(self.table1, values=dict(name="foo"))
+ i = insert(self.table1).values(dict(name="foo"))
i = i.on_conflict_do_update(
index_elements=[self.table1.c.myid],
set_=dict(name=i.excluded.name),
@@ -2367,7 +2379,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
)
def test_do_update_unnamed_pk_constraint_target(self):
- i = insert(self.table_with_metadata, values=dict(myid=1, name="foo"))
+ i = insert(self.table_with_metadata).values(dict(myid=1, name="foo"))
i = i.on_conflict_do_update(
constraint=self.table_with_metadata.primary_key,
set_=dict(name=i.excluded.name),
@@ -2380,7 +2392,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
)
def test_do_update_pk_constraint_index_elements_target(self):
- i = insert(self.table_with_metadata, values=dict(myid=1, name="foo"))
+ i = insert(self.table_with_metadata).values(dict(myid=1, name="foo"))
i = i.on_conflict_do_update(
index_elements=self.table_with_metadata.primary_key,
set_=dict(name=i.excluded.name),
@@ -2393,7 +2405,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
)
def test_do_update_named_unique_constraint_target(self):
- i = insert(self.table1, values=dict(name="foo"))
+ i = insert(self.table1).values(dict(name="foo"))
i = i.on_conflict_do_update(
constraint=self.unique_constr, set_=dict(myid=i.excluded.myid)
)
@@ -2405,7 +2417,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
)
def test_do_update_string_constraint_target(self):
- i = insert(self.table1, values=dict(name="foo"))
+ i = insert(self.table1).values(dict(name="foo"))
i = i.on_conflict_do_update(
constraint=self.unique_constr.name, set_=dict(myid=i.excluded.myid)
)
@@ -2418,7 +2430,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
def test_do_nothing_quoted_string_constraint_target(self):
"""test #6696"""
- i = insert(self.table1, values=dict(name="foo"))
+ i = insert(self.table1).values(dict(name="foo"))
i = i.on_conflict_do_nothing(constraint="Some Constraint Name")
self.assert_compile(
i,
@@ -2442,7 +2454,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
uq,
)
- i = insert(self.table1, values=dict(name="foo"))
+ i = insert(self.table1).values(dict(name="foo"))
i = i.on_conflict_do_nothing(constraint=uq)
self.assert_compile(
@@ -2455,7 +2467,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
def test_do_nothing_quoted_named_constraint_target(self):
"""test #6696"""
- i = insert(self.table1, values=dict(name="foo"))
+ i = insert(self.table1).values(dict(name="foo"))
unique_constr = UniqueConstraint(
self.table1.c.myid, name="Some Constraint Name"
)
@@ -2468,7 +2480,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
)
def test_do_update_index_elements_where_target(self):
- i = insert(self.table1, values=dict(name="foo"))
+ i = insert(self.table1).values(dict(name="foo"))
i = i.on_conflict_do_update(
index_elements=self.goofy_index.expressions,
index_where=self.goofy_index.dialect_options["postgresql"][
@@ -2485,9 +2497,8 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
)
def test_do_update_index_elements_where_target_multivalues(self):
- i = insert(
- self.table1,
- values=[dict(name="foo"), dict(name="bar"), dict(name="bat")],
+ i = insert(self.table1).values(
+ [dict(name="foo"), dict(name="bar"), dict(name="bat")],
)
i = i.on_conflict_do_update(
index_elements=self.goofy_index.expressions,
@@ -2512,7 +2523,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
)
def test_do_update_unnamed_index_target(self):
- i = insert(self.table1, values=dict(name="foo"))
+ i = insert(self.table1).values(dict(name="foo"))
unnamed_goofy = Index(
None, self.table1.c.name, postgresql_where=self.table1.c.name > "m"
@@ -2530,7 +2541,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
)
def test_do_update_unnamed_exclude_constraint_target(self):
- i = insert(self.table1, values=dict(name="foo"))
+ 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)
)
@@ -2543,7 +2554,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
)
def test_do_update_add_whereclause(self):
- i = insert(self.table1, values=dict(name="foo"))
+ 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),
@@ -2563,7 +2574,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
)
def test_do_update_add_whereclause_references_excluded(self):
- i = insert(self.table1, values=dict(name="foo"))
+ 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),
@@ -2579,7 +2590,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
)
def test_do_update_additional_colnames(self):
- i = insert(self.table1, values=dict(name="bar"))
+ i = insert(self.table1).values(dict(name="bar"))
i = i.on_conflict_do_update(
constraint=self.excl_constr_anon,
set_=dict(name="somename", unknown="unknown"),
@@ -2604,7 +2615,7 @@ class InsertOnConflictTest(fixtures.TestBase, AssertsCompiledSQL):
)
def test_on_conflict_as_cte(self):
- i = insert(self.table1, values=dict(name="foo"))
+ i = insert(self.table1).values(dict(name="foo"))
i = (
i.on_conflict_do_update(
constraint=self.excl_constr_anon,