"""Tests for contract logic."""

from royalties.logic import contract


def test_get_account_ids_by_contract_ids(account_contract_fixtures: None) -> None:
    """Test getting account IDs by contract IDs."""
    # Records created in account_contract_fixtures
    result = contract.get_account_ids_by_contract_ids([1, 2, 3])

    assert result == [1, 1, 1]


def test_get_account_ids_by_contract_ids_no_result() -> None:
    """Test getting account ID by contract ID when no result is found."""
    result = contract.get_account_ids_by_contract_ids([1, 2, 3])

    assert result == []