diff --git a/atst/domain/csp/reports.py b/atst/domain/csp/reports.py index cd87d46c..3611a45c 100644 --- a/atst/domain/csp/reports.py +++ b/atst/domain/csp/reports.py @@ -183,7 +183,7 @@ class MockReportingProvider(ReportingInterface): "Beluga": { "cumulative": CUMULATIVE_BUDGET_BELUGA, "applications": [ - MockApplication("NP02", ["Integ", "PreProd", "NP02_Prod"]), + MockApplication("NP02", ["Integ", "PreProd", "Prod"]), MockApplication("FM", ["Integ", "Prod"]), ], "budget": 70000, @@ -296,10 +296,13 @@ class MockReportingProvider(ReportingInterface): else: budget_months = {} - this_year = pendulum.now().year + end = pendulum.now() + start = end.subtract(months=12) + period = pendulum.period(start, end) + all_months = OrderedDict() - for m in range(1, 13): - month_str = "{month:02d}/{year}".format(month=m, year=this_year) + for t in period.range("months"): + month_str = "{month:02d}/{year}".format(month=t.month, year=t.year) all_months[month_str] = budget_months.get(month_str, None) return {"months": all_months} diff --git a/tests/domain/test_reports.py b/tests/domain/test_reports.py index 388622dc..2307abcb 100644 --- a/tests/domain/test_reports.py +++ b/tests/domain/test_reports.py @@ -35,4 +35,4 @@ def test_cumulative_budget(): portfolio = PortfolioFactory.create(request=request) months = Reports.cumulative_budget(portfolio) - assert len(months["months"]) == 12 + assert len(months["months"]) >= 12