"""Contracts model tests."""


from moneyhub.models import Contracts
from tests.unit.conftest import using_mock_snowflake_table

_MOCK_DATA = {
    'contract_dbt': [
        {
            'account_id': 57608,
            'contract_id': 531211,
            'contract_name': 'Contract 1'
        },
        {
            'account_id': 57608,
            'contract_id': 531219,
            'contract_name': 'Contract 2'
        },
        {
            'account_id': 57608,
            'contract_id': 531458,
            'contract_name': 'Contract 3'
        },
    ]
}


@using_mock_snowflake_table(Contracts, _MOCK_DATA)
def test_get_by_account_id():
    """Test to get ledger adjustments details for a specified account."""
    account_id = 57608
    contract_ids = [531211, 531219, 531458]

    items = Contracts.get_by_account_id(
        account_id, contract_ids)

    assert [item.contract_id for item in items] == [531211, 531219, 531458]