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