get new headers

Author: KazuCocoa

appium/webdriver/appium_connection.py

@@ -28,6 +28,19 @@
 _HEADER_IDEMOTENCY_KEY = 'X-Idempotency-Key'
 
 
+def _get_new_headers(key: str, headers: dict[str, str]) -> dict[str, str]:
+    """Return a new dictionary of heafers without the given key.
+    The key match is case-insensitive."""
+    new_headers = dict()
+
+    key = key.lower()
+    for k, v in headers.items():
+        if k.lower() == key:
+            continue
+        new_headers[k] = v
+    return new_headers
+
+
 class AppiumConnection(RemoteConnection):
     """
     A subclass of selenium.webdriver.remote.remote_connection.Remoteconnection.
@@ -51,7 +64,7 @@ def get_remote_connection_headers(cls, parsed_url: 'ParseResult', keep_alive: bo
         if parsed_url.path.endswith('/session'):
             # https://github.com/appium/appium-base-driver/pull/400
             cls.extra_headers[_HEADER_IDEMOTENCY_KEY] = str(uuid.uuid4())
-        elif _HEADER_IDEMOTENCY_KEY in cls.extra_headers:
-            del cls.extra_headers[_HEADER_IDEMOTENCY_KEY]
+        else:
+            cls.extra_headers = _get_new_headers(_HEADER_IDEMOTENCY_KEY, cls.extra_headers)
 
         return {**super().get_remote_connection_headers(parsed_url, keep_alive=keep_alive), **cls.extra_headers}