Auto close fetcher session

ComplianceAsCode · Aug 31, 2020 · 8849812 · 8849812
1 parent e256d2f
commit 8849812
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/CHANGES.md b/CHANGES.md
@@ -1,3 +1,7 @@
+# UNRELEASED
+
+- [FIXED] ComplianceFetcher session object is auto-closed now in tearDownClass.
+
 # 1.2.7
 
 - [CHANGED] Removed PyYAML dependency to resolve downstream dependency issues.

diff --git a/compliance/fetch.py b/compliance/fetch.py
@@ -28,7 +28,11 @@
 class ComplianceFetcher(unittest.TestCase):
     """Compliance fetcher automation TestCase class."""
 
-    _multiprocess_can_split_ = True
+    @classmethod
+    def tearDownClass(cls):
+        """Perform clean up."""
+        if hasattr(cls, '_session'):
+            cls._session.close()
 
     @classmethod
     def session(cls, url=None, creds=None, **headers):