"""Test for the recipient logic."""

from collaborator.logic import recipient


def test_get_by_id_success(mocker):
    """Test getting recipient by an id."""
    recipient_id = 1234

    mock_get_by_id = mocker.patch.object(recipient.RecipientPersister, "get_recipient")

    recipient.get_by_id(recipient_id)

    mock_get_by_id.assert_called_with(recipient_id)


def test_get_by_ids_success(mocker):
    """Test getting recipient by an id."""
    recipient_ids = [1234, 5678]

    mock_get_by_id = mocker.patch.object(
        recipient.RecipientPersister, "get_recipients_by_id"
    )

    recipient.get_by_ids(recipient_ids)

    mock_get_by_id.assert_called_with(recipient_ids)