From 1c552da7a4564815b4ded3948a0fb60a8ac4ad53 Mon Sep 17 00:00:00 2001 From: Vito Sansevero Date: Sat, 17 Oct 2026 16:29:30 -0700 Subject: [PATCH] test(history): support Python 3.10 package fixture cleanup (#209) --- tests/test_extension_web_dir.py | 5 ++++- tests/test_init_package.py | 8 ++++++-- 2 files changed, 10 insertions(+), 3 deletions(-) diff --git a/tests/test_extension_web_dir.py b/tests/test_extension_web_dir.py index dc21e7a..018a1cd 100644 --- a/tests/test_extension_web_dir.py +++ b/tests/test_extension_web_dir.py @@ -4,6 +4,7 @@ ComfyUI serves every .js file under WEB_DIRECTORY into the canvas page, so the directory must hold nothing but the node extension. """ +import contextlib import os import sys import tempfile @@ -26,7 +27,9 @@ class TestExtensionWebDir(unittest.TestCase): cls.addClassCleanup(cls.tmp.cleanup) output_dir = os.path.join(cls.tmp.name, "output") os.makedirs(output_dir) - cls.package, _ = cls.enterClassContext( + package_context = contextlib.ExitStack() + cls.addClassCleanup(package_context.close) + cls.package, _ = package_context.enter_context( load_package( cls.package_name, os.path.join(cls.tmp.name, "prompts.db"), output_dir ) diff --git a/tests/test_init_package.py b/tests/test_init_package.py index f5b7312..df95d4e 100644 --- a/tests/test_init_package.py +++ b/tests/test_init_package.py @@ -191,7 +191,9 @@ class PackageTestCase(unittest.TestCase): cls.output_dir = os.path.join(cls.tmp.name, "output") os.makedirs(cls.output_dir) cls.db_path = cls.make_db_path(cls.tmp.name) - cls.package, cls.server = cls.enterClassContext( + package_context = contextlib.ExitStack() + cls.addClassCleanup(package_context.close) + cls.package, cls.server = package_context.enter_context( load_package( cls.package_name, cls.db_path, cls.output_dir, server=cls.make_server() ) @@ -350,8 +352,10 @@ class TestConsoleBanner(unittest.TestCase): name = "prompt_manager_pkg_cp1252" stream = io.TextIOWrapper(io.BytesIO(), encoding="cp1252", errors="strict") + package_context = contextlib.ExitStack() + self.addCleanup(package_context.close) with contextlib.redirect_stdout(stream): - package, _ = self.enterContext( + package, _ = package_context.enter_context( load_package(name, os.path.join(tmp.name, "p.db"), output_dir) ) self.addCleanup(sys.modules.pop, name, None)