fix: Python 3.13 OTel Initialization

Author: hallvictoria

runtimes/v2/azure_functions_runtime/handle_event.py

@@ -18,7 +18,10 @@
 )
 from .loader import index_function_app, process_indexed_function
 from .logging import logger
-from .otel import otel_manager, configure_opentelemetry, initialize_azure_monitor
+from .otel import (configure_opentelemetry,
+                   initialize_azure_monitor,
+                   otel_manager,
+                   update_opentelemetry_status)
 from .version import VERSION
 
 from .bindings.context import get_context
@@ -86,6 +89,7 @@ async def worker_init_request(request):
 
     if is_envvar_true(PYTHON_ENABLE_OPENTELEMETRY):
         otel_manager.set_otel_libs_available(True)
+        update_opentelemetry_status()
 
     if (otel_manager.get_azure_monitor_available()
             or otel_manager.get_otel_libs_available()):
@@ -337,6 +341,7 @@ async def function_environment_reload_request(request):
         capabilities = {}
         if is_envvar_true(PYTHON_ENABLE_OPENTELEMETRY):
             otel_manager.set_otel_libs_available(True)
+            update_opentelemetry_status()
 
         if is_envvar_true(PYTHON_APPLICATIONINSIGHTS_ENABLE_TELEMETRY):
             initialize_azure_monitor()

runtimes/v2/azure_functions_runtime/otel.py

@@ -107,10 +107,15 @@ def initialize_azure_monitor():
 
 
 def configure_opentelemetry(invocation_context):
+    trace_context_propagator = otel_manager.get_trace_context_propagator()
+    context_api = otel_manager.get_context_api()
+    if trace_context_propagator is None or context_api is None:
+        return
+
     carrier = {TRACEPARENT: invocation_context.trace_context.trace_parent,
                TRACESTATE: invocation_context.trace_context.trace_state}
-    ctx = otel_manager.get_trace_context_propagator().extract(carrier)
-    otel_manager.get_context_api().attach(ctx)
+    ctx = trace_context_propagator.extract(carrier)
+    context_api.attach(ctx)
 
 
 otel_manager = OTelManager()

runtimes/v2/tests/unittests/test_opentelemetry.py

@@ -1,11 +1,13 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
+import asyncio
 import os
 import unittest
 
 import tests.protos as protos
 
 from azure_functions_runtime.handle_event import otel_manager, worker_init_request
-from azure_functions_runtime.otel import (initialize_azure_monitor,
+from azure_functions_runtime.otel import (configure_opentelemetry,
+                                          initialize_azure_monitor,
                                           update_opentelemetry_status,
                                           OTelManager)
 from azure_functions_runtime.logging import logger
@@ -19,6 +21,24 @@
 
 class TestOpenTelemetry(unittest.TestCase):
 
+    def setUp(self):
+        self._otel_manager_state = (
+            otel_manager.get_azure_monitor_available(),
+            otel_manager.get_otel_libs_available(),
+            otel_manager.get_context_api(),
+            otel_manager.get_trace_context_propagator(),
+        )
+
+    def tearDown(self):
+        (azure_monitor_available,
+         otel_libs_available,
+         context_api,
+         trace_context_propagator) = self._otel_manager_state
+        otel_manager.set_azure_monitor_available(azure_monitor_available)
+        otel_manager.set_otel_libs_available(otel_libs_available)
+        otel_manager.set_context_api(context_api)
+        otel_manager.set_trace_context_propagator(trace_context_propagator)
+
     def test_update_opentelemetry_status_import_error(self):
         with patch.dict('sys.modules', {
             'opentelemetry': None,
@@ -43,6 +63,12 @@ def test_update_opentelemetry_status_success(
         self.assertIsNotNone(otel_manager.get_context_api())
         self.assertIsNotNone(otel_manager.get_trace_context_propagator())
 
+    def test_configure_opentelemetry_noops_when_not_initialized(self):
+        otel_manager.set_context_api(None)
+        otel_manager.set_trace_context_propagator(None)
+
+        configure_opentelemetry(MagicMock())
+
     @patch('builtins.__import__')
     @patch("azure_functions_runtime.otel.update_opentelemetry_status")
     def test_initialize_azure_monitor_success(
@@ -146,7 +172,7 @@ async def test_init_request_initialize_azure_monitor_disabled_app_setting(
         self.assertNotIn("WorkerOpenTelemetryEnabled", capabilities)
 
     @patch.dict(os.environ, {'PYTHON_ENABLE_OPENTELEMETRY': 'true'})
-    async def test_init_request_enable_opentelemetry_enabled_app_setting(
+    def test_init_request_enable_opentelemetry_enabled_app_setting(
         self,
     ):
         worker_request = WorkerRequest(name='worker_init_request',
@@ -155,13 +181,15 @@ async def test_init_request_enable_opentelemetry_enabled_app_setting(
                                            FUNCTION_APP_DIRECTORY)),
                                        properties={'host': '123',
                                                    'protos': protos})
-        init_response = await worker_init_request(worker_request)
+        init_response = asyncio.run(worker_init_request(worker_request))
 
         self.assertEqual(init_response.result.status,
                          protos.StatusResult.Success)
 
-        # Verify otel_libs_available is set to True
-        self.assertTrue(otel_manager.get_azure_monitor_available())
+        # Verify OpenTelemetry propagation is initialized
+        self.assertTrue(otel_manager.get_otel_libs_available())
+        self.assertIsNotNone(otel_manager.get_context_api())
+        self.assertIsNotNone(otel_manager.get_trace_context_propagator())
         # Verify that WorkerOpenTelemetryEnabled capability is set to _TRUE
         capabilities = init_response.capabilities
         self.assertIn("WorkerOpenTelemetryEnabled", capabilities)