"""Integration tests helpers."""

from typing import Any

import requests
from mypy_boto3_dynamodb import DynamoDBClient

from tests.integration.conftest import seed_test_pp_identity


def assert_get_identity_roles_empty(url: str, bearer_token: str) -> None:
    """Verify the roles returned by endpoint are empty."""
    response = requests.get(url, headers={"Authorization": f"Bearer {bearer_token}"})

    assert response.status_code == 200, f"Response: {response.text}"
    assert response.json() == {
        "cursor": {"cursor": None, "shorthand": None},
        "errors": {},
        "tenants": {},
    }


def assert_get_identity_roles(
    url: Any,
    boto_client: DynamoDBClient,
    bearer_token: str,
    bearer_token_identity_uuid: str,
) -> None:
    """Verify the roles returned by endpoint match expected."""
    seed_test_pp_identity(
        boto_client,
        bearer_token_identity_uuid,
        "573d0372-7f2f-48a6-8deb-c9a6558f9549",
        "audience_development_analyst",
    )

    response = requests.get(url, headers={"Authorization": f"Bearer {bearer_token}"})

    assert response.status_code == 200, f"Response: {response.text}"
    assert response.json() == {
        "cursor": {"cursor": None, "shorthand": None},
        "errors": {},
        "tenants": {
            "573d0372-7f2f-48a6-8deb-c9a6558f9549": {
                "roles": [{"role": "audience_development_analyst"}],
                "tenant_type": "account",
                "tenant_uuid": "573d0372-7f2f-48a6-8deb-c9a6558f9549",
            }
        },
    }