Fix a memory leak in CDP Mode

Author: mdmintz

seleniumbase/plugins/sb_manager.py

@@ -1367,6 +1367,12 @@ def SB(
                     "%s%s%s%s%s"
                     % (c1, left_space, end_text, right_space, cr)
                 )
+        if hasattr(sb_config, "_cdp_aclose"):
+            import asyncio
+            with suppress(Exception):
+                loop = asyncio.get_event_loop()
+                asyncio.set_event_loop(loop)
+                loop.run_until_complete(sb_config._cdp_aclose())
         gc.collect()
     if test and test_name and not test_passed and raise_test_failure:
         raise exception

seleniumbase/undetected/cdp_driver/connection.py

@@ -19,6 +19,7 @@
 )
 import websockets
 from websockets.protocol import State
+from seleniumbase import config as sb_config
 from . import cdp_util as util
 import mycdp as cdp
 import mycdp.network
@@ -270,6 +271,7 @@ async def aopen(self, **kw):
                     max_size=MAX_SIZE,
                 )
                 self.listener = Listener(self)
+                sb_config._cdp_aclose = self.aclose
             except (Exception,) as e:
                 logger.debug("Exception during opening of websocket: %s", e)
                 if self.listener:
@@ -415,6 +417,7 @@ async def send(
             if not _is_update:
                 await self._register_handlers()
             await self.websocket.send(tx.message)
+            sb_config._cdp_aclose = self.aclose
             try:
                 return await tx
             except ProtocolException as e: