summaryrefslogtreecommitdiffstats
path: root/test/mock.js
blob: 5ff9f63147ec92bcaa87ad4d3315a0a88e1f7924 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
/* eslint-disable no-console */

var Q = require('q');
var _ = require('lodash');
var tmp = require('tmp');
var path = require('path');

var Book = require('../').Book;
var NodeFS = require('../lib/fs/node');
var fs = require('../lib/utils/fs');

require('./assertions');

// Create filesystem instance for testing
var nodeFS = new NodeFS();

function setupFS(files) {
    return Q.nfcall(tmp.dir.bind(tmp)).get(0)
    .then(function(rootFolder) {
        return _.chain(_.pairs(files))
            .sortBy(0)
            .reduce(function(prev, pair) {
                return prev.then(function() {
                    var filename = path.resolve(rootFolder, pair[0]);
                    var buf = pair[1];

                    if (_.isObject(buf)) buf = JSON.stringify(buf);
                    if (_.isString(buf)) buf = new Buffer(buf, 'utf-8');

                    return fs.mkdirp(path.dirname(filename))
                    .then(function() {
                        return fs.writeFile(filename, buf);
                    });
                });
            }, Q())
            .value()
            .then(function() {
                return rootFolder;
            });
    });
}

// Setup a mock book for testing using a map of files
function setupBook(files, opts) {
    opts = opts || {};
    opts.log = function() { };

    return setupFS(files)
    .then(function(folder) {
        opts.fs = nodeFS;
        opts.root = folder;

        return new Book(opts);
    });
}

// Setup a book with default README/SUMMARY
function setupDefaultBook(files, summary, opts) {
    var summaryContent = '# Summary \n\n' +
        _.map(summary, function(article) {
            return '* [' + article.title +'](' + article.path + ')';
        })
        .join('\n');

    return setupBook(_.defaults(files || {}, {
        'README.md': 'Hello',
        'SUMMARY.md': summaryContent
    }), opts);
}

// Output a book with a specific generator
function outputDefaultBook(Output, files, summary, opts) {
    return setupDefaultBook(files, summary, opts)
    .then(function(book) {
        // Parse the book
        return book.parse()

        // Start generation
        .then(function() {
            var output = new Output(book);
            return output.generate()
                .thenResolve(output);
        });
    });
}

// Output a book with a specific generator
function outputBook(Output, files, opts) {
    return setupBook(files, opts)
    .then(function(book) {
        // Parse the book
        return book.parse()

        // Start generation
        .then(function() {
            var output = new Output(book);
            return output.generate()
                .thenResolve(output);
        });
    });
}

// Log an error
function logError(err) {
    console.log(err.stack || err);
}

module.exports = {
    fs: nodeFS,
    setupFS: setupFS,
    setupBook: setupBook,
    outputBook: outputBook,
    setupDefaultBook: setupDefaultBook,
    outputDefaultBook: outputDefaultBook,
    logError: logError
};