Add tests

Author: naokazuterada

markdowntoc/markdowntoc_insert.py

@@ -74,7 +74,7 @@ def get_toc_open_tag(self):
         return toc_open_tags
 
     def get_toc_close_tag(self, start):
-        close_tags = self.view.find_all("^<!-- /MarkdownTOC -->\n")
+        close_tags = self.view.find_all(r'<!--[\s\n]+/MarkdownTOC[\s\n]+-->\n')
         close_tags = self.remove_items_in_codeblock(close_tags)
         for close_tag in close_tags:
             if start < close_tag.begin():

tests/base.py

@@ -39,13 +39,13 @@ def moveTo(self, pos):
     def getTOC_text(self):
         """Find TOC in the document, and return the list texts in it."""
         toc_region = self.view.find(
-            "<!-- MarkdownTOC .*-->(.|\n)+?<!-- /MarkdownTOC -->",
+            r'<!--[\s\n]+MarkdownTOC[\s\n]+[^>]*-->\n(.|\n)+?<!--[\s\n]+/MarkdownTOC[\s\n]+-->',
             sublime.IGNORECASE)
         toc_all = self.view.substr(toc_region)
 
         # pick toc contents
-        toc_contents = re.sub(r'<!-- MarkdownTOC .* -->', '', toc_all)
-        toc_contents = re.sub(r'<!-- /MarkdownTOC -->', '', toc_contents)
+        toc_contents = re.sub(r'<!--[\s\n]+MarkdownTOC[\s\n]+[^>]*-->', '', toc_all)
+        toc_contents = re.sub(r'<!--[\s\n]+/MarkdownTOC[\s\n]+-->', '', toc_contents)
         toc_contents = toc_contents.rstrip()
 
         return toc_contents

tests/tag_variation.py

@@ -0,0 +1,96 @@
+# coding:utf-8
+from base import TestBase
+
+class TagVariation(TestBase):
+    """Test for attributes 'style'"""
+
+    # for debug
+    # def tearDown(self):
+    #     pass
+
+    def test_1(self):
+        """Tag Variation 1"""
+        text = \
+"""
+
+<!-- MarkdownTOC autolink="true" bracket="square" style="ordered" -->
+
+<!-- /MarkdownTOC -->
+
+# foo
+"""
+        toc = self.init_update(text)['toc']
+        self.assert_In('1. [foo][foo]', toc)
+
+    def test_2(self):
+        """Tag Variation 2"""
+        text = \
+"""
+
+<!-- MarkdownTOC
+autolink="true" bracket="square" style="ordered" -->
+
+<!-- /MarkdownTOC -->
+
+# foo
+"""
+        toc = self.init_update(text)['toc']
+        self.assert_In('1. [foo][foo]', toc)
+
+    def test_3(self):
+        """Tag Variation 3"""
+        text = \
+"""
+
+<!--
+MarkdownTOC
+autolink="true"
+bracket="square"
+style="ordered" -->
+
+<!-- /MarkdownTOC -->
+
+# foo
+"""
+        toc = self.init_update(text)['toc']
+        self.assert_In('1. [foo][foo]', toc)
+
+    def test_4(self):
+        """Tag Variation 4"""
+        text = \
+"""
+
+<!--
+MarkdownTOC
+autolink="true"
+bracket="square"
+style="ordered"
+-->
+
+<!-- /MarkdownTOC -->
+
+# foo
+"""
+        toc = self.init_update(text)['toc']
+        self.assert_In('1. [foo][foo]', toc)
+
+    def test_5(self):
+        """Tag Variation 5"""
+        text = \
+"""
+
+<!--
+MarkdownTOC
+autolink="true"
+bracket="square"
+style="ordered"
+-->
+
+<!--
+/MarkdownTOC
+-->
+
+# foo
+"""
+        toc = self.init_update(text)['toc']
+        self.assert_In('1. [foo][foo]', toc)