Add test for custom print style

4 files changed, 38 insertions, 0 deletions

diff --git a/test/books/style-print/README.md b/test/books/style-print/README.md
new file mode 100644
index 0000000..09ade40
--- /dev/null
+++ b/test/books/style-print/README.md
@@ -0,0 +1,3 @@
+# Readme
+
+Default description for the book.
diff --git a/test/books/style-print/SUMMARY.md b/test/books/style-print/SUMMARY.md
new file mode 100644
index 0000000..ac9323c
--- /dev/null
+++ b/test/books/style-print/SUMMARY.md
@@ -0,0 +1 @@
+# Summary
diff --git a/test/books/style-print/styles/print.css b/test/books/style-print/styles/print.css
new file mode 100644
index 0000000..b05faf8
--- /dev/null
+++ b/test/books/style-print/styles/print.css
@@ -0,0 +1,3 @@
+body {
+    color: red;
+}
diff --git a/test/ebook.js b/test/ebook.js
index c9bb924..9b353d2 100644
--- a/test/ebook.js
+++ b/test/ebook.js
@@ -21,4 +21,35 @@ describe('eBook generator', function () {
             book.should.have.file("gitbook/style.css");
         });
     });
+
+    describe('Custom styles', function() {
+        var book;
+
+        before(function() {
+            return books.generate("style-print", "ebook")
+                .then(function(_book) {
+                    book = _book;
+                });
+        });
+
+        it('should remove default print.css', function() {
+            var PAGE = fs.readFileSync(
+                path.join(book.options.output, "index.html"),
+                { encoding: "utf-8" }
+            );
+            PAGE.should.be.html({
+                "link": {
+                    count: 1,
+                    attributes: {
+                        href: "./styles/print.css"
+                    }
+                }
+            });
+        });
+
+        it('should correctly print.css', function() {
+            book.should.have.file("styles");
+            book.should.have.file("styles/print.css");
+        });
+    })
 });