"""Test ows-abacus-account requests."""

from unittest.mock import patch

from payments_generate.ows_abacus_account import get_eligible_accounts


@patch('payments_generate.ows_abacus_account.raise_abacus_account_error')
def test_get_eligible_accounts_success(
    mock_raise_service_error,
    mock_eligible_accounts,
    request_engine
):
    """Test successfully getting eligible accounts."""
    payment_group_id = 33
    request_engine['ows-abacus-account'].add_spec(
        'GET',
        '/eligible-accounts',
        status=200,
        response=mock_eligible_accounts
    )

    res = get_eligible_accounts(payment_group_id)
    assert res == mock_eligible_accounts
    mock_raise_service_error.assert_not_called()


@patch('payments_generate.ows_abacus_account.raise_abacus_account_error')
def test_get_eligible_accounts_failure(mock_raise_service_error, request_engine):
    """Test failure getting eligible accounts."""
    payment_group_id = 333
    request_engine['ows-abacus-account'].add_spec(
        'GET',
        '/eligible-accounts',
        status=400,
        response='error'
    )

    res = get_eligible_accounts(payment_group_id)
    assert res is None
    mock_raise_service_error.assert_called_once_with(
        'ERROR in GET /eligible-accounts/',
        payment_group_id=payment_group_id
    )