Browse Source

pipe-test fixes

v0.7.4-release
Ryan Dahl 14 years ago
parent
commit
81ac0d5088
  1. 2
      lib/stream.js
  2. 68
      test/disabled/pipe-test.js

2
lib/stream.js

@ -23,7 +23,7 @@ Stream.prototype.pipe = function (dest, options) {
* source gets the 'end' event. * source gets the 'end' event.
*/ */
if (!options || options.end === false) { if (!options || options.end !== false) {
source.on("end", function () { source.on("end", function () {
dest.end(); dest.end();
}); });

68
test/disabled/pipe-test.js

@ -2,18 +2,23 @@ var common = require('../common');
var assert = require('assert'); var assert = require('assert');
var http = require('http'); var http = require('http');
var net = require('net'); var net = require('net');
var sys = require('sys');
var webPort = common.PORT
var tcpPort = webPort + 1;
var listenCount = 0; var listenCount = 0;
var gotThanks = false; var gotThanks = false;
var tcpLengthSeen; var tcpLengthSeen = 0;
var bufferSize = 5 * 1024 * 1024;
/* /*
* 5MB of random buffer. * 5MB of random buffer.
*/ */
var buffer = Buffer(1024 * 1024 * 5); var buffer = Buffer(bufferSize);
for (var i = 0; i < buffer.length; i++) { for (var i = 0; i < buffer.length; i++) {
buffer[i] = parseInt(Math.random()*10000); buffer[i] = parseInt(Math.random()*10000) % 256;
} }
@ -24,67 +29,86 @@ var web = http.Server(function (req, res) {
var socket = net.Stream(); var socket = net.Stream();
socket.connect(tcpPort); socket.connect(tcpPort);
socket.on('connect', function () {
console.log('socket connected');
req.pipe(socket);
});
req.pipe(socket);
req.on('end', function () { req.on('end', function () {
res.writeHead(200); res.writeHead(200);
res.write("thanks"); res.write("thanks");
res.end(); res.end();
console.log("response with 'thanks'");
});
req.connection.on('error', function (e) {
console.log("http server-side error: " + e.message);
process.exit(1);
}); });
}); });
var webPort = common.PORT
web.listen(webPort, startClient); web.listen(webPort, startClient);
var tcp = net.Server(function (socket) { var tcp = net.Server(function (s) {
tcp.close(); tcp.close();
console.log("tcp server connection");
var i = 0; var i = 0;
socket.on('data', function (d) { s.on('data', function (d) {
process.stdout.write(".");
tcpLengthSeen += d.length;
for (var j = 0; j < d.length; j++) { for (var j = 0; j < d.length; j++) {
assert.equal(i % 256, d[i]); //assert.equal(buffer[i], d[j]);
i++; i++;
} }
}); });
socket.on('end', function () { s.on('end', function () {
tcpLengthSeen = i; console.log("tcp socket disconnect");
socket.end(); s.end();
});
s.on('error', function (e) {
console.log("tcp server-side error: " + e.message);
process.exit(1);
}); });
}); });
var tcpPort = webPort + 1;
tcp.listen(tcpPort, startClient); tcp.listen(tcpPort, startClient);
function startClient () { function startClient () {
listenCount++; listenCount++;
console.log("listenCount %d" , listenCount); if (listenCount < 2) return;
if (listenCount < 2) {
console.log("dont start client %d" , listenCount);
return;
}
console.log("start client"); console.log("Making request");
var client = http.createClient(common.PORT); var client = http.createClient(common.PORT);
var req = client.request('GET', '/'); var req = client.request('GET', '/', { 'content-length': buffer.length });
req.write(buffer); req.write(buffer);
req.end(); req.end();
req.on('response', function (res) { req.on('response', function (res) {
console.log('Got response');
res.setEncoding('utf8'); res.setEncoding('utf8');
res.on('data', function (s) { res.on('data', function (string) {
assert.equal("thanks", s); assert.equal("thanks", string);
gotThanks = true; gotThanks = true;
}); });
}); });
client.on('error', function (e) {
console.log("http client-side error: " + e.message);
process.exit(2);
});
} }
process.on('exit', function () { process.on('exit', function () {
assert.ok(gotThanks); assert.ok(gotThanks);
assert.equal(1024*1024*5, tcpLengthSeen); assert.equal(bufferSize, tcpLengthSeen);
}); });

Loading…
Cancel
Save