summaryrefslogtreecommitdiff
path: root/tests/custom_managers
diff options
context:
space:
mode:
authorLoïc Bistuer <loic.bistuer@gmail.com>2016-04-17 18:55:55 +0700
committerLoïc Bistuer <loic.bistuer@gmail.com>2016-05-17 12:07:22 +0700
commited0ff913c648b16c4471fc9a9441d1ee48cb5420 (patch)
treebf9cd3fad9f4c9abbec1967817842fdacad0938a /tests/custom_managers
parent3a47d42fa33012b2156bf04058d933df6b3082d2 (diff)
Fixed #10506, #13793, #14891, #25201 -- Introduced new APIs to specify models' default and base managers.
This deprecates use_for_related_fields. Old API: class CustomManager(models.Model): use_for_related_fields = True class Model(models.Model): custom_manager = CustomManager() New API: class Model(models.Model): custom_manager = CustomManager() class Meta: base_manager_name = 'custom_manager' Refs #20932, #25897. Thanks Carl Meyer for the guidance throughout this work. Thanks Tim Graham for writing the docs.
Diffstat (limited to 'tests/custom_managers')
-rw-r--r--tests/custom_managers/models.py12
-rw-r--r--tests/custom_managers/tests.py33
2 files changed, 43 insertions, 2 deletions
diff --git a/tests/custom_managers/models.py b/tests/custom_managers/models.py
index 22ba66cb8f..51773b18b8 100644
--- a/tests/custom_managers/models.py
+++ b/tests/custom_managers/models.py
@@ -172,6 +172,18 @@ class Car(models.Model):
return self.name
+class FastCarAsBase(Car):
+ class Meta:
+ proxy = True
+ base_manager_name = 'fast_cars'
+
+
+class FastCarAsDefault(Car):
+ class Meta:
+ proxy = True
+ default_manager_name = 'fast_cars'
+
+
class RestrictedManager(models.Manager):
def get_queryset(self):
return super(RestrictedManager, self).get_queryset().filter(is_public=True)
diff --git a/tests/custom_managers/tests.py b/tests/custom_managers/tests.py
index ee1cc141a5..b7a433ef3b 100644
--- a/tests/custom_managers/tests.py
+++ b/tests/custom_managers/tests.py
@@ -6,8 +6,9 @@ from django.utils import six
from .models import (
Book, Car, CustomManager, CustomQuerySet, DeconstructibleCustomManager,
- FunPerson, OneToOneRestrictedModel, Person, PersonFromAbstract,
- PersonManager, PublishedBookManager, RelatedModel, RestrictedModel,
+ FastCarAsBase, FastCarAsDefault, FunPerson, OneToOneRestrictedModel,
+ Person, PersonFromAbstract, PersonManager, PublishedBookManager,
+ RelatedModel, RestrictedModel,
)
@@ -558,6 +559,34 @@ class TestCars(TestCase):
],
lambda c: c.name
)
+ # explicit default manager
+ self.assertQuerysetEqual(
+ FastCarAsDefault.cars.order_by("name"), [
+ "Corvette",
+ "Neon",
+ ],
+ lambda c: c.name
+ )
+ self.assertQuerysetEqual(
+ FastCarAsDefault._default_manager.all(), [
+ "Corvette",
+ ],
+ lambda c: c.name
+ )
+ # explicit base manager
+ self.assertQuerysetEqual(
+ FastCarAsBase.cars.order_by("name"), [
+ "Corvette",
+ "Neon",
+ ],
+ lambda c: c.name
+ )
+ self.assertQuerysetEqual(
+ FastCarAsBase._base_manager.all(), [
+ "Corvette",
+ ],
+ lambda c: c.name
+ )
class CustomManagersRegressTestCase(TestCase):