From aa1d1f1ea72887a3efb78f4950e27bc79dfa3766 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Mon, 27 Feb 2023 11:46:33 +0100 Subject: Adding upstream version 11.2.3. Signed-off-by: Daniel Baumann --- tests/test_tokens.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) (limited to 'tests/test_tokens.py') diff --git a/tests/test_tokens.py b/tests/test_tokens.py index d30c445..0888555 100644 --- a/tests/test_tokens.py +++ b/tests/test_tokens.py @@ -18,6 +18,18 @@ class TestTokens(unittest.TestCase): for sql, comment in sql_comment: self.assertEqual(tokenizer.tokenize(sql)[0].comments, comment) + def test_token_line(self): + tokens = Tokenizer().tokenize( + """SELECT /* + line break + */ + 'x + y', + x""" + ) + + self.assertEqual(tokens[-1].line, 6) + def test_jinja(self): tokenizer = Tokenizer() @@ -26,6 +38,7 @@ class TestTokens(unittest.TestCase): SELECT {{ x }}, {{- x -}}, + {# it's a comment #} {% for x in y -%} a {{+ b }} {% endfor %}; -- cgit v1.2.3