[TESTS][Telemetry Utils]

tests/telemetry/user_utils.py

@@ -0,0 +1,86 @@
+import uuid
+
+from swarms.telemetry.user_utils import (
+    generate_unique_identifier,
+    generate_user_id,
+    get_machine_id,
+    get_system_info,
+)
+
+# Helper functions tests
+def test_generate_user_id():
+    # Generate user IDs and ensure they are UUID strings
+    user_id = generate_user_id()
+    assert isinstance(user_id, str)
+    assert uuid.UUID(user_id, version=4)
+
+
+def test_get_machine_id():
+    # Get machine ID and ensure it's a valid SHA-256 hash
+    machine_id = get_machine_id()
+    assert isinstance(machine_id, str)
+    assert len(machine_id) == 64
+    assert all(char in "0123456789abcdef" for char in machine_id)
+
+
+def test_get_system_info():
+    # Get system information and ensure it's a dictionary with expected keys
+    system_info = get_system_info()
+    assert isinstance(system_info, dict)
+    expected_keys = [
+        "platform",
+        "platform_release",
+        "platform_version",
+        "architecture",
+        "hostname",
+        "ip_address",
+        "mac_address",
+        "processor",
+        "python_version",
+    ]
+    assert all(key in system_info for key in expected_keys)
+
+
+def test_generate_unique_identifier():
+    # Generate unique identifiers and ensure they are valid UUID strings
+    unique_id = generate_unique_identifier()
+    assert isinstance(unique_id, str)
+    assert uuid.UUID(
+        unique_id, version=5, namespace=uuid.NAMESPACE_DNS
+    )
+
+
+def test_generate_user_id_edge_case():
+    # Test generate_user_id with multiple calls
+    user_ids = set()
+    for _ in range(100):
+        user_id = generate_user_id()
+        user_ids.add(user_id)
+    assert len(user_ids) == 100  # Ensure generated IDs are unique
+
+
+def test_get_machine_id_edge_case():
+    # Test get_machine_id with multiple calls
+    machine_ids = set()
+    for _ in range(100):
+        machine_id = get_machine_id()
+        machine_ids.add(machine_id)
+    assert len(machine_ids) == 100  # Ensure generated IDs are unique
+
+
+def test_get_system_info_edge_case():
+    # Test get_system_info for consistency
+    system_info1 = get_system_info()
+    system_info2 = get_system_info()
+    assert (
+        system_info1 == system_info2
+    )  # Ensure system info remains the same
+
+
+def test_generate_unique_identifier_edge_case():
+    # Test generate_unique_identifier for uniqueness
+    unique_ids = set()
+    for _ in range(100):
+        unique_id = generate_unique_identifier()
+        unique_ids.add(unique_id)
+    assert len(unique_ids) == 100  # Ensure generated IDs are unique