|
|
var fs = require('graceful-fs') var test = require('tap').test var path = require('path') var writeStream = require('../index.js')
var rename = fs.rename fs.rename = function (from, to, cb) { setTimeout(function () { rename(from, to, cb) }, 100) }
test('basic', function (t) { // open 10 write streams to the same file.
// then write to each of them, and to the target
// and verify at the end that each of them does their thing
var target = path.resolve(__dirname, 'test.txt') var n = 10
// We run all of our assertions twice:
// once for finish, once for close
// There are 6 assertions, two fixed, plus 4 lines in the file.
t.plan(n * 2 * 6)
var streams = [] for (var i = 0; i < n; i++) { var s = writeStream(target) s.on('finish', verifier('finish', i)) s.on('close', verifier('close', i)) streams.push(s) }
function verifier (ev, num) { return function () { if (ev === 'close') { t.equal(this.__emittedFinish, true, num + '. closed only after finish') } else { this.__emittedFinish = true t.equal(ev, 'finish', num + '. finished') }
// make sure that one of the atomic streams won.
var res = fs.readFileSync(target, 'utf8') var lines = res.trim().split(/\n/) lines.forEach(function (line, lineno) { var first = lines[0].match(/\d+$/)[0] var cur = line.match(/\d+$/)[0] t.equal(cur, first, num + '. line ' + lineno + ' matches') })
var resExpr = /^first write \d+\nsecond write \d+\nthird write \d+\nfinal write \d+\n$/ t.similar(res, resExpr, num + '. content matches') } }
// now write something to each stream.
streams.forEach(function (stream, i) { stream.write('first write ' + i + '\n') })
// wait a sec for those writes to go out.
setTimeout(function () { // write something else to the target.
fs.writeFileSync(target, 'brutality!\n')
// write some more stuff.
streams.forEach(function (stream, i) { stream.write('second write ' + i + '\n') })
setTimeout(function () { // Oops! Deleted the file!
fs.unlinkSync(target)
// write some more stuff.
streams.forEach(function (stream, i) { stream.write('third write ' + i + '\n') })
setTimeout(function () { fs.writeFileSync(target, 'brutality TWO!\n') streams.forEach(function (stream, i) { stream.end('final write ' + i + '\n') }) }, 50) }, 50) }, 50) })
test('cleanup', function (t) { fs.readdirSync(__dirname).filter(function (f) { return f.match(/^test.txt/) }).forEach(function (file) { fs.unlinkSync(path.resolve(__dirname, file)) }) t.end() })
|