registry flaky test

Author: Victoria Hall

.github/Scripts/fwpc-tests.sh

@@ -1,2 +1,2 @@
 #!/usr/bin/env bash
-python -m pytest -q -n auto --dist loadfile --reruns 4 --instafail --cov=./azure_functions_worker --cov-report xml --cov-branch --cov-append tests/endtoend/test_worker_process_count_functions.py tests/endtoend/test_threadpool_thread_count_functions.py
+python -m pytest -q -n 1 --dist loadfile --reruns 4 --instafail --cov=./azure_functions_worker --cov-report xml --cov-branch --cov-append tests/endtoend/test_worker_process_count_functions.py tests/endtoend/test_threadpool_thread_count_functions.py

tests/endtoend/test_threadpool_thread_count_functions.py

@@ -21,6 +21,7 @@ def setUpClass(cls):
 
     def tearDown(self):
         os.environ.pop('PYTHON_THREADPOOL_THREAD_COUNT')
+
         super().tearDown()
 
     @classmethod

tests/endtoend/test_worker_process_count_functions.py

@@ -14,6 +14,9 @@ class TestWorkerProcessCount(testutils.WebHostTestCase):
     """
     @classmethod
     def setUpClass(cls):
+        cls.env_variables['PYTHON_THREADPOOL_THREAD_COUNT'] = '1'
+        cls.env_variables['FUNCTIONS_WORKER_PROCESS_COUNT'] = '2'
+
         os.environ["PYTHON_THREADPOOL_THREAD_COUNT"] = "1"
         os.environ["FUNCTIONS_WORKER_PROCESS_COUNT"] = "2"
 
@@ -22,6 +25,7 @@ def setUpClass(cls):
     def tearDown(self):
         os.environ.pop('PYTHON_THREADPOOL_THREAD_COUNT')
         os.environ.pop('FUNCTIONS_WORKER_PROCESS_COUNT')
+
         super().tearDown()
 
     @classmethod

tests/unittests/test_enable_debug_logging_functions.py

@@ -20,6 +20,7 @@
 """
 
 
+@testutils.retryable_test(4, 5)
 class TestDebugLoggingEnabledFunctions(testutils.WebHostTestCase):
     """
     Tests for cx debug logging enabled case.
@@ -38,7 +39,6 @@ def tearDownClass(cls):
     def get_script_dir(cls):
         return testutils.UNIT_TESTS_FOLDER / 'log_filtering_functions'
 
-    @testutils.retryable_test(4, 5)
     def test_debug_logging_enabled(self):
         """
         Verify when cx debug logging is enabled, cx function debug logs

tests/unittests/test_functions_registry.py

@@ -1,7 +1,7 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License.
 
-import unittest
+from tests.utils import testutils
 
 from azure.functions import Function
 from azure.functions.decorators.blob import BlobInput
@@ -11,17 +11,22 @@
 from azure_functions_worker.functions import FunctionLoadError
 
 
-class TestFunctionsRegistry(unittest.TestCase):
+class TestFunctionsRegistry(testutils.AsyncTestCase):
 
-    def setUp(self) -> None:
+    def setUp(self):
         def dummy():
             return "test"
 
         self.dummy = dummy
         self.func = Function(self.dummy, "test.py")
         self.function_registry = functions.Registry()
 
-    def test_add_indexed_function_invalid_direction(self):
+    async def test_add_indexed_function_invalid_direction(self):
+        # Ensures that azure-functions is loaded and BINDING_REGISTRY
+        # is not None
+        async with testutils.start_mockhost() as host:
+            await host.init_worker()
+
         trigger1 = HttpTrigger(name="req1", route="test")
         binding = BlobInput(name="$return", path="testpath",
                             connection="testconnection")