mirror of
https://github.com/open-telemetry/opentelemetry-python-contrib.git
synced 2025-07-29 21:23:55 +08:00
f8758c6902
* Update .pylintrc * fn * Update CHANGELOG.md * commments * Add deployment.environment to functions detector * Revert "Add deployment.environment to functions detector" This reverts commit 5411759711b8bc9976705deb416d5ffd8f65590f. * Remove deployment.environment from readme * Release 0.1.5 --------- Co-authored-by: jeremydvoss <jerevoss@gmail.com>
111 lines
4.5 KiB
Python
111 lines
4.5 KiB
Python
# Copyright The OpenTelemetry Authors
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
# pylint: disable=no-name-in-module
|
|
from opentelemetry.resource.detector.azure.functions import (
|
|
AzureFunctionsResourceDetector,
|
|
)
|
|
|
|
TEST_WEBSITE_SITE_NAME = "TEST_WEBSITE_SITE_NAME"
|
|
TEST_REGION_NAME = "TEST_REGION_NAME"
|
|
TEST_WEBSITE_INSTANCE_ID = "TEST_WEBSITE_INSTANCE_ID"
|
|
|
|
TEST_WEBSITE_RESOURCE_GROUP = "TEST_WEBSITE_RESOURCE_GROUP"
|
|
TEST_WEBSITE_OWNER_NAME = "TEST_WEBSITE_OWNER_NAME"
|
|
TEST_WEBSITE_MEMORY_LIMIT_MB = "1024"
|
|
|
|
|
|
class TestAzureAppServiceResourceDetector(unittest.TestCase):
|
|
@patch.dict(
|
|
"os.environ",
|
|
{
|
|
"FUNCTIONS_WORKER_RUNTIME": "1",
|
|
"WEBSITE_SITE_NAME": TEST_WEBSITE_SITE_NAME,
|
|
"REGION_NAME": TEST_REGION_NAME,
|
|
"WEBSITE_INSTANCE_ID": TEST_WEBSITE_INSTANCE_ID,
|
|
"WEBSITE_RESOURCE_GROUP": TEST_WEBSITE_RESOURCE_GROUP,
|
|
"WEBSITE_OWNER_NAME": TEST_WEBSITE_OWNER_NAME,
|
|
"WEBSITE_MEMORY_LIMIT_MB": TEST_WEBSITE_MEMORY_LIMIT_MB,
|
|
},
|
|
clear=True,
|
|
)
|
|
@patch("opentelemetry.resource.detector.azure.functions.getpid")
|
|
def test_on_functions(self, pid_mock):
|
|
pid_mock.return_value = 1000
|
|
resource = AzureFunctionsResourceDetector().detect()
|
|
attributes = resource.attributes
|
|
self.assertEqual(attributes["service.name"], TEST_WEBSITE_SITE_NAME)
|
|
self.assertEqual(attributes["cloud.provider"], "azure")
|
|
self.assertEqual(attributes["cloud.platform"], "azure_functions")
|
|
self.assertEqual(attributes["process.pid"], 1000)
|
|
|
|
self.assertEqual(
|
|
attributes["cloud.resource_id"],
|
|
f"/subscriptions/{TEST_WEBSITE_OWNER_NAME}/resourceGroups/{TEST_WEBSITE_RESOURCE_GROUP}/providers/Microsoft.Web/sites/{TEST_WEBSITE_SITE_NAME}",
|
|
)
|
|
|
|
self.assertEqual(attributes["cloud.region"], TEST_REGION_NAME)
|
|
self.assertEqual(attributes["faas.instance"], TEST_WEBSITE_INSTANCE_ID)
|
|
self.assertEqual(attributes["faas.max_memory"], 1024)
|
|
|
|
@patch.dict(
|
|
"os.environ",
|
|
{
|
|
"FUNCTIONS_WORKER_RUNTIME": "1",
|
|
"WEBSITE_SITE_NAME": TEST_WEBSITE_SITE_NAME,
|
|
"REGION_NAME": TEST_REGION_NAME,
|
|
"WEBSITE_INSTANCE_ID": TEST_WEBSITE_INSTANCE_ID,
|
|
"WEBSITE_RESOURCE_GROUP": TEST_WEBSITE_RESOURCE_GROUP,
|
|
"WEBSITE_OWNER_NAME": TEST_WEBSITE_OWNER_NAME,
|
|
"WEBSITE_MEMORY_LIMIT_MB": "error",
|
|
},
|
|
clear=True,
|
|
)
|
|
@patch("opentelemetry.resource.detector.azure.functions.getpid")
|
|
def test_on_functions_error_memory(self, pid_mock):
|
|
pid_mock.return_value = 1000
|
|
resource = AzureFunctionsResourceDetector().detect()
|
|
attributes = resource.attributes
|
|
self.assertEqual(attributes["service.name"], TEST_WEBSITE_SITE_NAME)
|
|
self.assertEqual(attributes["cloud.provider"], "azure")
|
|
self.assertEqual(attributes["cloud.platform"], "azure_functions")
|
|
self.assertEqual(attributes["process.pid"], 1000)
|
|
|
|
self.assertEqual(
|
|
attributes["cloud.resource_id"],
|
|
f"/subscriptions/{TEST_WEBSITE_OWNER_NAME}/resourceGroups/{TEST_WEBSITE_RESOURCE_GROUP}/providers/Microsoft.Web/sites/{TEST_WEBSITE_SITE_NAME}",
|
|
)
|
|
|
|
self.assertEqual(attributes["cloud.region"], TEST_REGION_NAME)
|
|
self.assertEqual(attributes["faas.instance"], TEST_WEBSITE_INSTANCE_ID)
|
|
self.assertIsNone(attributes.get("faas.max_memory"))
|
|
|
|
@patch.dict(
|
|
"os.environ",
|
|
{
|
|
"WEBSITE_SITE_NAME": TEST_WEBSITE_SITE_NAME,
|
|
"REGION_NAME": TEST_REGION_NAME,
|
|
"WEBSITE_INSTANCE_ID": TEST_WEBSITE_INSTANCE_ID,
|
|
"WEBSITE_RESOURCE_GROUP": TEST_WEBSITE_RESOURCE_GROUP,
|
|
"WEBSITE_OWNER_NAME": TEST_WEBSITE_OWNER_NAME,
|
|
"WEBSITE_MEMORY_LIMIT_MB": TEST_WEBSITE_MEMORY_LIMIT_MB,
|
|
},
|
|
clear=True,
|
|
)
|
|
def test_off_app_service(self):
|
|
resource = AzureFunctionsResourceDetector().detect()
|
|
self.assertEqual(resource.attributes, {})
|