From 91a9efe0511ab5b1c769d90c04b8b329fe3173ad Mon Sep 17 00:00:00 2001 From: Andi Albrecht Date: Mon, 17 Jun 2013 08:10:06 +0200 Subject: Allow whitespaces in certain keywords (fixes #97). --- tests/test_tokenize.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) (limited to 'tests') diff --git a/tests/test_tokenize.py b/tests/test_tokenize.py index 02af141..3b529d9 100644 --- a/tests/test_tokenize.py +++ b/tests/test_tokenize.py @@ -161,3 +161,29 @@ class TestStream(unittest.TestCase): tokens = list(lex.get_tokens(stream)) self.assertEqual(len(tokens), 2) self.assertEqual(tokens[1][0], Error) + + +def test_parse_join(): + p = sqlparse.parse('LEFT JOIN foo')[0] + assert len(p.tokens) == 3 + assert p.tokens[0].ttype is Keyword + p = sqlparse.parse('LEFT OUTER JOIN foo')[0] + assert len(p.tokens) == 3 + assert p.tokens[0].ttype is Keyword + + +def test_parse_endifloop(): + p = sqlparse.parse('END IF')[0] + assert len(p.tokens) == 1 + assert p.tokens[0].ttype is Keyword + p = sqlparse.parse('END IF')[0] + assert len(p.tokens) == 1 + p = sqlparse.parse('END\t\nIF')[0] + assert len(p.tokens) == 1 + assert p.tokens[0].ttype is Keyword + p = sqlparse.parse('END LOOP')[0] + assert len(p.tokens) == 1 + assert p.tokens[0].ttype is Keyword + p = sqlparse.parse('END LOOP')[0] + assert len(p.tokens) == 1 + assert p.tokens[0].ttype is Keyword -- cgit v1.2.1