dbcli/sqlglot
1
0
Fork 0
Code Issues Activity

Merging upstream version 11.4.1.

Browse source 
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
Daniel Baumann 2025-02-13 15:46:19 +01:00
parent ecb42ec17f
commit 63746a3e92
Signed by: daniel
GPG key ID: FBB4F0E80A80222F
89 changed files with 35352 additions and 33081 deletions
Download patch file Download diff file Expand all files Collapse all files

40
tests/test_parser.py
View file

@ -102,6 +102,13 @@ class TestParser(unittest.TestCase):
self.assertEqual(expressions[1].sql(), "ADD JAR s3://a")
self.assertEqual(expressions[2].sql(), "SELECT 1")
def test_lambda_struct(self):
expression = parse_one("FILTER(a.b, x -> x.id = id)")
lambda_expr = expression.expression
self.assertIsInstance(lambda_expr.this.this, exp.Dot)
self.assertEqual(lambda_expr.sql(), "x -> x.id = id")
def test_transactions(self):
expression = parse_one("BEGIN TRANSACTION")
self.assertIsNone(expression.this)
@ -280,6 +287,39 @@ class TestParser(unittest.TestCase):
self.assertIsInstance(parse_one("TIMESTAMP()"), exp.Func)
self.assertIsInstance(parse_one("map.x"), exp.Column)
def test_set_expression(self):
set_ = parse_one("SET")
self.assertEqual(set_.sql(), "SET")
self.assertIsInstance(set_, exp.Set)
set_session = parse_one("SET SESSION x = 1")
self.assertEqual(set_session.sql(), "SET SESSION x = 1")
self.assertIsInstance(set_session, exp.Set)
set_item = set_session.expressions[0]
self.assertIsInstance(set_item, exp.SetItem)
self.assertIsInstance(set_item.this, exp.EQ)
self.assertIsInstance(set_item.this.this, exp.Identifier)
self.assertIsInstance(set_item.this.expression, exp.Literal)
self.assertEqual(set_item.args.get("kind"), "SESSION")
set_to = parse_one("SET x TO 1")
self.assertEqual(set_to.sql(), "SET x = 1")
self.assertIsInstance(set_to, exp.Set)
set_as_command = parse_one("SET DEFAULT ROLE ALL TO USER")
self.assertEqual(set_as_command.sql(), "SET DEFAULT ROLE ALL TO USER")
self.assertIsInstance(set_as_command, exp.Command)
self.assertEqual(set_as_command.this, "SET")
self.assertEqual(set_as_command.expression, " DEFAULT ROLE ALL TO USER")
def test_pretty_config_override(self):
self.assertEqual(parse_one("SELECT col FROM x").sql(), "SELECT col FROM x")
with patch("sqlglot.pretty", True):