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