from datetime import datetime

from finance import income


def test_income_integration() -> None:
    inc = income.Income(amount=3000.0)

    tests = (
        (datetime(2024, 3, 12), datetime(2024, 4, 2), 3000.0),
        (datetime(2024, 3, 1), datetime(2024, 3, 15), 3000.0),
        (datetime(2024, 2, 25), datetime(2024, 3, 1), 3000.0),
        (datetime(2024, 2, 25), datetime(2024, 6, 12), 12000.0),
    )

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