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