dbcli/sqlglot
1
0
Fork 0
Code Issues Activity

Merging upstream version 17.3.0.

Browse source 
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
Daniel Baumann 2025-02-13 20:44:18 +01:00
parent 335ae02913
commit 133b8dfc8d
Signed by: daniel
GPG key ID: FBB4F0E80A80222F
43 changed files with 5488 additions and 5047 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/test_expressions.py
View file

@ -442,12 +442,15 @@ class TestExpressions(unittest.TestCase):
expression.find(exp.Table).replace(parse_one("y"))
self.assertEqual(expression.sql(), "SELECT c, b FROM y")
def test_pop(self):
def test_arg_deletion(self):
# Using the pop helper method
expression = parse_one("SELECT a, b FROM x")
expression.find(exp.Column).pop()
self.assertEqual(expression.sql(), "SELECT b FROM x")
expression.find(exp.Column).pop()
self.assertEqual(expression.sql(), "SELECT FROM x")
expression.pop()
self.assertEqual(expression.sql(), "SELECT FROM x")
@ -455,6 +458,15 @@ class TestExpressions(unittest.TestCase):
expression.find(exp.With).pop()
self.assertEqual(expression.sql(), "SELECT * FROM x")
# Manually deleting by setting to None
expression = parse_one("SELECT * FROM foo JOIN bar")
self.assertEqual(len(expression.args.get("joins", [])), 1)
expression.set("joins", None)
self.assertEqual(expression.sql(), "SELECT * FROM foo")
self.assertEqual(expression.args.get("joins", []), [])
self.assertIsNone(expression.args.get("joins"))
def test_walk(self):
expression = parse_one("SELECT * FROM (SELECT * FROM x)")
self.assertEqual(len(list(expression.walk())), 9)
@ -539,6 +551,9 @@ class TestExpressions(unittest.TestCase):
self.assertIsInstance(parse_one("ARRAY(time, foo)"), exp.Array)
self.assertIsInstance(parse_one("STANDARD_HASH('hello', 'sha256')"), exp.StandardHash)
self.assertIsInstance(parse_one("DATE(foo)"), exp.Date)
self.assertIsInstance(parse_one("HEX(foo)"), exp.Hex)
self.assertIsInstance(parse_one("TO_HEX(foo)", read="bigquery"), exp.Hex)
self.assertIsInstance(parse_one("TO_HEX(MD5(foo))", read="bigquery"), exp.MD5)
def test_column(self):
column = parse_one("a.b.c.d")