def test_get_queryset(self):
"""
Test :py:meth:`.MultipleObjectMixin.get_queryset`.
"""
mixin = MultipleObjectMixin()
mixin.document_class = Mock()
self.assertEqual(mixin.document_class.objects, mixin.get_queryset())
def test_get_queryset(self):
"""
Test :py:meth:`.MultipleObjectMixin.get_queryset`.
"""
mixin = MultipleObjectMixin()
mixin.document_class = Mock()
self.assertEqual(mixin.document_class.objects, mixin.get_queryset())
def test_get_filtered_queryset_no_filters(self):
"""
Test :py:meth:`~.MultipleObjectMixin.get_filtered_queryset`.
This tests the result without filters.
"""
mixin = MultipleObjectMixin()
mixin.get_filter_fields = Mock(return_value={})
mixin.get_queryset = Mock()
self.assertEqual(
mixin.get_queryset.return_value, mixin.get_filtered_queryset())
def test_get_filtered_queryset_no_filters(self):
"""
Test :py:meth:`~.MultipleObjectMixin.get_filtered_queryset`.
This tests the result without filters.
"""
mixin = MultipleObjectMixin()
mixin.get_filter_fields = Mock(return_value={})
mixin.get_queryset = Mock()
self.assertEqual(mixin.get_queryset.return_value,
mixin.get_filtered_queryset())
def test_get_filtered_queryset(self):
"""
Test :py:meth:`~.MultipleObjectMixin.get_filtered_queryset`.
"""
queryset = Mock()
mixin = MultipleObjectMixin()
mixin.get_filter_fields = Mock(
return_value={'foo': 'bar', 'bar': 'foo'})
mixin.get_queryset = Mock(return_value=queryset)
self.assertEqual(
queryset.filter.return_value, mixin.get_filtered_queryset())
queryset.filter.assert_called_once_with(foo='bar', bar='foo')
def test_get_filtered_queryset(self):
"""
Test :py:meth:`~.MultipleObjectMixin.get_filtered_queryset`.
"""
queryset = Mock()
mixin = MultipleObjectMixin()
mixin.get_filter_fields = Mock(return_value={
'foo': 'bar',
'bar': 'foo'
})
mixin.get_queryset = Mock(return_value=queryset)
self.assertEqual(queryset.filter.return_value,
mixin.get_filtered_queryset())
queryset.filter.assert_called_once_with(foo='bar', bar='foo')
