test-http-set-cookies.js

// Copyright Joyent, Inc. and other Node contributors.
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to permit
// persons to whom the Software is furnished to do so, subject to the
// following conditions:
//
// The above copyright notice and this permission notice shall be included
// in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
// USE OR OTHER DEALINGS IN THE SOFTWARE.

'use strict';
require('../common');
const assert = require('assert');
const http2 = require('http2');
const Countdown = require('../common/countdown');

const countdown = new Countdown(2, () => {
  client.close();
  server.close();
});
const server = http2.createServer(function(req, res) {
  if (req.url === '/one') {
    res.writeHead(200, [['set-cookie', 'A'],
                        ['content-type', 'text/plain']]);
    res.end('one\n');
  } else {
    res.writeHead(200, [['set-cookie', 'A'],
                        ['set-cookie', 'B'],
                        ['content-type', 'text/plain']]);
    res.end('two\n');
  }
});
server.listen(0);

let client;
server.on('listening', function() {
  client = http2.connect('http://localhost:' + this.address().port);

  //
  // one set-cookie header
  //

  const req1 = client.request({
    ':path': '/one'
  });
  req1.on('response',(res) => {
    // set-cookie headers are always return in an array.
    // even if there is only one.
    assert.deepStrictEqual(res['set-cookie'], ['A']);
    assert.strictEqual(res['content-type'], 'text/plain');

    req1.on('data', function(chunk) {
      console.log(chunk.toString());
    });

    req1.on('end', function() {
      req1.close();
      countdown.dec();
    });
  });

  // Two set-cookie headers

  const req2 = client.request({
    ':path': '/two'
  });
  req2.on('response',(res) => {
    assert.deepStrictEqual(res['set-cookie'], ['A', 'B']);
    assert.strictEqual(res['content-type'], 'text/plain');

    req2.on('data', function(chunk) {
      console.log(chunk.toString());
    });

    req2.on('end', function() {
      req2.close();
      countdown.dec();
    });
  });
});