Added test for Issue 57

piranna · piranna · commit a911e95665b3 · 2012-01-23T10:51:33.000+01:00
diff --git a/tests/files/begintag_2.sql b/tests/files/begintag_2.sql
@@ -0,0 +1,13 @@
+CREATE TRIGGER IF NOT EXISTS remove_if_it_was_the_last_file_link
+-- Delete the direntry when is removed it's last static link
+    AFTER DELETE ON links
+    WHEN NOT EXISTS
+    (
+        SELECT * FROM links
+        WHERE child_entry = OLD.child_entry
+        LIMIT 1
+    )
+BEGIN
+    DELETE FROM dir_entries
+    WHERE dir_entries.inode = OLD.child_entry;
+END;
diff --git a/tests/test_split.py b/tests/test_split.py
@@ -69,6 +69,12 @@ def test_begintag(self):
         self.assertEqual(len(stmts), 3)
         self.ndiffAssertEqual(''.join(unicode(q) for q in stmts), sql)
 
+    def test_begintag_2(self):
+        sql = load_file('begintag_2.sql')
+        stmts = sqlparse.parse(sql)
+        self.assertEqual(len(stmts), 1)
+        self.ndiffAssertEqual(''.join(unicode(q) for q in stmts), sql)
+
     def test_dropif(self):
         sql = 'DROP TABLE IF EXISTS FOO;\n\nSELECT * FROM BAR;'
         stmts = sqlparse.parse(sql)
