From e949f1f611a601e82dca420c26d092fdd3475155 Mon Sep 17 00:00:00 2001
From: James Graham <james@hoppipolla.co.uk>
Date: Fri, 21 Aug 2020 15:14:51 +0000
Subject: [PATCH] Ensure HTTP connections are closed in websocket tests

Differential Revision: https://phabricator.services.mozilla.com/D87494

bugzilla-url: https://bugzilla.mozilla.org/show_bug.cgi?id=1653294
gecko-commit: 29be769939c2507fc75257e5bb09f067447eba47
gecko-integration-branch: autoland
gecko-reviewers: webdriver-reviewers, whimboo
---
 tools/webdriver/webdriver/client.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tools/webdriver/webdriver/client.py b/tools/webdriver/webdriver/client.py
index fe5189ec0b30a4..0f79483826b34b 100644
--- a/tools/webdriver/webdriver/client.py
+++ b/tools/webdriver/webdriver/client.py
@@ -427,6 +427,8 @@ def start(self):
         if self.session_id is not None:
             return
 
+        self.transport.close()
+
         body = {"capabilities": {}}
 
         if self.requested_capabilities is not None:
@@ -452,6 +454,7 @@ def end(self):
             pass
         finally:
             self.session_id = None
+            self.transport.close()
 
     def send_command(self, method, url, body=None, timeout=None):
         """