Add base test for generation

1 files changed, 11 insertions, 2 deletions

diff --git a/lib/book.js b/lib/book.js
index 062e02c..73f3847 100644
--- a/lib/book.js
+++ b/lib/book.js
@@ -6,7 +6,9 @@ var fs = require("./utils/fs");
 var Configuration = require("./configuration");
 var TemplateEngine = require("./template");
 var Plugin = require("./plugin");
+
 var parsers = require("./parsers");
+var generators = require("./generators");
 
 var Book = function(root, options, parent) {
 	// Root folder of the book
@@ -106,14 +108,21 @@ Book.prototype.parse = function() {
 
 // Generate the output
 Book.prototype.generate = function() {
-	var that = this;
+	var that = this, generator;
+
 	if (that.isMultilingual()) return that.generateMultiLingual();
 
 	return Q()
 
 	// Clean output folder
 	.then(function() {
-		return fs.clean(that.options.output);
+		return fs.remove(that.options.output);
+	})
+
+	// Create generator
+	.then(function() {
+		Generator = generators[that.options.generator];
+		if (!Generator) throw "Generator '"+that.options.generator+"' doesn't exist";
 	});
 };