Spaces:
Sleeping
Sleeping
| 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 | |