"""Integration tests for dataloader endpoints."""

import pytest

from royalties.tests.integration.conftest import ows_royalties_api_client


@pytest.mark.jira('PLATFORM-4341')
def test_statement_period_dataloader_authorization(
    statement_period_fixtures: None,
    auth_headers: dict[str, str],
    unauthorized_headers: dict[str, str],
) -> None:
    """Test authorization for the statement period dataloader endpoint."""
    ids = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
    authorized_client = ows_royalties_api_client(auth_headers)
    res = authorized_client.post_statement_periods_dataloader(ids)
    assert res.status_code == 200
    assert len(res.json()) == 10

    unauthorized_client = ows_royalties_api_client(unauthorized_headers)
    res = unauthorized_client.post_statement_periods_dataloader(ids)
    assert res.status_code == 403
    assert res.json()['message'] == 'User is forbidden'