"""Tests for statement_period_payment_entity handlers."""

from unittest.mock import patch

from owsresponse import response


@patch('royalties.blueprints.statement_period_payment_entity.logic')
def test_get_states_by_statement_period(mock_logic, fixture_client):
    """Test getting statement_period_payment_entity states by statement_period."""
    statement_period_id = 300
    mock_response = [
        {
            'statement_period_payment_entity_id': 1,
            'reference_payment_entity_id': 2,
            'statement_period_id': statement_period_id,
            'abacus_state_id': 1000,
            'action_name': 'close_balance',
            'action_status': 'init',
        }
    ]

    mock_logic.get_states_by_statement_period.return_value = response.Response(
        message=mock_response, status=200
    )

    res = fixture_client.get(
        f'/statement-period/{statement_period_id}/payment-entities/states'
    )

    assert res.status_code == 200
    mock_logic.get_states_by_statement_period.assert_called_once_with(
        statement_period_id
    )