diff options
| author | Loïc Bistuer <loic.bistuer@gmail.com> | 2016-04-17 18:55:55 +0700 |
|---|---|---|
| committer | Loïc Bistuer <loic.bistuer@gmail.com> | 2016-05-17 12:07:22 +0700 |
| commit | ed0ff913c648b16c4471fc9a9441d1ee48cb5420 (patch) | |
| tree | bf9cd3fad9f4c9abbec1967817842fdacad0938a /tests/custom_managers | |
| parent | 3a47d42fa33012b2156bf04058d933df6b3082d2 (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.py | 12 | ||||
| -rw-r--r-- | tests/custom_managers/tests.py | 33 |
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): |
