From 9a5cf7bf443e6fdd1e310a3d58a86128f3b4beb7 Mon Sep 17 00:00:00 2001 From: Gavin Aguiar Date: Wed, 3 Apr 2024 15:17:24 -0500 Subject: [PATCH] Fixed wpc test --- tests/endtoend/test_worker_process_count_functions.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tests/endtoend/test_worker_process_count_functions.py b/tests/endtoend/test_worker_process_count_functions.py index 519d8982..bb2a1e7a 100644 --- a/tests/endtoend/test_worker_process_count_functions.py +++ b/tests/endtoend/test_worker_process_count_functions.py @@ -25,9 +25,12 @@ class TestWorkerProcessCount(testutils.WebHostTestCase): cls._patch_environ.start() super().setUpClass() - def tearDown(self): - super().tearDown() - self._patch_environ.stop() + @classmethod + def tearDownClass(cls): + # Remove the PYTHON_SCRIPT_FILE_NAME environment variable + os.environ.pop('PYTHON_THREADPOOL_THREAD_COUNT') + os.environ.pop('FUNCTIONS_WORKER_PROCESS_COUNT') + super().tearDownClass() @classmethod def get_script_dir(cls):