fix: clean up HTTP session and pool during tear down phase

googleapis · Apr 5, 2022 · 9bb80e6 · 9bb80e6
1 parent dc040f5
commit 9bb80e6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/google/auth/transport/requests.py b/google/auth/transport/requests.py
@@ -149,6 +149,10 @@ def __init__(self, session=None):
 
         self.session = session
 
+    def __del__(self):
+        if hasattr(self, 'session') and self.session is not None:
+            self.session.close()
+
     def __call__(
         self,
         url,

diff --git a/google/auth/transport/urllib3.py b/google/auth/transport/urllib3.py
@@ -427,6 +427,10 @@ def __exit__(self, exc_type, exc_val, exc_tb):
         """Proxy to ``self.http``."""
         return self.http.__exit__(exc_type, exc_val, exc_tb)
 
+    def __del__(self):
+        if hasattr(self, 'http') and self.http is not None:
+            self.http.clear()
+
     @property
     def headers(self):
         """Proxy to ``self.http``."""