summaryrefslogtreecommitdiffstats
path: root/tests/test_parser.py
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2022-12-02 09:16:29 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2022-12-02 09:16:29 +0000
commit1a60bbae98d3b530924a6807a55f8250de19ea86 (patch)
tree87d3000f271a6604fff43db188731229aed918a8 /tests/test_parser.py
parentAdding upstream version 10.0.8. (diff)
downloadsqlglot-1a60bbae98d3b530924a6807a55f8250de19ea86.tar.xz
sqlglot-1a60bbae98d3b530924a6807a55f8250de19ea86.zip
Adding upstream version 10.1.3.upstream/10.1.3
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'tests/test_parser.py')
-rw-r--r--tests/test_parser.py67
1 files changed, 58 insertions, 9 deletions
diff --git a/tests/test_parser.py b/tests/test_parser.py
index c747ea3..fa7b589 100644
--- a/tests/test_parser.py
+++ b/tests/test_parser.py
@@ -15,6 +15,51 @@ class TestParser(unittest.TestCase):
self.assertIsInstance(parse_one("int", into=exp.DataType), exp.DataType)
self.assertIsInstance(parse_one("array<int>", into=exp.DataType), exp.DataType)
+ def test_parse_into_error(self):
+ expected_message = "Failed to parse into [<class 'sqlglot.expressions.From'>]"
+ expected_errors = [
+ {
+ "description": "Invalid expression / Unexpected token",
+ "line": 1,
+ "col": 1,
+ "start_context": "",
+ "highlight": "SELECT",
+ "end_context": " 1;",
+ "into_expression": exp.From,
+ }
+ ]
+ with self.assertRaises(ParseError) as ctx:
+ parse_one("SELECT 1;", "sqlite", [exp.From])
+ self.assertEqual(str(ctx.exception), expected_message)
+ self.assertEqual(ctx.exception.errors, expected_errors)
+
+ def test_parse_into_errors(self):
+ expected_message = "Failed to parse into [<class 'sqlglot.expressions.From'>, <class 'sqlglot.expressions.Join'>]"
+ expected_errors = [
+ {
+ "description": "Invalid expression / Unexpected token",
+ "line": 1,
+ "col": 1,
+ "start_context": "",
+ "highlight": "SELECT",
+ "end_context": " 1;",
+ "into_expression": exp.From,
+ },
+ {
+ "description": "Invalid expression / Unexpected token",
+ "line": 1,
+ "col": 1,
+ "start_context": "",
+ "highlight": "SELECT",
+ "end_context": " 1;",
+ "into_expression": exp.Join,
+ },
+ ]
+ with self.assertRaises(ParseError) as ctx:
+ parse_one("SELECT 1;", "sqlite", [exp.From, exp.Join])
+ self.assertEqual(str(ctx.exception), expected_message)
+ self.assertEqual(ctx.exception.errors, expected_errors)
+
def test_column(self):
columns = parse_one("select a, ARRAY[1] b, case when 1 then 1 end").find_all(exp.Column)
assert len(list(columns)) == 1
@@ -24,6 +69,9 @@ class TestParser(unittest.TestCase):
def test_float(self):
self.assertEqual(parse_one(".2"), parse_one("0.2"))
+ def test_unary_plus(self):
+ self.assertEqual(parse_one("+15"), exp.Literal.number(15))
+
def test_table(self):
tables = [t.sql() for t in parse_one("select * from a, b.c, .d").find_all(exp.Table)]
self.assertEqual(tables, ["a", "b.c", "d"])
@@ -157,8 +205,9 @@ class TestParser(unittest.TestCase):
def test_comments(self):
expression = parse_one(
"""
- --comment1
- SELECT /* this won't be used */
+ --comment1.1
+ --comment1.2
+ SELECT /*comment1.3*/
a, --comment2
b as B, --comment3:testing
"test--annotation",
@@ -169,13 +218,13 @@ class TestParser(unittest.TestCase):
"""
)
- self.assertEqual(expression.comment, "comment1")
- self.assertEqual(expression.expressions[0].comment, "comment2")
- self.assertEqual(expression.expressions[1].comment, "comment3:testing")
- self.assertEqual(expression.expressions[2].comment, None)
- self.assertEqual(expression.expressions[3].comment, "comment4 --foo")
- self.assertEqual(expression.expressions[4].comment, "")
- self.assertEqual(expression.expressions[5].comment, " space")
+ self.assertEqual(expression.comments, ["comment1.1", "comment1.2", "comment1.3"])
+ self.assertEqual(expression.expressions[0].comments, ["comment2"])
+ self.assertEqual(expression.expressions[1].comments, ["comment3:testing"])
+ self.assertEqual(expression.expressions[2].comments, None)
+ self.assertEqual(expression.expressions[3].comments, ["comment4 --foo"])
+ self.assertEqual(expression.expressions[4].comments, [""])
+ self.assertEqual(expression.expressions[5].comments, [" space"])
def test_type_literals(self):
self.assertEqual(parse_one("int 1"), parse_one("CAST(1 AS INT)"))