Handle wildcards in identifiers.

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]