File tree Expand file tree Collapse file tree 1 file changed +2
-6
lines changed Expand file tree Collapse file tree 1 file changed +2
-6
lines changed Original file line number Diff line number Diff line change 1
1
import asyncio
2
+ import os
2
3
import unittest
3
4
from unittest .mock import MagicMock , patch
4
5
@@ -56,15 +57,13 @@ def test_initialize_azure_monitor_import_error(
56
57
# Verify that otel_libs_available is set to False due to ImportError
57
58
self .assertFalse (self .dispatcher ._azure_monitor_available )
58
59
59
- @patch ( "azure_functions_worker.dispatcher.get_app_setting" )
60
+ @patch . dict ( os . environ , { 'PYTHON_ENABLE_OPENTELEMETRY' : 'true' } )
60
61
@patch ('builtins.__import__' )
61
62
def test_init_request_otel_capability_enabled_app_setting (
62
63
self ,
63
64
mock_imports ,
64
- mock_app_setting ,
65
65
):
66
66
mock_imports .return_value = MagicMock ()
67
- mock_app_setting .return_value = True
68
67
69
68
init_request = protos .StreamingMessage (
70
69
worker_init_request = protos .WorkerInitRequest (
@@ -85,13 +84,10 @@ def test_init_request_otel_capability_enabled_app_setting(
85
84
self .assertEqual (capabilities ["WorkerOpenTelemetryEnabled" ], "true" )
86
85
87
86
@patch ("azure_functions_worker.dispatcher.Dispatcher.initialize_azure_monitor" )
88
- @patch ("azure_functions_worker.dispatcher.get_app_setting" )
89
87
def test_init_request_otel_capability_disabled_app_setting (
90
88
self ,
91
- mock_app_setting ,
92
89
mock_initialize_azmon ,
93
90
):
94
- mock_app_setting .return_value = False
95
91
96
92
init_request = protos .StreamingMessage (
97
93
worker_init_request = protos .WorkerInitRequest (
You can’t perform that action at this time.
0 commit comments