gitbook/test/images.js

var fs = require("fs");
var _ = require("lodash");
var path = require("path");
var cheerio = require("cheerio");

describe("Images", function () {
    var book, readme, $, $img, srcs;

    before(function() {
        return books.generate("images", "ebook")
            .then(function(_book) {
                book = _book;

                readme = fs.readFileSync(
                    path.join(book.options.output, "index.html"),
                    { encoding: "utf-8" }
                );
                $ = cheerio.load(readme);
                $img = $("img");
                srcs = $img.map(function() {
                    return $(this).attr("src");
                });
            });
    });

    it("should detect all images", function() {
        _.uniq(srcs).should.have.lengthOf(4);
    });

    it("should keep image tags", function() {
        srcs.should.have.lengthOf(5);
    });

    it("should not have .svg files", function() {
        _.each(srcs, function(src) {
            path.extname(src).should.not.equal(".svg");
        });
    });

    it("should correctly convert svg images to png", function() {
        _.each(srcs, function(src) {
            book.should.have.file(src);
        });
    });

    it("should handle relative paths", function() {
        var PAGE = fs.readFileSync(
            path.join(book.options.output, "folder/PAGE.html"),
            { encoding: "utf-8" }
        );

        PAGE.should.be.html({
            "img[src=\"../test.png\"]": {
                count: 1
            }
        });
    });
});