"""Unit tests for reference_flowthrough_calculation schema."""

from abacus_contract.schemas.reference_flowthrough_calculation import (
    ReferenceFlowthroughCalculationSchema,
)
from abacus_contract.tests.utils.factories import ReferenceFlowthroughCalculationFactory


def test_reference_flowthrough_calculation_schema():
    """Test reference_flowthrough_calculation schema."""
    mock_reference_flowthrough_calculation = (
        ReferenceFlowthroughCalculationFactory.create()
    )
    res = ReferenceFlowthroughCalculationSchema().dump(
        mock_reference_flowthrough_calculation
    )

    assert res == {
        'reference_flowthrough_calculation_id': mock_reference_flowthrough_calculation.reference_flowthrough_calculation_id,
        'flowthrough_calculation_name': mock_reference_flowthrough_calculation.flowthrough_calculation_name,
        'flowthrough_calculation': mock_reference_flowthrough_calculation.flowthrough_calculation,
        'flowthrough_calculation_example': mock_reference_flowthrough_calculation.flowthrough_calculation_example,
        'flowthrough_calculation_example_summary': mock_reference_flowthrough_calculation.flowthrough_calculation_example_summary,
    }