chore: clean up thread container

Author: karenc-bq

tests/integration/container/test_aurora_failover.py

@@ -14,13 +14,15 @@
 
 from __future__ import annotations
 
+import gc
 from time import sleep
 from typing import TYPE_CHECKING, List
 
 import pytest
 
 from aws_advanced_python_wrapper.errors import (
     FailoverSuccessError, TransactionResolutionUnknownError)
+from aws_advanced_python_wrapper.host_monitoring_plugin import MonitoringThreadContainer
 from aws_advanced_python_wrapper.utils.properties import (Properties,
                                                           WrapperProperties)
 from .utils.conditions import (disable_on_features, enable_on_deployments,
@@ -56,6 +58,8 @@ def setup_method(self, request):
         self.logger.info(f"Starting test: {request.node.name}")
         yield
         self.logger.info(f"Ending test: {request.node.name}")
+        MonitoringThreadContainer.clean_up()
+        gc.collect()
 
     @pytest.fixture(scope='class')
     def aurora_utility(self):

tests/integration/container/test_read_write_splitting.py

@@ -24,6 +24,7 @@
     AwsWrapperError, FailoverFailedError, FailoverSuccessError,
     ReadWriteSplittingError, TransactionResolutionUnknownError)
 from aws_advanced_python_wrapper.host_list_provider import RdsHostListProvider
+from aws_advanced_python_wrapper.host_monitoring_plugin import MonitoringThreadContainer
 from aws_advanced_python_wrapper.sql_alchemy_connection_provider import \
     SqlAlchemyPooledConnectionProvider
 from aws_advanced_python_wrapper.utils.log import Logger
@@ -60,7 +61,10 @@ def setup_method(self, request):
         self.logger.info(f"Starting test: {request.node.name}")
         yield
         self.logger.info(f"Ending test: {request.node.name}")
-
+        
+        MonitoringThreadContainer.clean_up()
+        gc.collect()
+        
     @pytest.fixture(scope='class')
     def rds_utils(self):
         region: str = TestEnvironment.get_current().get_info().get_region()