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')); | ||
|  | }); |