"""Unit tests for StatementPeriodAdjustmentFile model."""

from abacus_state.models.statement_period_adjustment_file import (
    StatementPeriodAdjustmentFile,
)


def test_get_adjustment_files_by_statement_period_id(
    create_mock_statement_period_adjustment_file,
):
    """Test to get list of statement_period_adjustment_file ids."""
    statement_period_id = 1
    statement_period_adjustment_files = (
        StatementPeriodAdjustmentFile.get_adjustment_files_by_statement_period_id(
            statement_period_id
        )
    )
    assert len(statement_period_adjustment_files) == 2


def test_get_statement_period_adjustment_file(
    create_mock_statement_period_adjustment_file,
):
    """Test to get a statement_period_adjustment_file."""
    statement_period_adjustment_file_id = 1
    statement_period_adjustment_file = (
        StatementPeriodAdjustmentFile.get_statement_period_adjustment_file(
            statement_period_adjustment_file_id
        )
    )
    assert (
        statement_period_adjustment_file.statement_period_adjustment_file_id
        == statement_period_adjustment_file_id
    )