Skip to content

Commit 74fe555

Browse files
author
Victoria Hall
committed
fix mock env variable
1 parent d13be18 commit 74fe555

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

tests/unittests/test_opentelemetry.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import asyncio
2+
import os
23
import unittest
34
from unittest.mock import MagicMock, patch
45

@@ -56,15 +57,13 @@ def test_initialize_azure_monitor_import_error(
5657
# Verify that otel_libs_available is set to False due to ImportError
5758
self.assertFalse(self.dispatcher._azure_monitor_available)
5859

59-
@patch("azure_functions_worker.dispatcher.get_app_setting")
60+
@patch.dict(os.environ, {'PYTHON_ENABLE_OPENTELEMETRY': 'true'})
6061
@patch('builtins.__import__')
6162
def test_init_request_otel_capability_enabled_app_setting(
6263
self,
6364
mock_imports,
64-
mock_app_setting,
6565
):
6666
mock_imports.return_value = MagicMock()
67-
mock_app_setting.return_value = True
6867

6968
init_request = protos.StreamingMessage(
7069
worker_init_request=protos.WorkerInitRequest(
@@ -85,13 +84,10 @@ def test_init_request_otel_capability_enabled_app_setting(
8584
self.assertEqual(capabilities["WorkerOpenTelemetryEnabled"], "true")
8685

8786
@patch("azure_functions_worker.dispatcher.Dispatcher.initialize_azure_monitor")
88-
@patch("azure_functions_worker.dispatcher.get_app_setting")
8987
def test_init_request_otel_capability_disabled_app_setting(
9088
self,
91-
mock_app_setting,
9289
mock_initialize_azmon,
9390
):
94-
mock_app_setting.return_value = False
9591

9692
init_request = protos.StreamingMessage(
9793
worker_init_request=protos.WorkerInitRequest(

0 commit comments

Comments
 (0)