1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
// Verify that the 'upgrade' header causes an 'upgrade' event to be emitted to
// the HTTP client. This test uses a raw TCP server to better control server
// behavior.
require('../common');
var http = require('http');
var net = require('net');
var sys = require('sys');
var PORT = 5000 + Math.floor(Math.random() * 1000);
// Parse a string of data, returning an object if headers are complete, and
// undefined otherwise
var parseHeaders = function(data) {
var m = data.search(/\r\n\r\n/);
if (!m) {
return;
}
var o = {};
data.substring(0, m.index).split('\r\n').forEach(function(h) {
var foo = h.split(':');
if (foo.length < 2) {
return;
}
o[foo[0].trim().toLowerCase()] = foo[1].trim().toLowerCase();
});
return o;
};
// Create a TCP server
var srv = net.createServer(function(c) {
var data = '';
c.addListener('data', function(d) {
data += d.toString('utf8');
// We found the end of the headers; make sure that we have an 'upgrade'
// header and send back a response
var headers = parseHeaders(data);
if (!headers) {
return;
}
assert.ok('upgrade' in headers);
c.write('HTTP/1.1 101\r\n');
c.write('connection: upgrade\r\n');
c.write('upgrade: ' + headers.upgrade + '\r\n');
c.write('\r\n');
c.write('nurtzo');
c.end();
});
});
srv.listen(PORT, '127.0.0.1');
var gotUpgrade = false;
var hc = http.createClient(PORT, '127.0.0.1');
hc.addListener('upgrade', function(req, socket, upgradeHead) {
// XXX: This test isn't fantastic, as it assumes that the entire response
// from the server will arrive in a single data callback
assert.equal(upgradeHead, 'nurtzo');
socket.end();
srv.close();
gotUpgrade = true;
});
hc.request('/', {
'Connection' : 'Upgrade',
'Upgrade' : 'WebSocket'
}).end();
process.addListener('exit', function() {
assert.ok(gotUpgrade);
});
|
