"""Integration tests using stress test complicated user."""

from typing import Any

import requests
from mypy_boto3_dynamodb import DynamoDBClient

from pdp.fastapi.schemas.identity import RolesResponse
from tests.integration import config, utils


def test_get_roles_by_identity_stress_test(
    default_boto_client: DynamoDBClient,
    seed_test_table_stress_test_complicated_user: None,
    bearer_token_complicated_identity_user: str,
    caplog: Any,
) -> None:
    """GET /identity/{identity_uuid}/roles/ endpoint with complicated stress test user."""  # noqa: E501
    bearer_token_identity_uuid = utils.get_bearer_token_identity_uuid(
        bearer_token_complicated_identity_user
    )
    url = f"{config.QA_BASE_URL}/identity/{bearer_token_identity_uuid}/roles/"
    response = requests.get(
        url,
        headers={"Authorization": f"Bearer {bearer_token_complicated_identity_user}"},
    )  # noqa: E501
    roles_response = RolesResponse.model_validate_json(response.text)
    assert len(roles_response.tenants) == 596


def test_get_my_roles_stress_test(
    default_boto_client: DynamoDBClient,
    seed_test_table_stress_test_complicated_user: None,
    bearer_token_complicated_identity_user: str,
) -> None:
    """GET /identity/self/roles/ endpoint with complicated stress test user."""
    url = f"{config.QA_BASE_URL}/identity/self/roles/"
    response = requests.get(
        url,
        headers={"Authorization": f"Bearer {bearer_token_complicated_identity_user}"},
    )  # noqa: E501

    assert response.status_code == 200, f"Response: {response.text}"
    roles_response = RolesResponse.model_validate_json(response.text)
    assert len(roles_response.tenants) == 596


def test_get_my_allowed_tenants_stress_test(
    default_boto_client: DynamoDBClient,
    seed_test_table_stress_test_complicated_user: None,
    bearer_token_complicated_identity_user: str,
) -> None:
    """GET /identity/self/allowed-tenants/ endpoint with complicated stress test user."""  # noqa: E501
    url = f"{config.QA_BASE_URL}/identity/self/allowed-tenants/"
    response = requests.post(
        url,
        json={"action": "view", "resource_type": "audience"},
        headers={"Authorization": f"Bearer {bearer_token_complicated_identity_user}"},
    )

    assert response.status_code == 200, f"Response: {response.text}"
    assert response.json() == {
        "action": "view",
        "resource_type": "audience",
        "tenants": [],
    }