"""Advances logic tests."""

from unittest.mock import patch

from moneyhub.logic import advances as logic
from tests.utils.factories import CombinedAdvancesFactory


@patch('moneyhub.logic.advances.models')
def test_get_by_account_and_statement_period(mock_models):
    """Test get_by_account_and_statement_period method."""
    advances = [
        CombinedAdvancesFactory.build(),
        CombinedAdvancesFactory.build(),
    ]
    mock_models.CombinedAdvances.get_by_account_and_statement_periods.return_value = \
        advances
    result = logic.get_by_account_and_statement_periods(24601, [123], 10001,)

    assert result == advances
    mock_models.CombinedAdvances.get_by_account_and_statement_periods.\
        assert_called_once_with(24601, [123], 10001)