def test_open_report_filtering_by_date_range(self):
BasketFactory.create(status=Basket.OPEN)
data = {
'start_date': datetime.date(2020, 5, 1),
'end_date': datetime.date(2020, 6, 1),
'formatter': 'CSV'
}
generator = OpenBasketReportGenerator(**data)
assert generator.queryset.count() == 1
def test_submitted_report_filtering_by_date_range(self):
date_submitted = datetime.datetime(2020, 7, 3).replace(tzinfo=utc)
BasketFactory.create(status=Basket.SUBMITTED, date_submitted=date_submitted)
data = {
'start_date': datetime.date(2020, 7, 1),
'end_date': datetime.date(2020, 8, 1),
'formatter': 'CSV'
}
generator = SubmittedBasketReportGenerator(**data)
assert generator.queryset.count() == 1
