1from unittest import mock 2 3from django.db import models 4 5 6class QuerySet(models.QuerySet): 7 8 def __bool__(self): 9 return True 10 11 12class MockQuerySet: 13 """ 14 Generate a mock that is suitably similar to a QuerySet 15 """ 16 17 def __new__(self): 18 m = mock.Mock(spec_set=QuerySet()) 19 m.filter.return_value = m 20 m.all.return_value = m 21 return m 22