fix: improve test coverage and fix flakey test

Author: taeold

src/functions_framework/aio/__init__.py

@@ -229,7 +229,7 @@ def create_asgi_app_from_module(target, source, signature_type, source_module, s
         A Starlette ASGI application instance
     """
     enable_id_logging = _enable_execution_id_logging()
-    if enable_id_logging:
+    if enable_id_logging:  # pragma: no cover
         _configure_app_execution_id_logging()
 
     function = _function_registry.get_user_function(source, source_module, target)

tests/test_cli.py

@@ -29,7 +29,7 @@
 from functions_framework._cli import _cli
 
 
-@pytest.fixture
+@pytest.fixture(autouse=True)
 def clean_registries():
     """Clean up both REGISTRY_MAP and ASGI_FUNCTIONS registries."""
     original_registry_map = _function_registry.REGISTRY_MAP.copy()
@@ -145,8 +145,12 @@ def test_asgi_cli(monkeypatch):
     assert asgi_server.run.calls == [pretend.call("0.0.0.0", 8080)]
 
 
-def test_cli_auto_detects_asgi_decorator(clean_registries):
+@pytest.mark.parametrize("log_execution_id", [None, "true"])
+def test_cli_auto_detects_asgi_decorator(monkeypatch, log_execution_id):
     """Test that CLI auto-detects @aio decorated functions without --asgi flag."""
+    if log_execution_id:
+        monkeypatch.setenv("LOG_EXECUTION_ID", log_execution_id)
+    
     # Use the actual async_decorator.py test file which has @aio.http decorated functions
     test_functions_dir = pathlib.Path(__file__).parent / "test_functions" / "decorators"
     source = test_functions_dir / "async_decorator.py"

tests/test_decorator_functions.py

@@ -38,7 +38,7 @@
 TEST_FUNCTIONS_DIR = pathlib.Path(__file__).resolve().parent / "test_functions"
 
 
-@pytest.fixture
+@pytest.fixture(autouse=True)
 def clean_registries():
     """Clean up both REGISTRY_MAP and ASGI_FUNCTIONS registries."""
     original_registry_map = registry.REGISTRY_MAP.copy()
@@ -147,7 +147,7 @@ def test_aio_http_dict_response():
     assert resp.json() == {"message": "hello", "count": 42, "success": True}
 
 
-def test_aio_decorators_register_asgi_functions(clean_registries):
+def test_aio_decorators_register_asgi_functions():
     """Test that @aio decorators add function names to ASGI_FUNCTIONS registry."""
     from functions_framework.aio import cloud_event, http
 

tests/test_function_registry.py

@@ -13,9 +13,25 @@
 # limitations under the License.
 import os
 
+import pytest
+
 from functions_framework import _function_registry
 
 
+@pytest.fixture(autouse=True)
+def clean_registries():
+    """Clean up both REGISTRY_MAP and ASGI_FUNCTIONS registries."""
+    original_registry_map = _function_registry.REGISTRY_MAP.copy()
+    original_asgi = _function_registry.ASGI_FUNCTIONS.copy()
+    _function_registry.REGISTRY_MAP.clear()
+    _function_registry.ASGI_FUNCTIONS.clear()
+    yield
+    _function_registry.REGISTRY_MAP.clear()
+    _function_registry.REGISTRY_MAP.update(original_registry_map)
+    _function_registry.ASGI_FUNCTIONS.clear()
+    _function_registry.ASGI_FUNCTIONS.update(original_asgi)
+
+
 def test_get_function_signature():
     test_cases = [
         {