def test_clean_report_type(self):
factories.ReportTypeFactory(
report_type='Q1',
report_type_full='April Quarterly {One of 4 valid Report Codes on Form 5, RptCode}'
)
report_date = factories.ReportDateFactory(
report_type='Q1',
due_date=datetime.datetime(2015, 1, 2),
)
db.session.flush()
assert report_date.report_type_full == 'April Quarterly'
def test_reporting_dates_filters(self):
factories.ReportTypeFactory(report_type='YE', report_type_full='Year End')
factories.ReportDateFactory(due_date=datetime.datetime(2014, 1, 2))
factories.ReportDateFactory(report_year=2015)
factories.ReportDateFactory(report_type='YE')
factories.ReportDateFactory(create_date=datetime.datetime(2014, 3, 2))
factories.ReportDateFactory(update_date=datetime.datetime(2014, 4, 2))
filter_fields = (
('due_date', '2014-01-02'),
('report_year', 2015),
('report_type', 'YE'),
('min_create_date', '2014-03-02'),
('max_update_date', '2014-04-02'),
)
for field, example in filter_fields:
page = api.url_for(ReportingDatesView, **{field: example})
# returns at least one result
results = self._results(page)
assert len(results) > 0