aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--markdown/extensions/toc.py6
-rw-r--r--tests/test_extensions.py19
2 files changed, 25 insertions, 0 deletions
diff --git a/markdown/extensions/toc.py b/markdown/extensions/toc.py
index 99542ed..db6659c 100644
--- a/markdown/extensions/toc.py
+++ b/markdown/extensions/toc.py
@@ -234,6 +234,9 @@ class TocExtension(Extension):
super(TocExtension, self).__init__(*args, **kwargs)
def extendMarkdown(self, md, md_globals):
+ md.registerExtension(self)
+ self.md = md
+ self.reset()
tocext = self.TreeProcessorClass(md)
tocext.config = self.getConfigs()
# Headerid ext is set to '>prettify'. With this set to '_end',
@@ -243,6 +246,9 @@ class TocExtension(Extension):
# to redefine ids after toc is created. But we do want toc prettified.
md.treeprocessors.add("toc", tocext, "_end")
+ def reset(self):
+ self.md.toc = ''
+
def makeExtension(*args, **kwargs):
return TocExtension(*args, **kwargs)
diff --git a/tests/test_extensions.py b/tests/test_extensions.py
index e24118f..760069f 100644
--- a/tests/test_extensions.py
+++ b/tests/test_extensions.py
@@ -733,6 +733,25 @@ class TestTOC(unittest.TestCase):
'</div>\n'
)
+ def testReset(self):
+ """ Test TOC Reset. """
+ self.assertEqual(self.md.toc, '')
+ self.md.convert('# Header 1\n\n## Header 2')
+ self.assertEqual(
+ self.md.toc,
+ '<div class="toc">\n'
+ '<ul>\n' # noqa
+ '<li><a href="#header-1">Header 1</a>' # noqa
+ '<ul>\n' # noqa
+ '<li><a href="#header-2">Header 2</a></li>\n' # noqa
+ '</ul>\n' # noqa
+ '</li>\n' # noqa
+ '</ul>\n' # noqa
+ '</div>\n'
+ )
+ self.md.reset()
+ self.assertEqual(self.md.toc, '')
+
class TestSmarty(unittest.TestCase):
def setUp(self):