1
0
Fork 0
management/front/dkha-web-sz-main/node_modules/fs-write-stream-atomic/test/basic.js

98 lines
2.7 KiB
JavaScript
Raw Normal View History

2023-12-18 13:12:25 +08:00
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()
})