"""Tests for ows_royalties connector."""

import httpx
from owsclient.test import OwsClientMock

from src.connectors import ows_royalties


def test_get_statement_period_payment_entity_states(
    ows_client_mock: OwsClientMock,
) -> None:
    """Test successful get payment entity states."""
    mock_response = [
        {
            'statement_period_payment_entity_id': 1,
            'reference_payment_entity_id': 2,
            'statement_period_id': 300,
            'abacus_state_id': 10,
            'action_name': 'close_balance',
            'action_status': 'complete',
        },
    ]
    statement_period_id = 300

    ows_client_mock.get(
        'ows-royalties',
        f'/statement-period/{statement_period_id}/payment-entities/states',
    ).mock(
        return_value=httpx.Response(
            200,
            json=mock_response,
        )
    )

    response = ows_royalties.get_statement_period_payment_entity_states(
        statement_period_id
    )

    assert response.status_code == 200
    assert response.json() == mock_response