Fixed 'compact' tests

1 files changed, 21 insertions, 4 deletions

diff --git a/tests/test_functions.py b/tests/test_functions.py
index 5ab1586..ff8625a 100644
--- a/tests/test_functions.py
+++ b/tests/test_functions.py
@@ -15,7 +15,7 @@ from sqlparse.filters   import compact
 from sqlparse.functions import getcolumns, getlimit, IsType
 
 
-class Test_SQL(TestCase):
+class Test_IncludeStatement(TestCase):
     sql = """-- type: script
             -- return: integer
 
@@ -25,6 +25,24 @@ class Test_SQL(TestCase):
                             VALUES(:inode)
             LIMIT 1"""
 
+    def test_includeStatement(self):
+        stream = compact(tokenize(self.sql), 'tests/files')
+
+        result = Tokens2Unicode(stream)
+
+        self.assertEqual(result,
+            'INSERT INTO dir_entries(type)VALUES(:type);INSERT INTO '
+            'directories(inode)VALUES(:inode)LIMIT 1')
+
+
+class Test_SQL(TestCase):
+    sql = """-- type: script
+            -- return: integer
+
+            INSERT INTO directories(inode)
+                            VALUES(:inode)
+            LIMIT 1"""
+
     sql2 = """SELECT child_entry,asdf AS inode, creation
               FROM links
               WHERE parent_dir == :parent_dir AND name == :name
@@ -63,13 +81,12 @@ LIMIT 1"""
 
 class Test_Compact(Test_SQL):
     def test_compact1(self):
-        stream = compact(tokenize(self.sql), 'tests/files')
+        stream = compact(tokenize(self.sql))
 
         result = Tokens2Unicode(stream)
 
         self.assertEqual(result,
-            'INSERT INTO dir_entries(type)VALUES(:type);INSERT INTO '
-            'directories(inode)VALUES(:inode)LIMIT 1')
+                         'INSERT INTO directories(inode)VALUES(:inode)LIMIT 1')
 
     def test_compact2(self):
         stream = tokenize(self.sql2)