1 files changed, 17 insertions, 0 deletions
diff --git a/test/mitmproxy/addons/test_disable_h2c_upgrade.py b/test/mitmproxy/addons/test_disable_h2c_upgrade.py
new file mode 100644
index 00000000..6cab713d
--- /dev/null
+++ b/test/mitmproxy/addons/test_disable_h2c_upgrade.py
@@ -0,0 +1,17 @@
+from mitmproxy.addons import disable_h2c_upgrade
+from mitmproxy.test import tflow
+
+
+class TestTermLog:
+    def test_simple(self):
+        a = disable_h2c_upgrade.DisableH2CleartextUpgrade()
+
+        f = tflow.tflow()
+        f.request.headers['upgrade'] = 'h2c'
+        f.request.headers['connection'] = 'foo'
+        f.request.headers['http2-settings'] = 'bar'
+
+        a.request(f)
+        assert 'upgrade' not in f.request.headers
+        assert 'connection' not in f.request.headers
+        assert 'http2-settings' not in f.request.headers