diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_format.py | 5 | ||||
| -rw-r--r-- | tests/test_grouping.py | 12 |
2 files changed, 17 insertions, 0 deletions
diff --git a/tests/test_format.py b/tests/test_format.py index 3180d10..4c2742d 100644 --- a/tests/test_format.py +++ b/tests/test_format.py @@ -134,6 +134,11 @@ class TestFormatReindent(TestCaseBase): 'from table1,', ' table2', 'where 1 = 2'])) + s = 'select a.*, b.id from a, b' + self.ndiffAssertEqual(f(s), '\n'.join(['select a.*,', + ' b.id', + 'from a,', + ' b'])) def test_case(self): f = lambda sql: sqlparse.format(sql, reindent=True) diff --git a/tests/test_grouping.py b/tests/test_grouping.py index fa72275..26e5d10 100644 --- a/tests/test_grouping.py +++ b/tests/test_grouping.py @@ -43,6 +43,18 @@ class TestGrouping(TestCaseBase): self.ndiffAssertEqual(s, parsed.to_unicode()) self.assert_(isinstance(parsed.tokens[-1].tokens[3], Identifier)) + def test_identifier_wildcard(self): + p = sqlparse.parse('a.*, b.id')[0] + self.assert_(isinstance(p.tokens[0], IdentifierList)) + self.assert_(isinstance(p.tokens[0].tokens[0], Identifier)) + self.assert_(isinstance(p.tokens[0].tokens[-1], Identifier)) + + def test_identifier_name_wildcard(self): + p = sqlparse.parse('a.*')[0] + t = p.tokens[0] + self.assertEqual(t.get_name(), '*') + self.assertEqual(t.is_wildcard(), True) + def test_where(self): s = 'select * from foo where bar = 1 order by id desc' p = sqlparse.parse(s)[0] |