Tests

Azure · Feb 7, 2024 · ba7c197 · ba7c197
1 parent e6da61a
commit ba7c197
Show file tree

Hide file tree

Showing 3 changed files with 91 additions and 17 deletions.
diff --git a/sdk/monitor/azure-monitor-opentelemetry-exporter/CHANGELOG.md b/sdk/monitor/azure-monitor-opentelemetry-exporter/CHANGELOG.md
@@ -4,15 +4,15 @@
 
 ### Features Added
 
-- Add attachType to statsbeat
-    ([#XXXXX](https://github.com/Azure/azure-sdk-for-python/pull/XXXXX))
-
 ### Breaking Changes
 
 ### Bugs Fixed
 
 ### Other Changes
 
+- Add attachType to statsbeat
+    ([#34226](https://github.com/Azure/azure-sdk-for-python/pull/34226))
+
 ## 1.0.0b22 (2024-02-01)
 
 ### Features Added

diff --git a/...nitor/azure-monitor-opentelemetry-exporter/azure/monitor/opentelemetry/exporter/_utils.py b/...nitor/azure-monitor-opentelemetry-exporter/azure/monitor/opentelemetry/exporter/_utils.py
@@ -52,10 +52,10 @@ def _get_sdk_version_prefix():
         rp = 'a'
     if _is_on_function():
         rp = 'f'
-    # TODO: add VM scenario outside statsbeat
+    # TODO: Add VM scenario outside statsbeat
     # if _is_on_vm():
     #     rp = 'v'
-    # TODO: add AKS scenario
+    # TODO: Add AKS scenario
     # if _is_on_aks():
     #     rp = 'k'
 

diff --git a/sdk/monitor/azure-monitor-opentelemetry-exporter/tests/test_utils.py b/sdk/monitor/azure-monitor-opentelemetry-exporter/tests/test_utils.py
@@ -77,36 +77,110 @@ def test_create_telemetry_item(self, mock_ns_to_iso_str):
         )
         self.assertEqual(result, expected)
 
-    @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=True)
+    # Unknown
+
+    @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=False)
+    @patch("azure.monitor.opentelemetry.exporter._utils.platform.system", return_value="")
+    def test_get_sdk_version_prefix(self, mock_system, mock_getenv):
+        result = _utils._get_sdk_version_prefix()
+        self.assertEqual(result, "uum_")
+
+    @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=False)
     @patch("azure.monitor.opentelemetry.exporter._utils.platform.system", return_value="Linux")
-    def test_get_sdk_version_prefix_off_app_service(self, mock_system, mock_getenv):
+    def test_get_sdk_version_prefix_linux(self, mock_system, mock_getenv):
         result = _utils._get_sdk_version_prefix()
-        self.assertEqual(result, "")
+        self.assertEqual(result, "ulm_")
+
+    @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=False)
+    @patch("azure.monitor.opentelemetry.exporter._utils.platform.system", return_value="Windows")
+    def test_get_sdk_version_prefix_wiindows(self, mock_system, mock_getenv):
+        result = _utils._get_sdk_version_prefix()
+        self.assertEqual(result, "uwm_")
+
+    # App Service
+
+    @patch("azure.monitor.opentelemetry.exporter._utils.environ", {"WEBSITE_SITE_NAME": TEST_WEBSITE_SITE_NAME})
+    @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=False)
+    @patch("azure.monitor.opentelemetry.exporter._utils.platform.system", return_value="")
+    def test_get_sdk_version_prefix_app_service(self, mock_system, mock_getenv):
+        result = _utils._get_sdk_version_prefix()
+        self.assertEqual(result, "aum_")
 
     @patch("azure.monitor.opentelemetry.exporter._utils.environ", {"WEBSITE_SITE_NAME": TEST_WEBSITE_SITE_NAME})
     @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=False)
     @patch("azure.monitor.opentelemetry.exporter._utils.platform.system", return_value="Linux")
-    def test_get_sdk_version_prefix_app_service_disabled_attach(self, mock_system, mock_getenv):
+    def test_get_sdk_version_prefix_app_service_linux(self, mock_system, mock_getenv):
+        result = _utils._get_sdk_version_prefix()
+        self.assertEqual(result, "alm_")
+
+    @patch("azure.monitor.opentelemetry.exporter._utils.environ", {"WEBSITE_SITE_NAME": TEST_WEBSITE_SITE_NAME})
+    @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=False)
+    @patch("azure.monitor.opentelemetry.exporter._utils.platform.system", return_value="Windows")
+    def test_get_sdk_version_prefix_app_service_windows(self, mock_system, mock_getenv):
         result = _utils._get_sdk_version_prefix()
