mirror of https://gitee.com/openkylin/nodejs.git
41 lines
902 B
JavaScript
41 lines
902 B
JavaScript
'use strict';
|
|
|
|
const common = require('../common');
|
|
if (!common.hasCrypto)
|
|
common.skip('missing crypto');
|
|
const assert = require('assert');
|
|
const h2 = require('http2');
|
|
|
|
class MyServerRequest extends h2.Http2ServerRequest {
|
|
getUserAgent() {
|
|
return this.headers['user-agent'] || 'unknown';
|
|
}
|
|
}
|
|
|
|
const server = h2.createServer({
|
|
Http2ServerRequest: MyServerRequest
|
|
}, (req, res) => {
|
|
assert.strictEqual(req.getUserAgent(), 'node-test');
|
|
|
|
res.writeHead(200, { 'Content-Type': 'text/plain' });
|
|
res.end();
|
|
});
|
|
server.listen(0);
|
|
|
|
server.on('listening', common.mustCall(() => {
|
|
|
|
const client = h2.connect(`http://localhost:${server.address().port}`);
|
|
const req = client.request({
|
|
':path': '/',
|
|
'User-Agent': 'node-test'
|
|
});
|
|
|
|
req.on('response', common.mustCall());
|
|
|
|
req.resume();
|
|
req.on('end', common.mustCall(() => {
|
|
server.close();
|
|
client.destroy();
|
|
}));
|
|
}));
|