1 files changed, 18 insertions, 33 deletions
diff --git a/test/website.js b/test/website.js
index afbe7aa..91340a0 100644
--- a/test/website.js
+++ b/test/website.js
@@ -1,40 +1,25 @@
+var fs = require('fs');
 var path = require('path');
-var _ = require('lodash');
-var assert = require('assert');
 
-var fs = require("fs");
-var fsUtil = require("../lib/utils/fs");
+describe('Website generator', function () {
+    describe('Basic Book', function() {
+        var book;
 
+        before(function() {
+            return books.generate("basic", "website")
+                .then(function(_book) {
+                    book = _book;
+                });
+        });
 
-describe('Website Generator', function () {
-    it('should correctly generate a book to website', function(done) {
-        testGeneration(books[1], "website", function(output) {
-            assert(fs.existsSync(path.join(output, "index.html")));
-            assert(fs.existsSync(path.join(output, "search_index.json")));
-        }, done);
-    });
-
-    it('should correctly include styles in website', function(done) {
-        testGeneration(books[0], "website", function(output) {
-            assert(fs.existsSync(path.join(output, "styles/website.css")));
-
-            var INDEX = fs.readFileSync(path.join(output, "index.html")).toString();
-            assert(INDEX.indexOf("styles/website.css") > 0);
-        }, done);
-    });
-
-    it('should correctly include glossary in website', function(done) {
-        testGeneration(books[0], "website", function(output) {
-            assert(fs.existsSync(path.join(output, "GLOSSARY.html")));
-        }, done);
-    });
+        it('should correctly output an index.html', function() {
+            book.should.have.file("index.html");
+        });
 
-    it('should correctly generate a multilingual book to website', function(done) {
-        testGeneration(books[2], "website", function(output) {
-            assert(fs.existsSync(path.join(output, "index.html")));
-            assert(fs.existsSync(path.join(output, "gitbook")));
-            assert(fs.existsSync(path.join(output, "fr/index.html")));
-            assert(fs.existsSync(path.join(output, "en/index.html")));
-        }, done);
+        it('should correctly copy assets', function() {
+            book.should.have.file("gitbook");
+            book.should.have.file("gitbook/app.js");
+            book.should.have.file("gitbook/style.css");
+        });
     });
 });