Add tests for page hook

1 files changed, 34 insertions, 13 deletions

diff --git a/test/mock.js b/test/mock.js
index 5ff9f63..e15c4be 100644
--- a/test/mock.js
+++ b/test/mock.js
@@ -68,38 +68,52 @@ function setupDefaultBook(files, summary, opts) {
     }), opts);
 }
 
-// Output a book with a specific generator
-function outputDefaultBook(Output, files, summary, opts) {
-    return setupDefaultBook(files, summary, opts)
+// Prepare output for a book
+function setupOutput(Output, files, opts) {
+    return setupBook(files, opts)
     .then(function(book) {
         // Parse the book
         return book.parse()
 
         // Start generation
         .then(function() {
-            var output = new Output(book);
-            return output.generate()
-                .thenResolve(output);
+            return new Output(book);
         });
     });
 }
 
-// Output a book with a specific generator
-function outputBook(Output, files, opts) {
-    return setupBook(files, opts)
+// Prepare output for a book
+function setupDefaultOutput(Output, files, summary, opts) {
+    return setupDefaultBook(files, summary, opts)
     .then(function(book) {
         // Parse the book
         return book.parse()
 
         // Start generation
         .then(function() {
-            var output = new Output(book);
-            return output.generate()
-                .thenResolve(output);
+            return new Output(book);
         });
     });
 }
 
+// Output a book with a specific generator
+function outputDefaultBook(Output, files, summary, opts) {
+    return setupDefaultOutput(Output, files, summary, opts)
+    .then(function(output) {
+        return output.generate()
+            .thenResolve(output);
+    });
+}
+
+// Output a book with a specific generator
+function outputBook(Output, files, opts) {
+    return setupOutput(Output, files, opts)
+    .then(function(output) {
+        return output.generate()
+            .thenResolve(output);
+    });
+}
+
 // Log an error
 function logError(err) {
     console.log(err.stack || err);
@@ -107,10 +121,17 @@ function logError(err) {
 
 module.exports = {
     fs: nodeFS,
+
     setupFS: setupFS,
+
     setupBook: setupBook,
-    outputBook: outputBook,
     setupDefaultBook: setupDefaultBook,
+
+    setupOutput: setupOutput,
+    setupDefaultOutput: setupDefaultOutput,
+
+    outputBook: outputBook,
     outputDefaultBook: outputDefaultBook,
+
     logError: logError
 };