diff options
| author | Fredy Wijaya <fredy.wijaya@gmail.com> | 2018-12-01 01:39:25 -0600 |
|---|---|---|
| committer | Andi Albrecht <albrecht.andi@gmail.com> | 2018-12-03 13:40:26 +0100 |
| commit | 0e56a02e14b6e2d87b955928d819e77b0d3f5693 (patch) | |
| tree | 4981a94a57f4a8bd03884dcde9fdf9d88aa4399d /tests | |
| parent | 488505f6c448e7eb0e4a1915bdc5b6130d44a68a (diff) | |
| download | sqlparse-0e56a02e14b6e2d87b955928d819e77b0d3f5693.tar.gz | |
Fix reindent issue for parenthesis (fixes issue 427)
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_format.py | 5 | ||||
| -rw-r--r-- | tests/test_parse.py | 21 |
2 files changed, 26 insertions, 0 deletions
diff --git a/tests/test_format.py b/tests/test_format.py index 72af62e..f0dbcf2 100644 --- a/tests/test_format.py +++ b/tests/test_format.py @@ -362,6 +362,11 @@ class TestFormatReindent(object): 'from', ' (select *', ' from foo);']) + assert f("select f(1)") == 'select f(1)' + assert f("select f( 1 )") == 'select f(1)' + assert f("select f(\n\n\n1\n\n\n)") == 'select f( 1 )' + assert f("select f(\n\n\n 1 \n\n\n)") == 'select f( 1 )' + assert f("select f(\n\n\n 1 \n\n\n)") == 'select f( 1 )' def test_where(self): f = lambda sql: sqlparse.format(sql, reindent=True) diff --git a/tests/test_parse.py b/tests/test_parse.py index 30be32b..d191f65 100644 --- a/tests/test_parse.py +++ b/tests/test_parse.py @@ -433,3 +433,24 @@ def test_get_real_name(): assert len(stmts) == 1 assert 'a' == stmts[0].get_real_name() assert 't' == stmts[0].get_alias() + + +def test_parenthesis(): + tokens = sqlparse.parse("(\n\n1\n\n)")[0].tokens[0].tokens + assert list(map(lambda t: t.ttype, tokens)) == [T.Punctuation, + T.Newline, + T.Newline, + T.Number.Integer, + T.Newline, + T.Newline, + T.Punctuation] + tokens = sqlparse.parse("(\n\n 1 \n\n)")[0].tokens[0].tokens + assert list(map(lambda t: t.ttype, tokens)) == [T.Punctuation, + T.Newline, + T.Newline, + T.Whitespace, + T.Number.Integer, + T.Whitespace, + T.Newline, + T.Newline, + T.Punctuation] |