From 5595da0f26e45d15d6b15d1b158eb20971d187ed Mon Sep 17 00:00:00 2001
From: Parth Verma <parth.verma@postman.com>
Date: Tue, 13 Aug 2024 17:45:03 -0700
Subject: [PATCH] Added tests for http-2 specific error handling

---
 tests/test-errors-http2-auto-specific.js | 73 ++++++++++++++++++++++++
 tests/test-errors-http2-specific.js      | 73 ++++++++++++++++++++++++
 2 files changed, 146 insertions(+)
 create mode 100644 tests/test-errors-http2-auto-specific.js
 create mode 100644 tests/test-errors-http2-specific.js

diff --git a/tests/test-errors-http2-auto-specific.js b/tests/test-errors-http2-auto-specific.js
new file mode 100644
index 000000000..4ace82797
--- /dev/null
+++ b/tests/test-errors-http2-auto-specific.js
@@ -0,0 +1,73 @@
+'use strict'
+
+var request = require('../index').defaults({ protocolVersion: 'auto' }) 
+var tape = require('tape')
+var server = require('./server')
+var destroyable = require('server-destroy')
+
+const s = server.createHttp2Server();
+
+destroyable(s)
+
+tape('setup', function (t) {
+  s.listen(0, function () {
+    t.end()
+  })
+});
+
+function addTest (errorCode, data = {}) {
+  tape('test ' + errorCode, function (t) {
+    s.on('/' + errorCode, function (req, res) {
+      if (errorCode === 0) {
+        res.end();
+        return;
+      }
+      res.stream.close(errorCode);
+    })
+    data.uri = s.url + '/' + errorCode
+    request(
+      { ...data, strictSSL: false, protocolVersion: 'auto' },
+      function (err) {
+        if (errorCode === 0){
+          t.equal(err, null)
+          t.end()
+          return
+        }
+        if(errorCode === 8){
+          t.equal(err.message, `HTTP/2 Stream closed with error code NGHTTP2_CANCEL`)
+          t.end()
+          return
+        }
+        t.equal(err.message, `Stream closed with error code ${errorCodes[errorCode]}`)
+        t.end()
+      }
+    )
+  })
+}
+
+const errorCodes = [
+  "NGHTTP2_NO_ERROR",
+  "NGHTTP2_PROTOCOL_ERROR",
+  "NGHTTP2_INTERNAL_ERROR",
+  "NGHTTP2_FLOW_CONTROL_ERROR",
+  "NGHTTP2_SETTINGS_TIMEOUT",
+  "NGHTTP2_STREAM_CLOSED",
+  "NGHTTP2_FRAME_SIZE_ERROR",
+  "NGHTTP2_REFUSED_STREAM",
+  "NGHTTP2_CANCEL",
+  "NGHTTP2_COMPRESSION_ERROR",
+  "NGHTTP2_CONNECT_ERROR",
+  "NGHTTP2_ENHANCE_YOUR_CALM",
+  "NGHTTP2_INADEQUATE_SECURITY",
+  "NGHTTP2_HTTP_1_1_REQUIRED"
+]
+
+for(let i = 0; i < errorCodes.length; i++){
+  addTest(i);
+}
+
+tape('cleanup', function(t){
+  s.destroy(function(){
+    t.end()
+  })
+})
\ No newline at end of file
diff --git a/tests/test-errors-http2-specific.js b/tests/test-errors-http2-specific.js
new file mode 100644
index 000000000..52f545789
--- /dev/null
+++ b/tests/test-errors-http2-specific.js
@@ -0,0 +1,73 @@
+'use strict'
+
+var request = require('../index').defaults({ protocolVersion: 'http2' }) 
+var tape = require('tape')
+var server = require('./server')
+var destroyable = require('server-destroy')
+
+const s = server.createHttp2Server();
+
+destroyable(s)
+
+tape('setup', function (t) {
+  s.listen(0, function () {
+    t.end()
+  })
+});
+
+function addTest (errorCode, data = {}) {
+  tape('test ' + errorCode, function (t) {
+    s.on('/' + errorCode, function (req, res) {
+      if (errorCode === 0) {
+        res.end();
+        return;
+      }
+      res.stream.close(errorCode);
+    })
+    data.uri = s.url + '/' + errorCode
+    request(
+      { ...data, strictSSL: false, protocolVersion: 'http2' },
+      function (err, resp, body) {
+        if (errorCode === 0){
+          t.equal(err, null)
+          t.end()
+          return
+        }
+        if(errorCode === 8){
+          t.equal(err.message, `HTTP/2 Stream closed with error code NGHTTP2_CANCEL`)
+          t.end()
+          return
+        }
+        t.equal(err.message, `Stream closed with error code ${errorCodes[errorCode]}`)
+        t.end()
+      }
+    )
+  })
+}
+
+const errorCodes = [
+  "NGHTTP2_NO_ERROR",
+  "NGHTTP2_PROTOCOL_ERROR",
+  "NGHTTP2_INTERNAL_ERROR",
+  "NGHTTP2_FLOW_CONTROL_ERROR",
+  "NGHTTP2_SETTINGS_TIMEOUT",
+  "NGHTTP2_STREAM_CLOSED",
+  "NGHTTP2_FRAME_SIZE_ERROR",
+  "NGHTTP2_REFUSED_STREAM",
+  "NGHTTP2_CANCEL",
+  "NGHTTP2_COMPRESSION_ERROR",
+  "NGHTTP2_CONNECT_ERROR",
+  "NGHTTP2_ENHANCE_YOUR_CALM",
+  "NGHTTP2_INADEQUATE_SECURITY",
+  "NGHTTP2_HTTP_1_1_REQUIRED"
+]
+
+for(let i = 0; i < errorCodes.length; i++){
+  addTest(i);
+}
+
+tape('cleanup', function(t){
+  s.destroy(function(){
+    t.end()
+  })
+})
\ No newline at end of file