File: utils.py

package info (click to toggle)
django-filter 2.4.0-1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 1,396 kB
  • sloc: python: 7,483; javascript: 7,213; makefile: 144
file content (21 lines) | stat: -rw-r--r-- 382 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
 
from unittest import mock

from django.db import models


class QuerySet(models.QuerySet):

    def __bool__(self):
        return True


class MockQuerySet:
    """
    Generate a mock that is suitably similar to a QuerySet
    """

    def __new__(self):
        m = mock.Mock(spec_set=QuerySet())
        m.filter.return_value = m
        m.all.return_value = m
        return m