def test_rows_for_insert_or_update(self):
from google.cloud.spanner_dbapi.parse_utils import (
rows_for_insert_or_update, )
from google.cloud.spanner_dbapi.exceptions import Error
with self.assertRaises(Error):
rows_for_insert_or_update([0], [[]])
with self.assertRaises(Error):
rows_for_insert_or_update([0], None, ["0", "%s"])
cases = [
(
["id", "app", "name"],
[(5, "ap", "n"), (6, "bp", "m")],
None,
[(5, "ap", "n"), (6, "bp", "m")],
),
(
["app", "name"],
[("ap", "n"), ("bp", "m")],
None,
[("ap", "n"), ("bp", "m")],
),
(
["app", "name", "fn"],
["ap", "n", "f1", "bp", "m", "f2", "cp", "o", "f3"],
["(%s, %s, %s)", "(%s, %s, %s)", "(%s, %s, %s)"],
[("ap", "n", "f1"), ("bp", "m", "f2"), ("cp", "o", "f3")],
),
(
["app", "name", "fn", "ln"],
[
("ap", "n", (45, "nested"), "ll"),
("bp", "m", "f2", "mt"),
("fp", "cp", "o", "f3"),
],
None,
[
("ap", "n", (45, "nested"), "ll"),
("bp", "m", "f2", "mt"),
("fp", "cp", "o", "f3"),
],
),
(
["app", "name", "fn"],
["ap", "n", "f1"],
None,
[("ap", "n", "f1")],
),
]
for i, (columns, params, pyformat_args, want) in enumerate(cases):
with self.subTest(i=i):
got = rows_for_insert_or_update(columns, params, pyformat_args)
self.assertEqual(got, want)
def test_rows_for_insert_or_update(self):
cases = [
(
["id", "app", "name"],
[(5, "ap", "n"), (6, "bp", "m")],
None,
[(5, "ap", "n"), (6, "bp", "m")],
),
(
["app", "name"],
[("ap", "n"), ("bp", "m")],
None,
[("ap", "n"), ("bp", "m")],
),
(
["app", "name", "fn"],
["ap", "n", "f1", "bp", "m", "f2", "cp", "o", "f3"],
["(%s, %s, %s)", "(%s, %s, %s)", "(%s, %s, %s)"],
[("ap", "n", "f1"), ("bp", "m", "f2"), ("cp", "o", "f3")],
),
(
["app", "name", "fn", "ln"],
[
("ap", "n", (45, "nested"), "ll"),
("bp", "m", "f2", "mt"),
("fp", "cp", "o", "f3"),
],
None,
[
("ap", "n", (45, "nested"), "ll"),
("bp", "m", "f2", "mt"),
("fp", "cp", "o", "f3"),
],
),
(
["app", "name", "fn"],
["ap", "n", "f1"],
None,
[("ap", "n", "f1")],
),
]
for i, (columns, params, pyformat_args, want) in enumerate(cases):
with self.subTest(i=i):
got = rows_for_insert_or_update(columns, params, pyformat_args)
self.assertEqual(got, want)