-        self.assertEqual(result, "")
+        self.assertEqual(result, "awm_")
+
+    @patch("azure.monitor.opentelemetry.exporter._utils.environ", {"WEBSITE_SITE_NAME": TEST_WEBSITE_SITE_NAME})
+    @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=True)
+    @patch("azure.monitor.opentelemetry.exporter._utils.platform.system", return_value="")
+    def test_get_sdk_version_prefix_app_service_attach(self, mock_system, mock_getenv):
+        result = _utils._get_sdk_version_prefix()
+        self.assertEqual(result, "aui_")
 
     @patch("azure.monitor.opentelemetry.exporter._utils.environ", {"WEBSITE_SITE_NAME": TEST_WEBSITE_SITE_NAME})
     @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=True)
     @patch("azure.monitor.opentelemetry.exporter._utils.platform.system", return_value="Linux")
-    def test_get_sdk_version_prefix_linux_attach(self, mock_system, mock_getenv):
+    def test_get_sdk_version_prefix_app_service_linux_attach(self, mock_system, mock_getenv):
         result = _utils._get_sdk_version_prefix()
-        self.assertEqual(result, "al_")
+        self.assertEqual(result, "ali_")
 
     @patch("azure.monitor.opentelemetry.exporter._utils.environ", {"WEBSITE_SITE_NAME": TEST_WEBSITE_SITE_NAME})
     @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=True)
     @patch("azure.monitor.opentelemetry.exporter._utils.platform.system", return_value="Windows")
-    def test_get_sdk_version_prefix_windows_attach(self, mock_system, mock_getenv):
+    def test_get_sdk_version_prefix_app_service_windows_attach(self, mock_system, mock_getenv):
         result = _utils._get_sdk_version_prefix()
-        self.assertEqual(result, "aw_")
+        self.assertEqual(result, "awi_")
 
-    @patch("azure.monitor.opentelemetry.exporter._utils.environ", {"WEBSITE_SITE_NAME": TEST_WEBSITE_SITE_NAME})
+    # Function
+
+    @patch("azure.monitor.opentelemetry.exporter._utils.environ", {"FUNCTIONS_WORKER_RUNTIME": TEST_WEBSITE_SITE_NAME})
+    @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=False)
+    @patch("azure.monitor.opentelemetry.exporter._utils.platform.system", return_value="")
+    def test_get_sdk_version_prefix_function(self, mock_system, mock_getenv):
+        result = _utils._get_sdk_version_prefix()
+        self.assertEqual(result, "fum_")
+
+    @patch("azure.monitor.opentelemetry.exporter._utils.environ", {"FUNCTIONS_WORKER_RUNTIME": TEST_WEBSITE_SITE_NAME})
+    @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=False)
+    @patch("azure.monitor.opentelemetry.exporter._utils.platform.system", return_value="Linux")
+    def test_get_sdk_version_prefix_function_linux(self, mock_system, mock_getenv):
+        result = _utils._get_sdk_version_prefix()
+        self.assertEqual(result, "flm_")
+
+    @patch("azure.monitor.opentelemetry.exporter._utils.environ", {"FUNCTIONS_WORKER_RUNTIME": TEST_WEBSITE_SITE_NAME})
+    @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=False)
+    @patch("azure.monitor.opentelemetry.exporter._utils.platform.system", return_value="Windows")
+    def test_get_sdk_version_prefix_function_windows(self, mock_system, mock_getenv):
+        result = _utils._get_sdk_version_prefix()
+        self.assertEqual(result, "fwm_")
+
+    @patch("azure.monitor.opentelemetry.exporter._utils.environ", {"FUNCTIONS_WORKER_RUNTIME": TEST_WEBSITE_SITE_NAME})
     @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=True)
     @patch("azure.monitor.opentelemetry.exporter._utils.platform.system", return_value="")
-    def test_get_sdk_version_prefix_unknown_attach(self, mock_system, mock_getenv):
+    def test_get_sdk_version_prefix_function_attach(self, mock_system, mock_getenv):
+        result = _utils._get_sdk_version_prefix()
+        self.assertEqual(result, "fui_")
+
+    @patch("azure.monitor.opentelemetry.exporter._utils.environ", {"FUNCTIONS_WORKER_RUNTIME": TEST_WEBSITE_SITE_NAME})
+    @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=True)
+    @patch("azure.monitor.opentelemetry.exporter._utils.platform.system", return_value="Linux")
+    def test_get_sdk_version_prefix_function_linux_attach(self, mock_system, mock_getenv):
+        result = _utils._get_sdk_version_prefix()
+        self.assertEqual(result, "fli_")
+
+    @patch("azure.monitor.opentelemetry.exporter._utils.environ", {"FUNCTIONS_WORKER_RUNTIME": TEST_WEBSITE_SITE_NAME})
+    @patch("azure.monitor.opentelemetry.exporter._utils.isdir", return_value=True)
+    @patch("azure.monitor.opentelemetry.exporter._utils.platform.system", return_value="Windows")
+    def test_get_sdk_version_prefix_function_windows_attach(self, mock_system, mock_getenv):
         result = _utils._get_sdk_version_prefix()
-        self.assertEqual(result, "au_")
+        self.assertEqual(result, "fwi_")