queryset so sánh trong django testcase

Question

tôi có một cái nhìn rất đơn giản như sau

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

Để kiểm tra xem trên trong trường hợp thử nghiệm của tôi tôi có phương pháp sau đây mà lỗi ra.

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 
    # ... 

Lỗi tôi nhận được là AssertionError: False is not true. Tôi đã thử in cả bộ truy vấn và cả hai đều hoàn toàn giống nhau. Tại sao nó trả lại False khi cả hai đối tượng giống hệt nhau? Bất kỳ ý tưởng?

Hiện nay để khắc phục điều này, tôi đang sử dụng một hack khó chịu có độ dài kiểm tra như sau:

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 

Answer 1

Đối tượng queryset sẽ không giống hệt nhau nếu họ là kết quả của truy vấn khác nhau ngay cả khi họ có cùng giá trị trong kết quả của họ (so sánh ds1.query và ds2.query).

Nếu bạn chuyển đổi các truy vấn thiết lập một danh sách đầu tiên, bạn sẽ có thể làm một so sánh bình thường (giả sử họ có thứ tự sắp xếp cùng tất nhiên):

self.assertEqual(list(ds1), list(ds2)) 

Answer 2

Tìm thấy một giải pháp. Chúng tôi cần chuyển đổi Querysets thành danh sách được sắp xếp trước khi chúng tôi có thể so sánh chúng. Một cái gì đó như sau.

Class SomeTestCase(TestCase): 
    # ... 
    def test_simple_view(self): 
     # ... some other checks 
     docset1 = self.resonse.context['documents'] 
     docset2 = self.user.document_set.all() 
     self.assertTrue(list(sorted(docset1)) == len(sorted(docset))) 
    # ... 

Answer 3

thay thế này không cần sắp xếp:

self.assertQuerysetEqual(qs1, list(qs2), ordered=False) 

Xem assert reference.

Lưu ý: Chỉ dành cho django 1.4+.