from datetime import datetime
from decimal import Decimal

from finance import flow


def test_flow_integration() -> None:
    fl = flow.Flow(
        amount=Decimal(3000.0),
        since=datetime(2023, 1, 1),
        until=datetime(2026, 1, 5),
    )

    tests = (
        (datetime(2024, 3, 12), datetime(2024, 4, 2), Decimal(3000.0)),
        (datetime(2024, 3, 1), datetime(2024, 3, 15), Decimal(3000.0)),
        (datetime(2024, 2, 25), datetime(2024, 3, 1), Decimal(3000.0)),
        (datetime(2024, 2, 25), datetime(2024, 6, 12), Decimal(12000.0)),
        (datetime(2022, 1, 5), datetime(2022, 9, 14), Decimal(0.0)),
        (datetime(2022, 1, 5), datetime(2023, 1, 1), Decimal(3000.0)),
        (datetime(2022, 7, 4), datetime(2024, 12, 8), Decimal(72000.0)),
        (datetime(2025, 11, 4), datetime(2026, 5, 9), Decimal(6000.0)),
        (datetime(2026, 1, 5), datetime(2027, 1, 1), Decimal(0.0)),
        (datetime(2027, 5, 28), datetime(2028, 7, 7), Decimal(0.0)),
    )

    for test in tests:
        assert fl.integrate(start=test[0], end=test[1]) == test[2]