refactor(framework:skip) Add partner_id tests (#3872)

src/py/flwr/common/telemetry_test.py

@@ -15,12 +15,14 @@
 """Telemetry tests."""
 
 
+import os
 import time
 import unittest
 from typing import Callable
 from unittest import mock
+from uuid import uuid4
 
-from flwr.common.telemetry import EventType, _get_source_id, event
+from flwr.common.telemetry import EventType, _get_partner_id, _get_source_id, event
 
 
 class TelemetryTest(unittest.TestCase):
@@ -109,3 +111,39 @@ def _new_failing_get_home() -> None:
 
         # Assert
         self.assertEqual(source_id, except_value)
+
+    def test_get_partner_id(self) -> None:
+        """Test if _get_partner_id returns an ID successfully."""
+        # Prepare
+        generated_id = str(uuid4())
+        os.environ["FLWR_TELEMETRY_PARTNER_ID"] = generated_id
+
+        # Execute
+        partner_id = _get_partner_id()
+
+        # Assert
+        self.assertEqual(partner_id, generated_id)
+
+    def test_get_partner_id_no_env(self) -> None:
+        """Test if _get_partner_id returns unavailable without an env variable."""
+        # Prepare
+        os.environ["FLWR_TELEMETRY_PARTNER_ID"] = ""
+        expected_value = "unavailable"
+
+        # Execute
+        partner_id = _get_partner_id()
+
+        # Assert
+        self.assertEqual(partner_id, expected_value)
+
+    def test_get_partner_id_invalid(self) -> None:
+        """Test if _get_partner_id returns invalid with an incorrect env variable."""
+        # Prepare
+        os.environ["FLWR_TELEMETRY_PARTNER_ID"] = "not a valid ID"
+        expected_value = "invalid"
+
+        # Execute
+        partner_id = _get_partner_id()
+
+        # Assert
+        self.assertEqual(partner_id, expected_value)