from __future__ import annotations

import uuid
from types import SimpleNamespace


def test_standard_oss_instance_id_aligns_to_embedded_uuid():
    from langbot.pkg.workspace.identity import workspace_uuid_from_instance_id

    instance_uuid = 'a711d9e4-0953-443f-a0e9-7dd50193a79f'

    assert workspace_uuid_from_instance_id(instance_uuid) == instance_uuid
    assert workspace_uuid_from_instance_id(f'instance_{instance_uuid}') == instance_uuid


def test_custom_legacy_instance_id_maps_to_stable_valid_uuid():
    from langbot.pkg.workspace.identity import workspace_uuid_from_instance_id

    first = workspace_uuid_from_instance_id('instance_migration_test')
    second = workspace_uuid_from_instance_id('instance_migration_test')

    assert first == second
    assert str(uuid.UUID(first)) == first


def test_query_telemetry_identity_reports_instance_and_workspace():
    from langbot.pkg.telemetry.identity import workspace_identity

    identity = workspace_identity(SimpleNamespace(workspace_uuid='workspace-a', instance_uuid='instance-a'))

    assert identity == {
        'instance_id': 'instance-a',
        'workspace_uuid': 'workspace-a',
    }