|
@@ -411,3 +411,23 @@ class GetContextDataTest(unittest.TestCase):
|
|
|
# test that kwarg overrides values assigned higher up
|
|
|
context = test_view.get_context_data(test_name='test_value')
|
|
|
self.assertEqual(context['test_name'], 'test_value')
|
|
|
+
|
|
|
+
|
|
|
+class UseMultipleObjectMixinTest(unittest.TestCase):
|
|
|
+ rf = RequestFactory()
|
|
|
+
|
|
|
+ def test_use_queryset_from_view(self):
|
|
|
+ test_view = views.CustomMultipleObjectMixinView()
|
|
|
+ test_view.get(self.rf.get('/'))
|
|
|
+ # Don't pass queryset as argument
|
|
|
+ context = test_view.get_context_data()
|
|
|
+ self.assertEqual(context['object_list'], test_view.queryset)
|
|
|
+
|
|
|
+ def test_overwrite_queryset(self):
|
|
|
+ test_view = views.CustomMultipleObjectMixinView()
|
|
|
+ test_view.get(self.rf.get('/'))
|
|
|
+ queryset = [{'name': 'Lennon'}, {'name': 'Ono'}]
|
|
|
+ self.assertNotEqual(test_view.queryset, queryset)
|
|
|
+ # Overwrite the view's queryset with queryset from kwarg
|
|
|
+ context = test_view.get_context_data(object_list=queryset)
|
|
|
+ self.assertEqual(context['object_list'], queryset)
|