def test_a_args(self):
from google.cloud.spanner_dbapi.parser import ARGS
from google.cloud.spanner_dbapi.parser import a_args
from google.cloud.spanner_dbapi.parser import expect
from google.cloud.spanner_dbapi.parser import func
from google.cloud.spanner_dbapi.parser import pyfmt_str
cases = [
("()", "", a_args([])),
("(%s)", "", a_args([pyfmt_str])),
("(%s,)", "", a_args([pyfmt_str])),
("(%s),", ",", a_args([pyfmt_str])),
(
"(%s,%s, f1(%s, %s))",
"",
a_args([
pyfmt_str, pyfmt_str,
func("f1", a_args([pyfmt_str, pyfmt_str]))
]),
),
]
for text, want_unconsumed, want_parsed in cases:
with self.subTest(text=text):
got_unconsumed, got_parsed = expect(text, ARGS)
self.assertEqual(got_parsed, want_parsed)
self.assertEqual(got_unconsumed, want_unconsumed)
def test_func(self):
cases = [
("_91())", ")", func("_91", a_args([]))),
("_a()", "", func("_a", a_args([]))),
("___()", "", func("___", a_args([]))),
("abc()", "", func("abc", a_args([]))),
(
"AF112(%s, LOWER(%s, %s), rand(%s, %s, TAN(%s, %s)))",
"",
func(
"AF112",
a_args([
pyfmt_str,
func("LOWER", a_args([pyfmt_str, pyfmt_str])),
func(
"rand",
a_args([
pyfmt_str,
pyfmt_str,
func(
"TAN",
a_args([pyfmt_str, pyfmt_str]),
),
]),
),
]),
),
),
]
for text, want_unconsumed, want_parsed in cases:
with self.subTest(text=text):
got_unconsumed, got_parsed = expect(text, FUNC)
self.assertEqual(got_parsed, want_parsed)
self.assertEqual(got_unconsumed, want_unconsumed)
def test_expect_values(self):
cases = [
("VALUES ()", "", values([a_args([])])),
("VALUES", "", values([])),
("VALUES(%s)", "", values([a_args([pyfmt_str])])),
(" VALUES (%s) ", "", values([a_args([pyfmt_str])])),
("VALUES(%s, %s)", "", values([a_args([pyfmt_str, pyfmt_str])])),
(
"VALUES(%s, %s, LOWER(%s, %s))",
"",
values([
a_args([
pyfmt_str,
pyfmt_str,
func("LOWER", a_args([pyfmt_str, pyfmt_str])),
])
]),
),
(
"VALUES (UPPER(%s)), (%s)",
"",
values([
a_args([func("UPPER", a_args([pyfmt_str]))]),
a_args([pyfmt_str]),
]),
),
]
for text, want_unconsumed, want_parsed in cases:
with self.subTest(text=text):
got_unconsumed, got_parsed = expect(text, VALUES)
self.assertEqual(got_parsed, want_parsed)
self.assertEqual(got_unconsumed, want_unconsumed)
def test_terminal(self):
cases = [
("%s", "", pyfmt_str),
(" %s", "", pyfmt_str),
(" %s ", "", pyfmt_str),
]
for text, want_unconsumed, want_parsed in cases:
with self.subTest(text=text):
got_unconsumed, got_parsed = expect(text, TERMINAL)
self.assertEqual(got_parsed, want_parsed)
self.assertEqual(got_unconsumed, want_unconsumed)
def test_a_args_fail(self):
cases = [
("", "ARGS: supposed to begin with `\\(`"),
("(", "ARGS: supposed to end with `\\)`"),
(")", "ARGS: supposed to begin with `\\(`"),
("(%s,%s, f1(%s, %s), %s", "ARGS: supposed to end with `\\)`"),
]
for text, wantException in cases:
with self.subTest(text=text):
self.assertRaisesRegex(ProgrammingError, wantException,
lambda: expect(text, ARGS))
def test_expect(self):
from google.cloud.spanner_dbapi.parser import ARGS
from google.cloud.spanner_dbapi.parser import expect
from google.cloud.spanner_dbapi import exceptions
with self.assertRaises(exceptions.ProgrammingError):
expect(word="", token=ARGS)
with self.assertRaises(exceptions.ProgrammingError):
expect(word="ABC", token=ARGS)
with self.assertRaises(exceptions.ProgrammingError):
expect(word="(", token=ARGS)
with self.assertRaises(exceptions.ProgrammingError):
expect(word="", token="ABC")
def test_func_fail(self):
cases = [
("", "FUNC: `` does not begin with `a-zA-z` nor a `_`"),
("91", "FUNC: `91` does not begin with `a-zA-z` nor a `_`"),
("_91", "supposed to begin with `\\(`"),
("_91(", "supposed to end with `\\)`"),
("_.()", "supposed to begin with `\\(`"),
("_a.b()", "supposed to begin with `\\(`"),
]
for text, wantException in cases:
with self.subTest(text=text):
self.assertRaisesRegex(ProgrammingError, wantException,
lambda: expect(text, FUNC))
def test_terminal_fail(self):
cases = [
("", "TERMINAL: `` is not %s"),
("fdp", "TERMINAL: `fdp` is not %s"),
("%%s", "TERMINAL: `%%s` is not %s"),
("%sa", "TERMINAL: `%sa` is not %s"),
]
for text, wantException in cases:
with self.subTest(text=text):
self.assertRaisesRegex(
ProgrammingError,
wantException,
lambda: expect(text, TERMINAL),
)
def test_a_args_fail(self):
from google.cloud.spanner_dbapi.exceptions import ProgrammingError
from google.cloud.spanner_dbapi.parser import ARGS
from google.cloud.spanner_dbapi.parser import expect
cases = [
("", "ARGS: supposed to begin with `\\(`"),
("(", "ARGS: supposed to end with `\\)`"),
(")", "ARGS: supposed to begin with `\\(`"),
("(%s,%s, f1(%s, %s), %s", "ARGS: supposed to end with `\\)`"),
]
for text, wantException in cases:
with self.subTest(text=text):
self.assertRaisesRegex(ProgrammingError, wantException,
lambda: expect(text, ARGS))
def test_expect_values_fail(self):
cases = [
("", "VALUES: `` does not start with VALUES"),
(
"VALUES(%s, %s, (%s, %s))",
"FUNC: `\\(%s, %s\\)\\)` does not begin with `a-zA-z` nor a `_`",
),
("VALUES(%s),,", "ARGS: supposed to begin with `\\(` in `,`"),
]
for text, wantException in cases:
with self.subTest(text=text):
self.assertRaisesRegex(
ProgrammingError,
wantException,
lambda: expect(text, VALUES),
)
def test_func_fail(self):
from google.cloud.spanner_dbapi.exceptions import ProgrammingError
from google.cloud.spanner_dbapi.parser import FUNC
from google.cloud.spanner_dbapi.parser import expect
cases = [
("", "FUNC: `` does not begin with `a-zA-z` nor a `_`"),
("91", "FUNC: `91` does not begin with `a-zA-z` nor a `_`"),
("_91", "supposed to begin with `\\(`"),
("_91(", "supposed to end with `\\)`"),
("_.()", "supposed to begin with `\\(`"),
("_a.b()", "supposed to begin with `\\(`"),
]
for text, wantException in cases:
with self.subTest(text=text):
self.assertRaisesRegex(ProgrammingError, wantException,
lambda: expect(text, FUNC))
def test_expect_values(self):
from google.cloud.spanner_dbapi.parser import VALUES
from google.cloud.spanner_dbapi.parser import a_args
from google.cloud.spanner_dbapi.parser import expect
from google.cloud.spanner_dbapi.parser import func
from google.cloud.spanner_dbapi.parser import pyfmt_str
from google.cloud.spanner_dbapi.parser import values
cases = [
("VALUES ()", "", values([a_args([])])),
("VALUES", "", values([])),
("VALUES(%s)", "", values([a_args([pyfmt_str])])),
(" VALUES (%s) ", "", values([a_args([pyfmt_str])])),
("VALUES(%s, %s)", "", values([a_args([pyfmt_str, pyfmt_str])])),
(
"VALUES(%s, %s, LOWER(%s, %s))",
"",
values([
a_args([
pyfmt_str,
pyfmt_str,
func("LOWER", a_args([pyfmt_str, pyfmt_str])),
])
]),
),
(
"VALUES (UPPER(%s)), (%s)",
"",
values([
a_args([func("UPPER", a_args([pyfmt_str]))]),
a_args([pyfmt_str])
]),
),
]
for text, want_unconsumed, want_parsed in cases:
with self.subTest(text=text):
got_unconsumed, got_parsed = expect(text, VALUES)
self.assertEqual(got_parsed, want_parsed)
self.assertEqual(got_unconsumed, want_unconsumed)
def test_a_args(self):
cases = [
("()", "", a_args([])),
("(%s)", "", a_args([pyfmt_str])),
("(%s,)", "", a_args([pyfmt_str])),
("(%s),", ",", a_args([pyfmt_str])),
(
"(%s,%s, f1(%s, %s))",
"",
a_args([
pyfmt_str,
pyfmt_str,
func("f1", a_args([pyfmt_str, pyfmt_str])),
]),
),
]
for text, want_unconsumed, want_parsed in cases:
with self.subTest(text=text):
got_unconsumed, got_parsed = expect(text, ARGS)
self.assertEqual(got_parsed, want_parsed)
self.assertEqual(got_unconsumed, want_unconsumed)
def test_expect(self):
from google.cloud.spanner_dbapi.parser import ARGS
from google.cloud.spanner_dbapi.parser import EXPR
from google.cloud.spanner_dbapi.parser import FUNC
from google.cloud.spanner_dbapi.parser import expect
from google.cloud.spanner_dbapi.parser import pyfmt_str
from google.cloud.spanner_dbapi import exceptions
with self.assertRaises(exceptions.ProgrammingError):
expect(word="", token=ARGS)
with self.assertRaises(exceptions.ProgrammingError):
expect(word="ABC", token=ARGS)
with self.assertRaises(exceptions.ProgrammingError):
expect(word="(", token=ARGS)
expected = "", pyfmt_str
self.assertEqual(expect("%s", EXPR), expected)
expected = expect("function()", FUNC)
self.assertEqual(expect("function()", EXPR), expected)
with self.assertRaises(exceptions.ProgrammingError):
expect(word="", token="ABC")
