def test_get_datespan_filter(self):
form_data = {'date_range': '2015-06-25 to 2016-02-19'}
form = FilterFormESExportDownloadForm(self.project, pytz.utc, form_data)
self.assertTrue(form.is_valid())
datespan_filter = form._get_datespan_filter()
self.assertEqual(datespan_filter.lt, datetime.datetime(2016, 2, 20, tzinfo=pytz.utc))
self.assertEqual(datespan_filter.gte, datetime.datetime(2015, 6, 25, tzinfo=pytz.utc))
self.assertEqual(datespan_filter.lte, None)
self.assertEqual(datespan_filter.gt, None)
def test_get_group_filter(self):
"""
Confirm that FilterFormESExportDownloadForm._get_group_filter() returns
a filter with the correct group_id and correct base_filter.
"""
form_data = {
'type_or_group': 'group',
'group': 'some_group_id',
'date_range': '2015-06-25 to 2016-02-19',
}
form = FilterFormESExportDownloadForm(self.project, pytz.utc, form_data)
self.assertTrue(form.is_valid(), "Form had the following errors: {}".format(form.errors))
group_filter = form._get_group_filter()
self.assertEqual(group_filter.group_id, 'some_group_id')
self.assertEqual(group_filter.base_filter, FormSubmittedByFilter)
