155 lines
3.8 KiB
JavaScript
155 lines
3.8 KiB
JavaScript
const fs = require('fs')
|
|
const path = require('path')
|
|
|
|
const test = require('tape')
|
|
const async = require('async')
|
|
const clone = require('clone')
|
|
const rimraf = require('rimraf')
|
|
|
|
const mermaid = require('../lib')
|
|
|
|
const fileTestMermaid = path.join('test', 'fixtures', 'test.mermaid')
|
|
const isWin = /^win/.test(process.platform)
|
|
let phantomCmd
|
|
if (isWin) {
|
|
phantomCmd = 'node_modules/.bin/phantomjs.cmd'
|
|
} else {
|
|
phantomCmd = 'node_modules/.bin/phantomjs'
|
|
}
|
|
const singleFile = {
|
|
files: [fileTestMermaid],
|
|
outputDir: path.join(process.cwd(), 'test/tmp_single'),
|
|
phantomPath: path.join(process.cwd(), phantomCmd),
|
|
width: 1200,
|
|
css: path.join(__dirname, '..', 'dist', 'mermaid.css'),
|
|
sequenceConfig: null,
|
|
ganttConfig: null
|
|
}
|
|
const multiFile = {
|
|
files: [path.join('test', 'fixtures', 'test.mermaid'),
|
|
path.join('test', 'fixtures', 'test2.mermaid'),
|
|
path.join('test', 'fixtures', 'gantt.mermaid'),
|
|
path.join('test', 'fixtures', 'sequence.mermaid')
|
|
],
|
|
outputDir: 'test/tmp_multi',
|
|
phantomPath: path.join(process.cwd(), phantomCmd),
|
|
width: 1200,
|
|
css: path.join(__dirname, '..', 'dist', 'mermaid.css'),
|
|
sequenceConfig: null,
|
|
ganttConfig: null
|
|
}
|
|
|
|
test('output of single png', function (t) {
|
|
t.plan(3)
|
|
|
|
const expected = ['test.mermaid.png']
|
|
|
|
const opt = clone(singleFile)
|
|
opt.outputDir += '_png'
|
|
opt.png = true
|
|
|
|
mermaid.process(opt.files, opt, function (code) {
|
|
t.equal(code, 0, 'has clean exit code')
|
|
|
|
verifyFiles(expected, opt.outputDir, t)
|
|
})
|
|
})
|
|
|
|
test('output of multiple png', function (t) {
|
|
t.plan(3)
|
|
|
|
const expected = ['test.mermaid.png', 'test2.mermaid.png',
|
|
'gantt.mermaid.png', 'sequence.mermaid.png']
|
|
|
|
const opt = clone(multiFile)
|
|
opt.outputDir += '_png'
|
|
opt.png = true
|
|
|
|
mermaid.process(opt.files, opt, function (code) {
|
|
t.equal(code, 0, 'has clean exit code')
|
|
|
|
verifyFiles(expected, opt.outputDir, t)
|
|
})
|
|
})
|
|
|
|
test('output of single svg', function (t) {
|
|
t.plan(3)
|
|
|
|
const expected = ['test.mermaid.svg']
|
|
|
|
const opt = clone(singleFile)
|
|
opt.outputDir += '_svg'
|
|
opt.svg = true
|
|
|
|
mermaid.process(opt.files, opt, function (code) {
|
|
t.equal(code, 0, 'has clean exit code')
|
|
|
|
verifyFiles(expected, opt.outputDir, t)
|
|
})
|
|
})
|
|
|
|
test('output of multiple svg', function (t) {
|
|
t.plan(3)
|
|
|
|
const expected = ['test.mermaid.svg', 'test2.mermaid.svg',
|
|
'gantt.mermaid.svg', 'sequence.mermaid.svg']
|
|
|
|
const opt = clone(multiFile)
|
|
opt.outputDir += '_svg'
|
|
opt.svg = true
|
|
|
|
mermaid.process(opt.files, opt, function (code) {
|
|
t.equal(code, 0, 'has clean exit code')
|
|
|
|
verifyFiles(expected, opt.outputDir, t)
|
|
})
|
|
})
|
|
|
|
test('output including CSS', function (t) {
|
|
t.plan(5)
|
|
|
|
const expected = ['test.mermaid.png']
|
|
const opt = clone(singleFile)
|
|
const opt2 = clone(singleFile)
|
|
|
|
opt.png = true
|
|
opt.outputDir += '_css_png'
|
|
opt2.png = true
|
|
opt2.outputDir += '_css_png'
|
|
|
|
mermaid.process(opt.files, opt, function (code) {
|
|
t.equal(code, 0, 'has clean exit code')
|
|
const filename = path.join(opt.outputDir, path.basename(expected[0]))
|
|
const one = fs.statSync(filename)
|
|
|
|
opt2.css = path.join('test', 'fixtures', 'test.css')
|
|
|
|
mermaid.process(opt2.files, opt2, function (code) {
|
|
t.equal(code, 0, 'has clean exit code')
|
|
const two = fs.statSync(filename)
|
|
t.notEqual(one.size, two.size)
|
|
|
|
verifyFiles(expected, opt.outputDir, t)
|
|
})
|
|
})
|
|
})
|
|
|
|
function verifyFiles (expected, dir, t) {
|
|
async.each(
|
|
expected, function (file, cb) {
|
|
const filename = path.join(dir, path.basename(file))
|
|
fs.stat(filename, function (err, stat) {
|
|
cb(err)
|
|
})
|
|
}, function (err) {
|
|
t.notOk(err, 'all files passed')
|
|
const deleteTmps = true
|
|
const _rimraf = deleteTmps ? rimraf : function (dir, f) { f(0) }
|
|
_rimraf(dir, function (rmerr) {
|
|
t.notOk(rmerr, 'cleaned up')
|
|
t.end()
|
|
})
|
|
}
|
|
)
|
|
}
|