Add _linkRefs flag.

mkdoc · Mar 21, 2016 · ffc342d · ffc342d
1 parent 4471bd7
commit ffc342d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/collator.js b/collator.js
@@ -41,14 +41,16 @@ function flush(cb) {
     var doc = Node.createDocument()
       , para = Node.createNode(Node.PARAGRAPH);
 
+    doc._linkRefs = true;
+
     this.refs.forEach(function(ref) {
       para.appendChild(attach(ref));
     })
 
     // NOTE: do not append paragraph, push to the stream
     this.push(doc);
     this.push(para);
-    this.push(Node.createNode(Node.EOF));
+    this.push(Node.createNode(Node.EOF, {_linkRefs: true}));
   }
   cb();
 }

diff --git a/test/spec/links-callback.js b/test/spec/links-callback.js
@@ -24,12 +24,12 @@ describe('mkref:', function() {
     function onFinish() {
       var result = utils.result(target);
 
-      //console.dir(result);
-
       // open document
       expect(result[0]._type).to.eql(Node.DOCUMENT);
+
       // mock document paragraph
       expect(result[1]._type).to.eql(Node.PARAGRAPH);
+
       // eof main document
       expect(result[2]._type).to.eql(Node.EOF);