Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

comparing querysets in django TestCase

I have a very simple view as follows

def simple_view(request):     documents = request.user.document_set.all()     return render(request, 'simple.html', {'documents': documents}) 

To test the above view in my test case i have the following method which errors out.

Class SomeTestCase(TestCase):     # ...     def test_simple_view(self):         # ... some other checks         docset = self.resonse.context['documents']         self.assertTrue(self.user.document_set.all() == docset) # This line raises an error     # ... 

The error i get is AssertionError: False is not true. I have tried printing both the querysets and both are absolutely identical. Why would it return False when both the objects are identical ? Any Ideas ?

Currently to overcome this, I am using a nasty hack of checking lengths as follows:

ds1, ds2 = self.response.context['documents'], self.user.document_set.all() self.assertTrue(len([x for x in ds1 if x in ds2]) == len(ds1) == len(ds2)) # Makes sure each entry in ds1 exists in ds2 
like image 775
Amyth Avatar asked Apr 17 '13 11:04

Amyth


1 Answers

The queryset objects will not be identical if they are the result of different queries even if they have the same values in their result (compare ds1.query and ds2.query).

If you convert the query set to a list first, you should be able to do a normal comparison (assuming they have the same sort order of course):

self.assertEqual(list(ds1), list(ds2)) 
like image 54
Matti John Avatar answered Oct 29 '22 01:10

Matti John