From 52b9f8c1ea191ce9c1ae0cd485306460cb52f71b Mon Sep 17 00:00:00 2001
From: Waylan Limberg <waylan.limberg@icloud.com>
Date: Tue, 30 Dec 2014 11:00:27 -0500
Subject: Add reset support to TOC extension.

Now, whenever the TOC extensiuon is loaded, the Markdown class instance
will always have a toc attribute (md.toc). Calling md.reset() will also
reset the toc attribute which defaults to an empty string.
---
 tests/test_extensions.py | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'tests')

diff --git a/tests/test_extensions.py b/tests/test_extensions.py
index e24118f..2380f17 100644
--- a/tests/test_extensions.py
+++ b/tests/test_extensions.py
@@ -733,6 +733,14 @@ 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.assertTrue(self.md.toc.startswith('<div class="toc">'))
+        self.md.reset()
+        self.assertEqual(self.md.toc, '')
+
 
 class TestSmarty(unittest.TestCase):
     def setUp(self):
-- 
cgit v1.2.3