teardownclass

Author: hallvictoria

tests/endtoend/test_http_functions.py

@@ -229,6 +229,22 @@ def get_environment_variables(cls):
 @unittest.skipIf(sys.version_info.minor <= 7, "Skipping tests <= Python 3.7")
 class TestHttpFunctionsV2FastApiWithInitIndexing(
         TestHttpFunctionsWithInitIndexing):
+
+    @classmethod
+    def setUpClass(cls):
+        cls.env_variables[PYTHON_ENABLE_INIT_INDEXING] = '1'
+        os.environ[PYTHON_ENABLE_INIT_INDEXING] = "1"
+        super().setUpClass()
+
+    @classmethod
+    def tearDownClass(cls):
+        os.environ.pop(PYTHON_ENABLE_INIT_INDEXING)
+        super().tearDownClass()
+
+    @classmethod
+    def get_environment_variables(cls):
+        return cls.env_variables
+
     @classmethod
     def get_script_dir(cls):
         return testutils.E2E_TESTS_FOLDER / 'http_functions' / \

tests/endtoend/test_worker_process_count_functions.py

@@ -22,7 +22,8 @@ def setUpClass(cls):
 
         super().setUpClass()
 
-    def tearDown(self):
+    @classmethod
+    def tearDownClass(cls):
         os.environ.pop('PYTHON_THREADPOOL_THREAD_COUNT')
         os.environ.pop('FUNCTIONS_WORKER_PROCESS_COUNT')