mirror of
https://github.com/open-telemetry/opentelemetry-python-contrib.git
synced 2025-08-02 19:47:17 +08:00
Azure resource detectors (#1901)
This commit is contained in:
@ -0,0 +1,116 @@
|
||||
# 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
|
||||
|
||||
from opentelemetry.resource.detector.azure.app_service import (
|
||||
AzureAppServiceResourceDetector,
|
||||
)
|
||||
|
||||
TEST_WEBSITE_SITE_NAME = "TEST_WEBSITE_SITE_NAME"
|
||||
TEST_REGION_NAME = "TEST_REGION_NAME"
|
||||
TEST_WEBSITE_SLOT_NAME = "TEST_WEBSITE_SLOT_NAME"
|
||||
TEST_WEBSITE_HOSTNAME = "TEST_WEBSITE_HOSTNAME"
|
||||
TEST_WEBSITE_INSTANCE_ID = "TEST_WEBSITE_INSTANCE_ID"
|
||||
TEST_WEBSITE_HOME_STAMPNAME = "TEST_WEBSITE_HOME_STAMPNAME"
|
||||
|
||||
TEST_WEBSITE_RESOURCE_GROUP = "TEST_WEBSITE_RESOURCE_GROUP"
|
||||
TEST_WEBSITE_OWNER_NAME = "TEST_WEBSITE_OWNER_NAME"
|
||||
|
||||
class TestAzureAppServiceResourceDetector(unittest.TestCase):
|
||||
@patch.dict("os.environ", {
|
||||
"WEBSITE_SITE_NAME": TEST_WEBSITE_SITE_NAME,
|
||||
"REGION_NAME": TEST_REGION_NAME,
|
||||
"WEBSITE_SLOT_NAME": TEST_WEBSITE_SLOT_NAME,
|
||||
"WEBSITE_HOSTNAME": TEST_WEBSITE_HOSTNAME,
|
||||
"WEBSITE_INSTANCE_ID": TEST_WEBSITE_INSTANCE_ID,
|
||||
"WEBSITE_HOME_STAMPNAME": TEST_WEBSITE_HOME_STAMPNAME,
|
||||
"WEBSITE_RESOURCE_GROUP": TEST_WEBSITE_RESOURCE_GROUP,
|
||||
"WEBSITE_OWNER_NAME": TEST_WEBSITE_OWNER_NAME,
|
||||
}, clear=True)
|
||||
def test_on_app_service(self):
|
||||
resource = AzureAppServiceResourceDetector().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_app_service")
|
||||
|
||||
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["deployment.environment"], TEST_WEBSITE_SLOT_NAME)
|
||||
self.assertEqual(attributes["host.id"], TEST_WEBSITE_HOSTNAME)
|
||||
self.assertEqual(attributes["service.instance.id"], TEST_WEBSITE_INSTANCE_ID)
|
||||
self.assertEqual(attributes["azure.app.service.stamp"], TEST_WEBSITE_HOME_STAMPNAME)
|
||||
|
||||
@patch.dict("os.environ", {
|
||||
"WEBSITE_SITE_NAME": TEST_WEBSITE_SITE_NAME,
|
||||
"REGION_NAME": TEST_REGION_NAME,
|
||||
"WEBSITE_SLOT_NAME": TEST_WEBSITE_SLOT_NAME,
|
||||
"WEBSITE_HOSTNAME": TEST_WEBSITE_HOSTNAME,
|
||||
"WEBSITE_INSTANCE_ID": TEST_WEBSITE_INSTANCE_ID,
|
||||
"WEBSITE_HOME_STAMPNAME": TEST_WEBSITE_HOME_STAMPNAME,
|
||||
"WEBSITE_OWNER_NAME": TEST_WEBSITE_OWNER_NAME,
|
||||
}, clear=True)
|
||||
def test_on_app_service_no_resource_group(self):
|
||||
resource = AzureAppServiceResourceDetector().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_app_service")
|
||||
|
||||
self.assertTrue("cloud.resource_id" not in attributes)
|
||||
|
||||
self.assertEqual(attributes["cloud.region"], TEST_REGION_NAME)
|
||||
self.assertEqual(attributes["deployment.environment"], TEST_WEBSITE_SLOT_NAME)
|
||||
self.assertEqual(attributes["host.id"], TEST_WEBSITE_HOSTNAME)
|
||||
self.assertEqual(attributes["service.instance.id"], TEST_WEBSITE_INSTANCE_ID)
|
||||
self.assertEqual(attributes["azure.app.service.stamp"], TEST_WEBSITE_HOME_STAMPNAME)
|
||||
|
||||
@patch.dict("os.environ", {
|
||||
"WEBSITE_SITE_NAME": TEST_WEBSITE_SITE_NAME,
|
||||
"REGION_NAME": TEST_REGION_NAME,
|
||||
"WEBSITE_SLOT_NAME": TEST_WEBSITE_SLOT_NAME,
|
||||
"WEBSITE_HOSTNAME": TEST_WEBSITE_HOSTNAME,
|
||||
"WEBSITE_INSTANCE_ID": TEST_WEBSITE_INSTANCE_ID,
|
||||
"WEBSITE_HOME_STAMPNAME": TEST_WEBSITE_HOME_STAMPNAME,
|
||||
"WEBSITE_OWNER_NAME": TEST_WEBSITE_OWNER_NAME,
|
||||
}, clear=True)
|
||||
def test_on_app_service_no_owner(self):
|
||||
resource = AzureAppServiceResourceDetector().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_app_service")
|
||||
|
||||
self.assertTrue("cloud.resource_id" not in attributes)
|
||||
|
||||
self.assertEqual(attributes["cloud.region"], TEST_REGION_NAME)
|
||||
self.assertEqual(attributes["deployment.environment"], TEST_WEBSITE_SLOT_NAME)
|
||||
self.assertEqual(attributes["host.id"], TEST_WEBSITE_HOSTNAME)
|
||||
self.assertEqual(attributes["service.instance.id"], TEST_WEBSITE_INSTANCE_ID)
|
||||
self.assertEqual(attributes["azure.app.service.stamp"], TEST_WEBSITE_HOME_STAMPNAME)
|
||||
|
||||
@patch.dict("os.environ", {
|
||||
"REGION_NAME": TEST_REGION_NAME,
|
||||
"WEBSITE_SLOT_NAME": TEST_WEBSITE_SLOT_NAME,
|
||||
"WEBSITE_HOSTNAME": TEST_WEBSITE_HOSTNAME,
|
||||
"WEBSITE_INSTANCE_ID": TEST_WEBSITE_INSTANCE_ID,
|
||||
"WEBSITE_HOME_STAMPNAME": TEST_WEBSITE_HOME_STAMPNAME,
|
||||
"WEBSITE_OWNER_NAME": TEST_WEBSITE_OWNER_NAME,
|
||||
}, clear=True)
|
||||
def test_off_app_service(self):
|
||||
resource = AzureAppServiceResourceDetector().detect()
|
||||
self.assertEqual(resource.attributes, {})
|
Reference in New Issue
Block a user