Test ebook generation instead of pdf output

4 files changed, 20 insertions, 17 deletions

diff --git a/lib/generators/ebook.js b/lib/generators/ebook.js
index fa5dcbd..b173c19 100644
--- a/lib/generators/ebook.js
+++ b/lib/generators/ebook.js
@@ -12,10 +12,10 @@ var Generator = function(book, format) {
     BaseGenerator.apply(this, arguments);
 
     // eBook format
-    this.ebookFormat = format || "pdf";
+    this.ebookFormat = format;
 
     // Styles to use
-    this.styles = ["ebook", this.ebookFormat];
+    this.styles = _.compact(["ebook", this.ebookFormat]);
 };
 util.inherits(Generator, BaseGenerator);
 
@@ -47,6 +47,8 @@ Generator.prototype.finish = function() {
     	return that.writeSummary();
     })
     .then(function() {
+        if (!that.ebookFormat) return Q();
+
         var d = Q.defer();
 
         if (!that.options.cover && fs.existsSync(path.join(that.options.output, "cover.jpg"))) {
diff --git a/lib/generators/index.js b/lib/generators/index.js
index 9189196..e619188 100644
--- a/lib/generators/index.js
+++ b/lib/generators/index.js
@@ -4,6 +4,7 @@ var EbookGenerator = require("./ebook");
 module.exports = {
 	json: require("./json"),
 	website: require("./website"),
+	ebook: EbookGenerator,
 	pdf: _.partialRight(EbookGenerator, "pdf"),
 	mobi: _.partialRight(EbookGenerator, "mobi"),
 	epub: _.partialRight(EbookGenerator, "epub")
diff --git a/test/ebook.js b/test/ebook.js
new file mode 100644
index 0000000..e835a6e
--- /dev/null
+++ b/test/ebook.js
@@ -0,0 +1,15 @@
+var path = require('path');
+var _ = require('lodash');
+var assert = require('assert');
+
+var fs = require("fs");
+var fsUtil = require("../lib/utils/fs");
+
+
+describe('eBook Generator', function () {
+    it('should correctly generate ebook pages', function(done) {
+        testGeneration(books[1], "ebook", function(output) {
+            assert(fs.existsSync(path.join(output, "SUMMARY.html")));
+        }, done);
+    });
+});
diff --git a/test/pdf.js b/test/pdf.js
deleted file mode 100644
index 9b5cf42..0000000
--- a/test/pdf.js
+++ /dev/null
@@ -1,15 +0,0 @@
-var path = require('path');
-var _ = require('lodash');
-var assert = require('assert');
-
-var fs = require("fs");
-var fsUtil = require("../lib/utils/fs");
-
-
-describe('PDF Generator', function () {
-    it('should correctly generate a pdf', function(done) {
-        testGeneration(books[1], "pdf", function(output) {
-            assert(fs.existsSync(path.join(output, "book.pdf")));
-        }, done);
-    });
-});