You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
148 lines
4.2 KiB
148 lines
4.2 KiB
3 years ago
|
var Dicer = require('..');
|
||
|
var assert = require('assert'),
|
||
|
fs = require('fs'),
|
||
|
path = require('path'),
|
||
|
inspect = require('util').inspect;
|
||
|
|
||
|
var FIXTURES_ROOT = __dirname + '/fixtures/';
|
||
|
|
||
|
var t = 0,
|
||
|
group = path.basename(__filename, '.js') + '/';
|
||
|
|
||
|
var tests = [
|
||
|
{ source: 'many',
|
||
|
opts: { boundary: '----WebKitFormBoundaryWLHCs9qmcJJoyjKR' },
|
||
|
chsize: 16,
|
||
|
nparts: 7,
|
||
|
what: 'Extra trailer data pushed after finished'
|
||
|
},
|
||
|
];
|
||
|
|
||
|
function next() {
|
||
|
if (t === tests.length)
|
||
|
return;
|
||
|
var v = tests[t],
|
||
|
fixtureBase = FIXTURES_ROOT + v.source,
|
||
|
fd,
|
||
|
n = 0,
|
||
|
buffer = new Buffer(v.chsize),
|
||
|
state = { parts: [] };
|
||
|
|
||
|
fd = fs.openSync(fixtureBase + '/original', 'r');
|
||
|
|
||
|
var dicer = new Dicer(v.opts),
|
||
|
error,
|
||
|
partErrors = 0,
|
||
|
finishes = 0;
|
||
|
|
||
|
dicer.on('part', function(p) {
|
||
|
var part = {
|
||
|
body: undefined,
|
||
|
bodylen: 0,
|
||
|
error: undefined,
|
||
|
header: undefined
|
||
|
};
|
||
|
|
||
|
p.on('header', function(h) {
|
||
|
part.header = h;
|
||
|
}).on('data', function(data) {
|
||
|
// make a copy because we are using readSync which re-uses a buffer ...
|
||
|
var copy = new Buffer(data.length);
|
||
|
data.copy(copy);
|
||
|
data = copy;
|
||
|
if (!part.body)
|
||
|
part.body = [ data ];
|
||
|
else
|
||
|
part.body.push(data);
|
||
|
part.bodylen += data.length;
|
||
|
}).on('error', function(err) {
|
||
|
part.error = err;
|
||
|
++partErrors;
|
||
|
}).on('end', function() {
|
||
|
if (part.body)
|
||
|
part.body = Buffer.concat(part.body, part.bodylen);
|
||
|
state.parts.push(part);
|
||
|
});
|
||
|
}).on('error', function(err) {
|
||
|
error = err;
|
||
|
}).on('finish', function() {
|
||
|
assert(finishes++ === 0, makeMsg(v.what, 'finish emitted multiple times'));
|
||
|
|
||
|
if (v.dicerError)
|
||
|
assert(error !== undefined, makeMsg(v.what, 'Expected error'));
|
||
|
else
|
||
|
assert(error === undefined, makeMsg(v.what, 'Unexpected error'));
|
||
|
|
||
|
if (v.events && v.events.indexOf('part') > -1) {
|
||
|
assert.equal(state.parts.length,
|
||
|
v.nparts,
|
||
|
makeMsg(v.what,
|
||
|
'Part count mismatch:\nActual: '
|
||
|
+ state.parts.length
|
||
|
+ '\nExpected: '
|
||
|
+ v.nparts));
|
||
|
|
||
|
if (!v.npartErrors)
|
||
|
v.npartErrors = 0;
|
||
|
assert.equal(partErrors,
|
||
|
v.npartErrors,
|
||
|
makeMsg(v.what,
|
||
|
'Part errors mismatch:\nActual: '
|
||
|
+ partErrors
|
||
|
+ '\nExpected: '
|
||
|
+ v.npartErrors));
|
||
|
|
||
|
for (var i = 0, header, body; i < v.nparts; ++i) {
|
||
|
if (fs.existsSync(fixtureBase + '/part' + (i+1))) {
|
||
|
body = fs.readFileSync(fixtureBase + '/part' + (i+1));
|
||
|
if (body.length === 0)
|
||
|
body = undefined;
|
||
|
} else
|
||
|
body = undefined;
|
||
|
assert.deepEqual(state.parts[i].body,
|
||
|
body,
|
||
|
makeMsg(v.what,
|
||
|
'Part #' + (i+1) + ' body mismatch'));
|
||
|
if (fs.existsSync(fixtureBase + '/part' + (i+1) + '.header')) {
|
||
|
header = fs.readFileSync(fixtureBase
|
||
|
+ '/part' + (i+1) + '.header', 'binary');
|
||
|
header = JSON.parse(header);
|
||
|
} else
|
||
|
header = undefined;
|
||
|
assert.deepEqual(state.parts[i].header,
|
||
|
header,
|
||
|
makeMsg(v.what,
|
||
|
'Part #' + (i+1)
|
||
|
+ ' parsed header mismatch:\nActual: '
|
||
|
+ inspect(state.parts[i].header)
|
||
|
+ '\nExpected: '
|
||
|
+ inspect(header)));
|
||
|
}
|
||
|
}
|
||
|
++t;
|
||
|
next();
|
||
|
});
|
||
|
|
||
|
while (true) {
|
||
|
n = fs.readSync(fd, buffer, 0, buffer.length, null);
|
||
|
if (n === 0) {
|
||
|
setTimeout(function() {
|
||
|
dicer.write('\r\n\r\n\r\n');
|
||
|
dicer.end();
|
||
|
}, 50);
|
||
|
break;
|
||
|
}
|
||
|
dicer.write(n === buffer.length ? buffer : buffer.slice(0, n));
|
||
|
}
|
||
|
fs.closeSync(fd);
|
||
|
}
|
||
|
next();
|
||
|
|
||
|
function makeMsg(what, msg) {
|
||
|
return '[' + group + what + ']: ' + msg;
|
||
|
}
|
||
|
|
||
|
process.on('exit', function() {
|
||
|
assert(t === tests.length,
|
||
|
makeMsg('_exit', 'Only ran ' + t + '/' + tests.length + ' tests'));
|
||
|
});
|