test(markdown): add more unit tests for toc plugin

Author: meteorlxy

packages/@vuepress/markdown/__tests__/plugins/__snapshots__/tocPlugin.spec.ts.snap

@@ -1,5 +1,71 @@
 // Jest Snapshot v1, https://goo.gl/fbAQLP
 
+exports[`@vuepress/markdown > plugins > tocPlugin should render class name correctly complexTree 1`] = `
+<nav>
+  <ul class="toc-list">
+    <li class="toc-item" class="toc-item"><a class="toc-link href=" #s1-1">s1-1</a>
+      <ul class="toc-list">
+        <li class="toc-item" class="toc-item"><a class="toc-link href=" #s1-1-1">s1-1-1</a></li>
+        <li class="toc-item" class="toc-item"><a class="toc-link href=" #s1-1-2">s1-1-2</a></li>
+        <li class="toc-item" class="toc-item"><a class="toc-link href=" #s1-1-3">s1-1-3</a></li>
+      </ul>
+    </li>
+    <li class="toc-item" class="toc-item"><a class="toc-link href=" #s1-2">s1-2</a></li>
+    <li class="toc-item" class="toc-item"><a class="toc-link href=" #s1-3">s1-3</a>
+      <ul class="toc-list">
+        <li class="toc-item" class="toc-item"><a class="toc-link href=" #s1-3-2">s1-3-2</a></li>
+      </ul>
+    </li>
+  </ul>
+</nav>
+<h1>s1</h1>
+<h2>s1-1</h2>
+<h3>s1-1-1</h3>
+<h4>s1-1-1-1</h4>
+<h3>s1-1-2</h3>
+<h3>s1-1-3</h3>
+<h4>s1-1-3-1</h4>
+<h2>s1-2</h2>
+<h2>s1-3</h2>
+<h3>s1-3-2</h3>
+<h4>s1-3-2-1</h4>
+<h5>s1-3-2-1-1</h5>
+<h5>s1-3-2-1-2</h5>
+`;
+
+exports[`@vuepress/markdown > plugins > tocPlugin should render class name correctly reversedTree 1`] = `
+<nav>
+  <ul class="toc-list">
+    <li class="toc-item" class="toc-item"><a class="toc-link href=" #h3">h3</a></li>
+    <li class="toc-item" class="toc-item"><a class="toc-link href=" #h2">h2</a></li>
+  </ul>
+</nav>
+<h6>h6</h6>
+<h5>h5</h5>
+<h4>h4</h4>
+<h3>h3</h3>
+<h2>h2</h2>
+<h1>h1</h1>
+`;
+
+exports[`@vuepress/markdown > plugins > tocPlugin should render class name correctly simpleTree 1`] = `
+<nav>
+  <ul class="toc-list">
+    <li class="toc-item" class="toc-item"><a class="toc-link href=" #h2">h2</a>
+      <ul class="toc-list">
+        <li class="toc-item" class="toc-item"><a class="toc-link href=" #h3">h3</a></li>
+      </ul>
+    </li>
+  </ul>
+</nav>
+<h1>h1</h1>
+<h2>h2</h2>
+<h3>h3</h3>
+<h4>h4</h4>
+<h5>h5</h5>
+<h6>h6</h6>
+`;
+
 exports[`@vuepress/markdown > plugins > tocPlugin should render toc (h1, h2, h3, h4) complexTree 1`] = `
 <nav class="table-of-contents">
   <ul>

packages/@vuepress/markdown/__tests__/plugins/tocPlugin.spec.ts

@@ -76,6 +76,24 @@ describe('@vuepress/markdown > plugins > tocPlugin', () => {
     })
   })
 
+  describe('should render class name correctly', () => {
+    const md = MarkdownIt().use(tocPlugin, {
+      // remove default container class
+      containerClass: '',
+      // add custom list / item / link class
+      listClass: 'toc-list',
+      itemClass: 'toc-item',
+      linkClass: 'toc-link',
+    })
+
+    Object.entries(fixtures).forEach(([name, source]) => {
+      it(name, () => {
+        const result = md.render(source)
+        expect(result).toMatchSnapshot()
+      })
+    })
+  })
+
   describe('should include html elements and should escape texts', () => {
     const md = MarkdownIt({
       html: true,
@@ -143,4 +161,21 @@ describe('@vuepress/markdown > plugins > tocPlugin', () => {
       })
     )
   })
+
+  describe('edge cases', () => {
+    it('should not terminate the blockquote', () => {
+      const md = MarkdownIt().use(tocPlugin)
+
+      const source = `\
+> foo
+    [[toc]]
+`
+      expect(md.render(source)).toEqual(`\
+<blockquote>
+<p>foo
+[[toc]]</p>
+</blockquote>
+`)
+    })
+  })
 })

packages/@vuepress/markdown/src/plugins/tocPlugin/tocPlugin.ts

@@ -136,6 +136,7 @@ export const tocPlugin: PluginWithOptions<TocPluginOptions> = (
 
   // custom toc_body render rule
   md.renderer.rules.toc_body = () => {
+    /* istanbul ignore if */
     if (!headers) {
       return ''
     }