1 files changed, 18 insertions, 1 deletions
diff --git a/test/mock.js b/test/mock.js
index 919a992..973c74d 100644
--- a/test/mock.js
+++ b/test/mock.js
@@ -7,6 +7,7 @@ require('should');
 require('should-promised');
 
 var Book = require('../').Book;
+var Output = require('../lib/output');
 var NodeFS = require('../lib/fs/node');
 
 // Create filesystem instance for testing
@@ -57,7 +58,23 @@ function setupDefaultBook(files, opts) {
     }), opts);
 }
 
+// Output a book with a specific generator
+function outputDefaultBook(generator, files, opts) {
+    return setupDefaultBook(files, opts)
+    .then(function(book) {
+        // Parse the book
+        return book.parse()
+
+        // Start generation
+        .then(function() {
+            var output = new Output(book, generator);
+            return output.generate();
+        });
+    });
+}
+
 module.exports = {
     setupBook: setupBook,
-    setupDefaultBook: setupDefaultBook
+    setupDefaultBook: setupDefaultBook,
+    outputDefaultBook: outputDefaultBook
 };