"""PayeeCollaborator model tests."""

import pytest

from payee.models.payee_collaborator import PayeeCollaborator


@pytest.mark.parametrize(
    'args,expected_ids',
    [
        (
            dict(),
            {1, 2},
        ),
        (
            {'payee_ids': []},
            {1, 2},
        ),
        (
            {'payee_ids': [1]},
            {1},
        ),
    ],
)
def test_get_payees(
    args,
    expected_ids,
    payee_fixtures,
    payee_collaborator_fixtures,
):
    """Get payee collaborators."""

    payees, total_count = PayeeCollaborator.get_payee_collaborators(
        limit=2, offset=0, **args
    )
    assert total_count == len(expected_ids)
    returned_ids = {p.payee_collaborator_id for p in payees}
    assert returned_ids == expected_ids