test: add test for parseHeaders

ulivz · ulivz · commit 4f0b316260dc · 2018-06-21T00:22:23.000+08:00
diff --git a/lib/util/parseHeaders.js b/lib/util/parseHeaders.js
@@ -18,7 +18,7 @@ const removeMarkdownToken = str => String(str)
   .replace(/(\\)(\*|_|`)/g, '$2')               // remove escape char '\'
 
 exports.removeTailHtml = (str) => {
-  return String(str).replace(/<.*>\s*$/g, '')
+  return String(str).replace(/\s*?<.*>\s*$/g, '')
 }
 
 // only remove some md tokens.
diff --git a/test/util/parseHeaders.spec.js b/test/util/parseHeaders.spec.js
@@ -0,0 +1,37 @@
+import {
+  parseHeaders,
+  removeTailHtml
+} from '@/util/parseHeaders'
+
+describe('parseHeaders', () => {
+  test('should unescape html', () => {
+    const input = '&lt;div&gt;'
+    expect(parseHeaders(input)).toBe('<div>')
+  })
+
+  test('should remove markdown tokens correctly', () => {
+    const asserts = {
+      // #238
+      '[vue](vuejs.org)': 'vue',
+      '`vue`': 'vue',
+      '*vue*': 'vue',
+      '**vue**': 'vue',
+      '***vue***': 'vue',
+      '_vue_': 'vue',
+      '\\_vue\\_': '_vue_',
+      '\\*vue\\*': '*vue*',
+
+      // #564 For multiple markdown tokens
+      '`a` and `b`': 'a and b',
+      '***bold and italic***': 'bold and italic',
+      '**bold** and *italic*': 'bold and italic'
+    }
+    Object.keys(asserts).forEach(input => {
+      expect(parseHeaders(input)).toBe(asserts[input])
+    })
+  })
+
+  test('should remove tail html correctly', () => {
+    expect(removeTailHtml('# H1 <div></div>>')).toBe('# H1')
+  })
+})

Original file line number	Diff line number	Diff line change
`@@ -18,7 +18,7 @@ const removeMarkdownToken = str => String(str)`
`18`	`18`	.replace(/(\\)(\*\|_\|`)/g, '$2') // remove escape char '\'
`19`	`19`
`20`	`20`	`exports.removeTailHtml = (str) => {`
`21`		`- return String(str).replace(/<.>\s$/g, '')`
	`21`	`+ return String(str).replace(/\s?<.>\s*$/g, '')`
`22`	`22`	`}`
`23`	`23`
`24`	`24`	`// only remove some md tokens.`