1 files changed, 85 insertions, 0 deletions
diff --git a/test/parallel/test-http2-info-headers.js b/test/parallel/test-http2-info-headers.js
new file mode 100755
index 0000000000..c5d93d514f
--- /dev/null
+++ b/test/parallel/test-http2-info-headers.js
@@ -0,0 +1,85 @@
+// Flags: --expose-http2
+'use strict';
+
+const common = require('../common');
+const assert = require('assert');
+const h2 = require('http2');
+
+const server = h2.createServer();
+
+// we use the lower-level API here
+server.on('stream', common.mustCall(onStream));
+
+const status101regex =
+  /^HTTP status code 101 \(Switching Protocols\) is forbidden in HTTP\/2$/;
+const afterRespondregex =
+  /^Cannot specify additional headers after response initiated$/;
+
+function onStream(stream, headers, flags) {
+
+  assert.throws(() => stream.additionalHeaders({ ':status': 201 }),
+                common.expectsError({
+                  code: 'ERR_HTTP2_INVALID_INFO_STATUS',
+                  type: RangeError,
+                  message: /^Invalid informational status code: 201$/
+                }));
+
+  assert.throws(() => stream.additionalHeaders({ ':status': 101 }),
+                common.expectsError({
+                  code: 'ERR_HTTP2_STATUS_101',
+                  type: Error,
+                  message: status101regex
+                }));
+
+  // Can send more than one
+  stream.additionalHeaders({ ':status': 100 });
+  stream.additionalHeaders({ ':status': 100 });
+
+  stream.respond({
+    'content-type': 'text/html',
+    ':status': 200
+  });
+
+  assert.throws(() => stream.additionalHeaders({ abc: 123 }),
+                common.expectsError({
+                  code: 'ERR_HTTP2_HEADERS_AFTER_RESPOND',
+                  type: Error,
+                  message: afterRespondregex
+                }));
+
+  stream.end('hello world');
+}
+
+server.listen(0);
+
+server.on('listening', common.mustCall(() => {
+
+  const client = h2.connect(`http://localhost:${server.address().port}`);
+
+  const req = client.request({ ':path': '/'});
+
+  // The additionalHeaders method does not exist on client stream
+  assert.strictEqual(req.additionalHeaders, undefined);
+
+  // Additional informational headers
+  req.on('headers', common.mustCall((headers) => {
+    assert.notStrictEqual(headers, undefined);
+    assert.strictEqual(headers[':status'], 100);
+  }, 2));
+
+  // Response headers
+  req.on('response', common.mustCall((headers) => {
+    assert.notStrictEqual(headers, undefined);
+    assert.strictEqual(headers[':status'], 200);
+    assert.strictEqual(headers['content-type'], 'text/html');
+  }));
+
+  req.resume();
+
+  req.on('end', common.mustCall(() => {
+    server.close();
+    client.destroy();
+  }));
+  req.end();
+
+}));