test fixes

Author: hallvictoria

tests/unittests/test_handle_event.py

@@ -136,6 +136,7 @@ async def test_function_environment_reload_request(self, mock_index_function_app
                                            BASIC_FUNCTION_DIRECTORY)),
                                        properties={'host': '123',
                                                    'protos': test_protos})
+        handle_event.protos = test_protos
         result = await function_environment_reload_request(worker_request)
         self.assertEqual(result.capabilities, {})
         self.assertEqual(result.worker_metadata.runtime_name, "python")
@@ -152,6 +153,7 @@ async def test_function_environment_reload_request_with_streaming(
             self,
             mock_http_v2_enabled,
             mock_initialize_http_server):
+        handle_event.protos = test_protos
         worker_request = WorkerRequest(name='function_environment_reload_request',
                                        request=Request(FunctionRequest(
                                            'hello',
@@ -172,6 +174,7 @@ async def test_function_environment_reload_request_with_streaming(
            return_value=True)
     async def test_function_environment_reload_request_with_otel(self,
                                                                  mock_otel_enabled):
+        handle_event.protos = test_protos
         worker_request = WorkerRequest(name='function_environment_reload_request',
                                        request=Request(FunctionRequest(
                                            'hello',
@@ -189,6 +192,7 @@ async def test_function_environment_reload_request_with_otel(self,
     async def test_function_environment_reload_request_with_exception(self):
         # Even if an exception happens during indexing,
         # we still return success
+        handle_event.protos = test_protos
         worker_request = WorkerRequest(name='function_environment_reload_request',
                                        request=Request(FunctionRequest(
                                            'hello',

tests/unittests/test_opentelemetry.py

@@ -7,6 +7,7 @@
 from azure_functions_worker_v2.handle_event import otel_manager, worker_init_request
 from azure_functions_worker_v2.otel import (initialize_azure_monitor,
                                             update_opentelemetry_status)
+from azure_functions_worker_v2.logging import logger
 from tests.utils.constants import UNIT_TESTS_FOLDER
 from tests.utils.mock_classes import FunctionRequest, Request, WorkerRequest
 from unittest.mock import MagicMock, patch
@@ -17,12 +18,21 @@
 
 class TestOpenTelemetry(unittest.TestCase):
 
-    @patch('builtins.__import__', side_effect=ImportError)
-    def test_update_opentelemetry_status_import_error(self, mock_import_error):
-        update_opentelemetry_status()
-        # Verify that context variables are None due to ImportError
-        self.assertIsNone(otel_manager.get_context_api())
-        self.assertIsNone(otel_manager.get_trace_context_propagator())
+    def test_update_opentelemetry_status_import_error(self):
+        with patch.dict('sys.modules', {
+            'opentelemetry': None,
+            'opentelemetry.context': None,
+            'opentelemetry.trace': None,
+            'opentelemetry.trace.propagation': None,
+            'opentelemetry.trace.propagation.tracecontext': None,
+        }):
+            # Verify that context variables are None due to ImportError
+            with self.assertLogs(logger.name, 'ERROR') as cm:
+                update_opentelemetry_status()
+                self.assertTrue(
+                    any("Cannot import OpenTelemetry libraries."
+                        in message for message in cm.output)
+                )
 
     @patch('builtins.__import__')
     def test_update_opentelemetry_status_success(

tests/utils/mock_classes.py

@@ -22,7 +22,7 @@ def __init__(self, name: Any):
 class FunctionRequest:
     def __init__(self, capabilities: Any,
                  function_app_directory: Any,
-                 environment_variables: Optional[Any] = None):
+                 environment_variables: Optional[Any] = {}):
         self.capabilities = capabilities
         self.function_app_directory = function_app_directory
         self.environment_variables = environment_variables