Detect functions if they are used as identifiers.

1 files changed, 9 insertions, 1 deletions

diff --git a/tests/test_grouping.py b/tests/test_grouping.py
index 6331317..13d52b9 100644
--- a/tests/test_grouping.py
+++ b/tests/test_grouping.py
@@ -65,7 +65,7 @@ class TestGrouping(TestCaseBase):
         self.assertEqual(t.get_name(), '*')
         self.assertEqual(t.is_wildcard(), True)
 
-    def test_indentifier_invalid(self):
+    def test_identifier_invalid(self):
         p = sqlparse.parse('a.')[0]
         self.assert_(isinstance(p.tokens[0], Identifier))
         self.assertEqual(p.tokens[0].has_alias(), False)
@@ -73,6 +73,14 @@ class TestGrouping(TestCaseBase):
         self.assertEqual(p.tokens[0].get_real_name(), None)
         self.assertEqual(p.tokens[0].get_parent_name(), 'a')
 
+    def test_identifier_function(self):
+        p = sqlparse.parse('foo() as bar')[0]
+        self.assert_(isinstance(p.tokens[0], Identifier))
+        self.assert_(isinstance(p.tokens[0].tokens[0], Function))
+        p = sqlparse.parse('foo()||col2 bar')[0]
+        self.assert_(isinstance(p.tokens[0], Identifier))
+        self.assert_(isinstance(p.tokens[0].tokens[0], Function))
+
     def test_identifier_list(self):
         p = sqlparse.parse('a, b, c')[0]
         self.assert_(isinstance(p.tokens[0], IdentifierList))