fix: call walkTokens when calling marked with a callback(#2060)

markedjs · May 21, 2021 · 1d97308 · 1d97308 · vercel · May 21, 2021
1 parent 29a301b
commit 1d97308
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/marked.js b/src/marked.js
@@ -51,6 +51,9 @@ function marked(src, opt, callback) {
 
       if (!err) {
         try {
+          if (opt.walkTokens) {
+            marked.walkTokens(tokens, opt.walkTokens);
+          }
           out = Parser.parse(tokens, opt);
         } catch (e) {
           err = e;

diff --git a/test/unit/marked-spec.js b/test/unit/marked-spec.js
@@ -183,6 +183,20 @@ describe('use extension', () => {
     expect(walked).toBe(2);
   });
 
+  it('should use walkTokens in async', (done) => {
+    let walked = 0;
+    const extension = {
+      walkTokens(token) {
+        walked++;
+      }
+    };
+    marked.use(extension);
+    marked('text', () => {
+      expect(walked).toBe(2);
+      done();
+    });
+  });
+
   it('should use options from extension', () => {
     const extension = {
       headerIds: false