test: improve coverage by removing unnecessary pragma no cover annotations

Author: taeold

src/functions_framework/execution_id.py

@@ -54,7 +54,7 @@ def _get_current_context():
     context = execution_context_var.get()
     if context is not None:
         return context
-    return (  # pragma: no cover
+    return (
         flask.g.execution_id_context
         if flask.has_request_context() and "execution_id_context" in flask.g
         else None
@@ -188,7 +188,7 @@ async def async_wrapper(request, *args, **kwargs):
                 if inspect.iscoroutinefunction(view_function):
                     result = await view_function(request, *args, **kwargs)
                 else:
-                    result = view_function(request, *args, **kwargs)  # pragma: no cover
+                    result = view_function(request, *args, **kwargs)
 
                 # Only reset context on successful completion
                 # On exception, leave context available for exception handlers

tests/test_execution_id_async.py

@@ -284,3 +284,22 @@ def sync_func(request):
     result = wrapped(request)
 
     assert result == {"status": "ok"}
+
+
+def test_sync_function_called_from_async_context():
+    """Test that a sync function works correctly when called from async ASGI app."""
+    source = TEST_FUNCTIONS_DIR / "execution_id" / "async_main.py"
+    target = "sync_function_in_async_context"
+    app = create_asgi_app(target, source)
+    client = TestClient(app)
+    resp = client.post(
+        "/",
+        headers={
+            "Function-Execution-Id": TEST_EXECUTION_ID,
+            "Content-Type": "application/json",
+        },
+    )
+
+    result = resp.json()
+    assert result["execution_id"] == TEST_EXECUTION_ID
+    assert result["type"] == "sync"

tests/test_functions/execution_id/async_main.py

@@ -42,3 +42,9 @@ async def async_trace_test(request):
         "execution_id": context.execution_id if context else None,
         "span_id": context.span_id if context else None,
     }
+
+
+# Sync function to test the sync branch in async decorator
+def sync_function_in_async_context(request):
+    """A sync function that can be called from async context."""
+    return {"execution_id": request.headers.get("Function-Execution-Id"), "type": "sync"}