From cf49728f719975144a958f23ba5f3336fb81ae55 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sat, 27 Apr 2024 04:50:25 +0200 Subject: Merging upstream version 23.12.1. Signed-off-by: Daniel Baumann --- tests/test_transpile.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) (limited to 'tests/test_transpile.py') diff --git a/tests/test_transpile.py b/tests/test_transpile.py index 95fba30..22085b3 100644 --- a/tests/test_transpile.py +++ b/tests/test_transpile.py @@ -123,6 +123,10 @@ class TestTranspile(unittest.TestCase): "SELECT * FROM t1\n/*x*/\nUNION ALL SELECT * FROM t2", "SELECT * FROM t1 /* x */ UNION ALL SELECT * FROM t2", ) + self.validate( + "/* comment */ SELECT * FROM a UNION SELECT * FROM b", + "/* comment */ SELECT * FROM a UNION SELECT * FROM b", + ) self.validate( "SELECT * FROM t1\n/*x*/\nINTERSECT ALL SELECT * FROM t2", "SELECT * FROM t1 /* x */ INTERSECT ALL SELECT * FROM t2", @@ -524,6 +528,28 @@ INNER JOIN y USING (id)""", pretty=True, ) + self.validate( + """with x as ( + SELECT * + /* +NOTE: LEFT JOIN because blah blah blah + */ + FROM a +) +select * from x""", + """WITH x AS ( + SELECT + * + /* +NOTE: LEFT JOIN because blah blah blah + */ + FROM a +) +SELECT + * +FROM x""", + pretty=True, + ) def test_types(self): self.validate("INT 1", "CAST(1 AS INT)") -- cgit v1.2.3