def test_return_raw_queryset_if_form_is_invalid(self):
queryset = Mock()
params = {'drf__ham__gte_0': 'Yay!', 'drf__ham__gte_1': None, 'drf__ham__lte_0': None, 'drf__ham__lte_1': None}
filter_ = DateTimeRangeFilter('spam', self.request, params, Mock(), Mock(), 'ham')
return_value = filter_.queryset(self.request, queryset)
self.assertEqual(return_value, queryset)
def test_queryset_ignore_null_fields(self):
queryset = Mock()
params = {'drf__ham__gte_0': None, 'drf__ham__gte_1': None, 'drf__ham__lte_0': None, 'drf__ham__lte_1': None}
filter_ = DateTimeRangeFilter('spam', self.request, params, Mock(), Mock(), 'ham')
return_value = filter_.queryset(self.request, queryset)
self.assertEqual(return_value, queryset.filter.return_value)
self.assertItemsEqual([call()], queryset.filter.call_args_list)
def test_queryset_filters_by_date_to(self):
queryset = Mock()
data_end = timezone.make_aware(datetime(2014, 1, 2, 3, 4, 5), timezone.get_current_timezone())
params = {'drf__ham__lte_0': data_end.strftime('%Y-%m-%d'), 'drf__ham__lte_1': data_end.strftime('%H:%M:%S'),
'drf__ham__gte_0': None, 'drf__ham__gte_1': None}
filter_ = DateTimeRangeFilter('spam', self.request, params, Mock(), Mock(), 'ham')
return_value = filter_.queryset(self.request, queryset)
self.assertEqual(return_value, queryset.filter.return_value)
self.assertEqual(data_end, queryset.filter.call_args_list[0][1]['ham__lte'])
def test_queryset_filters_by_date_from(self):
queryset = Mock()
params = {'drf__ham__gte_0': '2014-01-02', 'drf__ham__gte_1': '03:04:05',
'drf__ham__lte_0': None, 'drf__ham__lte_1': None}
filter_ = DateTimeRangeFilter('spam', self.request, params, Mock(), Mock(), 'ham')
return_value = filter_.queryset(self.request, queryset)
timezoned_expected_result = timezone.make_aware(datetime(2014, 1, 2, 3, 4, 5), timezone.get_current_timezone())
self.assertEqual(return_value, queryset.filter.return_value)
self.assertItemsEqual([call(ham__gte=timezoned_expected_result)], queryset.filter.call_args_list)
def test_return_raw_queryset_if_form_is_invalid(self):
queryset = Mock()
params = {
'drf__ham__gte_0': 'Yay!',
'drf__ham__gte_1': None,
'drf__ham__lte_0': None,
'drf__ham__lte_1': None
}
filter_ = DateTimeRangeFilter('spam', self.request, params, Mock(),
Mock(), 'ham')
return_value = filter_.queryset(self.request, queryset)
self.assertEqual(return_value, queryset)
def test_queryset_ignore_null_fields(self):
queryset = Mock()
params = {
'drf__ham__gte_0': None,
'drf__ham__gte_1': None,
'drf__ham__lte_0': None,
'drf__ham__lte_1': None
}
filter_ = DateTimeRangeFilter('spam', self.request, params, Mock(),
Mock(), 'ham')
return_value = filter_.queryset(self.request, queryset)
self.assertEqual(return_value, queryset.filter.return_value)
self.assertCountEqual([call()], queryset.filter.call_args_list)
def test_queryset_filters_by_date_to(self):
queryset = Mock()
data_end = timezone.make_aware(datetime(2014, 1, 2, 3, 4, 5),
timezone.get_current_timezone())
params = {
'drf__ham__lte_0': data_end.strftime('%Y-%m-%d'),
'drf__ham__lte_1': data_end.strftime('%H:%M:%S'),
'drf__ham__gte_0': None,
'drf__ham__gte_1': None
}
filter_ = DateTimeRangeFilter('spam', self.request, params, Mock(),
Mock(), 'ham')
return_value = filter_.queryset(self.request, queryset)
self.assertEqual(return_value, queryset.filter.return_value)
self.assertEqual(data_end,
queryset.filter.call_args_list[0][1]['ham__lte'])
def test_queryset_filters_by_date_from(self):
queryset = Mock()
params = {
'drf__ham__gte_0': '2014-01-02',
'drf__ham__gte_1': '03:04:05',
'drf__ham__lte_0': None,
'drf__ham__lte_1': None
}
filter_ = DateTimeRangeFilter('spam', self.request, params, Mock(),
Mock(), 'ham')
return_value = filter_.queryset(self.request, queryset)
timezoned_expected_result = timezone.make_aware(
datetime(2014, 1, 2, 3, 4, 5), timezone.get_current_timezone())
self.assertEqual(return_value, queryset.filter.return_value)
self.assertCountEqual([call(ham__gte=timezoned_expected_result)],
queryset.filter.call_args_list)
