summaryrefslogtreecommitdiff
path: root/tests/custom_managers
diff options
context:
space:
mode:
authordjango-bot <ops@djangoproject.com>2022-02-03 20:24:19 +0100
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2022-02-07 20:37:05 +0100
commit9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch)
treef0506b668a013d0063e5fba3dbf4863b466713ba /tests/custom_managers
parentf68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff)
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/custom_managers')
-rw-r--r--tests/custom_managers/models.py67
-rw-r--r--tests/custom_managers/tests.py401
2 files changed, 298 insertions, 170 deletions
diff --git a/tests/custom_managers/models.py b/tests/custom_managers/models.py
index 50d4bfb446..56448b45e9 100644
--- a/tests/custom_managers/models.py
+++ b/tests/custom_managers/models.py
@@ -9,9 +9,7 @@ There are two reasons you might want to customize a ``Manager``: to add extra
returns.
"""
-from django.contrib.contenttypes.fields import (
- GenericForeignKey, GenericRelation,
-)
+from django.contrib.contenttypes.fields import GenericForeignKey, GenericRelation
from django.db import models
@@ -27,8 +25,10 @@ class PublishedBookManager(models.Manager):
class AnnotatedBookManager(models.Manager):
def get_queryset(self):
- return super().get_queryset().annotate(
- favorite_avg=models.Avg('favorite_books__favorite_thing_id')
+ return (
+ super()
+ .get_queryset()
+ .annotate(favorite_avg=models.Avg("favorite_books__favorite_thing_id"))
)
@@ -46,10 +46,12 @@ class CustomQuerySet(models.QuerySet):
def optout_public_method(self, *args, **kwargs):
return self.all()
+
optout_public_method.queryset_only = True
def _optin_private_method(self, *args, **kwargs):
return self.all()
+
_optin_private_method.queryset_only = False
@@ -72,12 +74,13 @@ CustomManager = BaseCustomManager.from_queryset(CustomQuerySet)
class CustomInitQuerySet(models.QuerySet):
# QuerySet with an __init__() method that takes an additional argument.
- def __init__(self, custom_optional_arg=None, model=None, query=None, using=None, hints=None):
+ def __init__(
+ self, custom_optional_arg=None, model=None, query=None, using=None, hints=None
+ ):
super().__init__(model=model, query=query, using=using, hints=hints)
class DeconstructibleCustomManager(BaseCustomManager.from_queryset(CustomQuerySet)):
-
def __init__(self, a, b, c=1, d=2):
super().__init__(a)
@@ -97,17 +100,21 @@ class Person(models.Model):
last_name = models.CharField(max_length=30)
fun = models.BooleanField(default=False)
- favorite_book = models.ForeignKey('Book', models.SET_NULL, null=True, related_name='favorite_books')
- favorite_thing_type = models.ForeignKey('contenttypes.ContentType', models.SET_NULL, null=True)
+ favorite_book = models.ForeignKey(
+ "Book", models.SET_NULL, null=True, related_name="favorite_books"
+ )
+ favorite_thing_type = models.ForeignKey(
+ "contenttypes.ContentType", models.SET_NULL, null=True
+ )
favorite_thing_id = models.IntegerField(null=True)
- favorite_thing = GenericForeignKey('favorite_thing_type', 'favorite_thing_id')
+ favorite_thing = GenericForeignKey("favorite_thing_type", "favorite_thing_id")
objects = PersonManager()
fun_people = FunPeopleManager()
boring_people = BoringPeopleManager()
custom_queryset_default_manager = CustomQuerySet.as_manager()
- custom_queryset_custom_manager = CustomManager('hello')
+ custom_queryset_custom_manager = CustomManager("hello")
custom_init_queryset_manager = CustomInitQuerySet.as_manager()
def __str__(self):
@@ -119,14 +126,16 @@ class FunPerson(models.Model):
last_name = models.CharField(max_length=30)
fun = models.BooleanField(default=True)
favorite_book = models.ForeignKey(
- 'Book',
+ "Book",
models.SET_NULL,
null=True,
- related_name='fun_people_favorite_books',
+ related_name="fun_people_favorite_books",
+ )
+ favorite_thing_type = models.ForeignKey(
+ "contenttypes.ContentType", models.SET_NULL, null=True
)
- favorite_thing_type = models.ForeignKey('contenttypes.ContentType', models.SET_NULL, null=True)
favorite_thing_id = models.IntegerField(null=True)
- favorite_thing = GenericForeignKey('favorite_thing_type', 'favorite_thing_id')
+ favorite_thing = GenericForeignKey("favorite_thing_type", "favorite_thing_id")
objects = FunPeopleManager()
@@ -135,24 +144,24 @@ class Book(models.Model):
title = models.CharField(max_length=50)
author = models.CharField(max_length=30)
is_published = models.BooleanField(default=False)
- authors = models.ManyToManyField(Person, related_name='books')
- fun_authors = models.ManyToManyField(FunPerson, related_name='books')
+ authors = models.ManyToManyField(Person, related_name="books")
+ fun_authors = models.ManyToManyField(FunPerson, related_name="books")
favorite_things = GenericRelation(
Person,
- content_type_field='favorite_thing_type',
- object_id_field='favorite_thing_id',
+ content_type_field="favorite_thing_type",
+ object_id_field="favorite_thing_id",
)
fun_people_favorite_things = GenericRelation(
FunPerson,
- content_type_field='favorite_thing_type',
- object_id_field='favorite_thing_id',
+ content_type_field="favorite_thing_type",
+ object_id_field="favorite_thing_id",
)
published_objects = PublishedBookManager()
annotated_objects = AnnotatedBookManager()
class Meta:
- base_manager_name = 'annotated_objects'
+ base_manager_name = "annotated_objects"
class ConfusedBook(models.Model):
@@ -160,14 +169,14 @@ class ConfusedBook(models.Model):
author = models.CharField(max_length=30)
favorite_things = GenericRelation(
Person,
- content_type_field='favorite_thing_type',
- object_id_field='favorite_thing_id',
+ content_type_field="favorite_thing_type",
+ object_id_field="favorite_thing_id",
)
less_favorite_things = GenericRelation(
FunPerson,
- content_type_field='favorite_thing_type',
- object_id_field='favorite_thing_id',
- related_query_name='favorite_things',
+ content_type_field="favorite_thing_type",
+ object_id_field="favorite_thing_id",
+ related_query_name="favorite_things",
)
@@ -187,13 +196,13 @@ class Car(models.Model):
class FastCarAsBase(Car):
class Meta:
proxy = True
- base_manager_name = 'fast_cars'
+ base_manager_name = "fast_cars"
class FastCarAsDefault(Car):
class Meta:
proxy = True
- default_manager_name = 'fast_cars'
+ default_manager_name = "fast_cars"
class RestrictedManager(models.Manager):
diff --git a/tests/custom_managers/tests.py b/tests/custom_managers/tests.py
index b397816668..6b90a6ce37 100644
--- a/tests/custom_managers/tests.py
+++ b/tests/custom_managers/tests.py
@@ -2,39 +2,50 @@ from django.db import models
from django.test import TestCase
from .models import (
- Book, Car, ConfusedBook, CustomManager, CustomQuerySet,
- DeconstructibleCustomManager, FastCarAsBase, FastCarAsDefault, FunPerson,
- OneToOneRestrictedModel, Person, PersonFromAbstract, PersonManager,
- PublishedBookManager, RelatedModel, RestrictedModel,
+ Book,
+ Car,
+ ConfusedBook,
+ CustomManager,
+ CustomQuerySet,
+ DeconstructibleCustomManager,
+ FastCarAsBase,
+ FastCarAsDefault,
+ FunPerson,
+ OneToOneRestrictedModel,
+ Person,
+ PersonFromAbstract,
+ PersonManager,
+ PublishedBookManager,
+ RelatedModel,
+ RestrictedModel,
)
class CustomManagerTests(TestCase):
custom_manager_names = [
- 'custom_queryset_default_manager',
- 'custom_queryset_custom_manager',
+ "custom_queryset_default_manager",
+ "custom_queryset_custom_manager",
]
@classmethod
def setUpTestData(cls):
cls.b1 = Book.published_objects.create(
- title="How to program", author="Rodney Dangerfield", is_published=True)
+ title="How to program", author="Rodney Dangerfield", is_published=True
+ )
cls.b2 = Book.published_objects.create(
- title="How to be smart", author="Albert Einstein", is_published=False)
+ title="How to be smart", author="Albert Einstein", is_published=False
+ )
cls.p1 = Person.objects.create(first_name="Bugs", last_name="Bunny", fun=True)
- cls.droopy = Person.objects.create(first_name="Droopy", last_name="Dog", fun=False)
+ cls.droopy = Person.objects.create(
+ first_name="Droopy", last_name="Dog", fun=False
+ )
def test_custom_manager_basic(self):
"""
Test a custom Manager method.
"""
- self.assertQuerysetEqual(
- Person.objects.get_fun_people(), [
- "Bugs Bunny"
- ],
- str
- )
+ self.assertQuerysetEqual(Person.objects.get_fun_people(), ["Bugs Bunny"], str)
def test_queryset_copied_to_default(self):
"""
@@ -58,7 +69,10 @@ class CustomManagerTests(TestCase):
# Methods with queryset_only=False are copied even if they are private.
manager._optin_private_method()
# Methods with queryset_only=True aren't copied even if they are public.
- msg = "%r object has no attribute 'optout_public_method'" % manager.__class__.__name__
+ msg = (
+ "%r object has no attribute 'optout_public_method'"
+ % manager.__class__.__name__
+ )
with self.assertRaisesMessage(AttributeError, msg):
manager.optout_public_method()
@@ -74,7 +88,7 @@ class CustomManagerTests(TestCase):
self.assertIs(queryset._filter_CustomQuerySet, True)
# Specialized querysets inherit from our custom queryset.
- queryset = manager.values_list('first_name', flat=True).filter()
+ queryset = manager.values_list("first_name", flat=True).filter()
self.assertEqual(list(queryset), ["Bugs"])
self.assertIs(queryset._filter_CustomQuerySet, True)
@@ -86,7 +100,7 @@ class CustomManagerTests(TestCase):
"""
The custom manager __init__() argument has been set.
"""
- self.assertEqual(Person.custom_queryset_custom_manager.init_arg, 'hello')
+ self.assertEqual(Person.custom_queryset_custom_manager.init_arg, "hello")
def test_manager_attributes(self):
"""
@@ -127,20 +141,30 @@ class CustomManagerTests(TestCase):
Custom managers respond to usual filtering methods
"""
self.assertQuerysetEqual(
- Book.published_objects.all(), [
+ Book.published_objects.all(),
+ [
"How to program",
],
- lambda b: b.title
+ lambda b: b.title,
)
def test_fk_related_manager(self):
- Person.objects.create(first_name="Bugs", last_name="Bunny", fun=True, favorite_book=self.b1)
- Person.objects.create(first_name="Droopy", last_name="Dog", fun=False, favorite_book=self.b1)
- FunPerson.objects.create(first_name="Bugs", last_name="Bunny", fun=True, favorite_book=self.b1)
- FunPerson.objects.create(first_name="Droopy", last_name="Dog", fun=False, favorite_book=self.b1)
+ Person.objects.create(
+ first_name="Bugs", last_name="Bunny", fun=True, favorite_book=self.b1
+ )
+ Person.objects.create(
+ first_name="Droopy", last_name="Dog", fun=False, favorite_book=self.b1
+ )
+ FunPerson.objects.create(
+ first_name="Bugs", last_name="Bunny", fun=True, favorite_book=self.b1
+ )
+ FunPerson.objects.create(
+ first_name="Droopy", last_name="Dog", fun=False, favorite_book=self.b1
+ )
self.assertQuerysetEqual(
- self.b1.favorite_books.order_by('first_name').all(), [
+ self.b1.favorite_books.order_by("first_name").all(),
+ [
"Bugs",
"Droopy",
],
@@ -148,21 +172,24 @@ class CustomManagerTests(TestCase):
ordered=False,
)
self.assertQuerysetEqual(
- self.b1.fun_people_favorite_books.all(), [
+ self.b1.fun_people_favorite_books.all(),
+ [
"Bugs",
],
lambda c: c.first_name,
ordered=False,
)
self.assertQuerysetEqual(
- self.b1.favorite_books(manager='boring_people').all(), [
+ self.b1.favorite_books(manager="boring_people").all(),
+ [
"Droopy",
],
lambda c: c.first_name,
ordered=False,
)
self.assertQuerysetEqual(
- self.b1.favorite_books(manager='fun_people').all(), [
+ self.b1.favorite_books(manager="fun_people").all(),
+ [
"Bugs",
],
lambda c: c.first_name,
@@ -171,16 +198,25 @@ class CustomManagerTests(TestCase):
def test_fk_related_manager_reused(self):
self.assertIs(self.b1.favorite_books, self.b1.favorite_books)
- self.assertIn('favorite_books', self.b1._state.related_managers_cache)
+ self.assertIn("favorite_books", self.b1._state.related_managers_cache)
def test_gfk_related_manager(self):
- Person.objects.create(first_name="Bugs", last_name="Bunny", fun=True, favorite_thing=self.b1)
- Person.objects.create(first_name="Droopy", last_name="Dog", fun=False, favorite_thing=self.b1)
- FunPerson.objects.create(first_name="Bugs", last_name="Bunny", fun=True, favorite_thing=self.b1)
- FunPerson.objects.create(first_name="Droopy", last_name="Dog", fun=False, favorite_thing=self.b1)
+ Person.objects.create(
+ first_name="Bugs", last_name="Bunny", fun=True, favorite_thing=self.b1
+ )
+ Person.objects.create(
+ first_name="Droopy", last_name="Dog", fun=False, favorite_thing=self.b1
+ )
+ FunPerson.objects.create(
+ first_name="Bugs", last_name="Bunny", fun=True, favorite_thing=self.b1
+ )
+ FunPerson.objects.create(
+ first_name="Droopy", last_name="Dog", fun=False, favorite_thing=self.b1
+ )
self.assertQuerysetEqual(
- self.b1.favorite_things.all(), [
+ self.b1.favorite_things.all(),
+ [
"Bugs",
"Droopy",
],
@@ -188,21 +224,24 @@ class CustomManagerTests(TestCase):
ordered=False,
)
self.assertQuerysetEqual(
- self.b1.fun_people_favorite_things.all(), [
+ self.b1.fun_people_favorite_things.all(),
+ [
"Bugs",
],
lambda c: c.first_name,
ordered=False,
)
self.assertQuerysetEqual(
- self.b1.favorite_things(manager='boring_people').all(), [
+ self.b1.favorite_things(manager="boring_people").all(),
+ [
"Droopy",
],
lambda c: c.first_name,
ordered=False,
)
self.assertQuerysetEqual(
- self.b1.favorite_things(manager='fun_people').all(), [
+ self.b1.favorite_things(manager="fun_people").all(),
+ [
"Bugs",
],
lambda c: c.first_name,
@@ -215,14 +254,14 @@ class CustomManagerTests(TestCase):
self.b1.fun_people_favorite_things,
)
self.assertIn(
- 'fun_people_favorite_things',
+ "fun_people_favorite_things",
self.b1._state.related_managers_cache,
)
def test_gfk_related_manager_not_reused_when_alternate(self):
self.assertIsNot(
- self.b1.favorite_things(manager='fun_people'),
- self.b1.favorite_things(manager='fun_people'),
+ self.b1.favorite_things(manager="fun_people"),
+ self.b1.favorite_things(manager="fun_people"),
)
def test_gfk_related_manager_no_overlap_when_not_hidden(self):
@@ -232,36 +271,43 @@ class CustomManagerTests(TestCase):
cause those separate managers to clash.
"""
book = ConfusedBook.objects.create(
- title='How to program', author='Rodney Dangerfield',
+ title="How to program",
+ author="Rodney Dangerfield",
)
person = Person.objects.create(
- first_name='Bugs', last_name='Bunny', fun=True, favorite_thing=book,
+ first_name="Bugs",
+ last_name="Bunny",
+ fun=True,
+ favorite_thing=book,
)
fun_person = FunPerson.objects.create(
- first_name='Droopy', last_name='Dog', fun=False, favorite_thing=book,
+ first_name="Droopy",
+ last_name="Dog",
+ fun=False,
+ favorite_thing=book,
)
# The managers don't collide in the internal cache.
self.assertIsNot(book.favorite_things, book.less_favorite_things)
self.assertIs(book.favorite_things, book.favorite_things)
self.assertIs(book.less_favorite_things, book.less_favorite_things)
# Both managers are cached separately despite the collision in names.
- self.assertIn('favorite_things', book._state.related_managers_cache)
- self.assertIn('less_favorite_things', book._state.related_managers_cache)
+ self.assertIn("favorite_things", book._state.related_managers_cache)
+ self.assertIn("less_favorite_things", book._state.related_managers_cache)
# "less_favorite_things" isn't available as a reverse related manager,
# so never ends up in the cache.
self.assertQuerysetEqual(fun_person.favorite_things.all(), [book])
with self.assertRaises(AttributeError):
fun_person.less_favorite_things
- self.assertIn('favorite_things', fun_person._state.related_managers_cache)
+ self.assertIn("favorite_things", fun_person._state.related_managers_cache)
self.assertNotIn(
- 'less_favorite_things',
+ "less_favorite_things",
fun_person._state.related_managers_cache,
)
# The GenericRelation doesn't exist for Person, only FunPerson, so the
# exception prevents the cache from being polluted.
with self.assertRaises(AttributeError):
person.favorite_things
- self.assertNotIn('favorite_things', person._state.related_managers_cache)
+ self.assertNotIn("favorite_things", person._state.related_managers_cache)
def test_m2m_related_manager(self):
bugs = Person.objects.create(first_name="Bugs", last_name="Bunny", fun=True)
@@ -270,11 +316,14 @@ class CustomManagerTests(TestCase):
self.b1.authors.add(droopy)
bugs = FunPerson.objects.create(first_name="Bugs", last_name="Bunny", fun=True)
self.b1.fun_authors.add(bugs)
- droopy = FunPerson.objects.create(first_name="Droopy", last_name="Dog", fun=False)
+ droopy = FunPerson.objects.create(
+ first_name="Droopy", last_name="Dog", fun=False
+ )
self.b1.fun_authors.add(droopy)
self.assertQuerysetEqual(
- self.b1.authors.order_by('first_name').all(), [
+ self.b1.authors.order_by("first_name").all(),
+ [
"Bugs",
"Droopy",
],
@@ -282,21 +331,24 @@ class CustomManagerTests(TestCase):
ordered=False,
)
self.assertQuerysetEqual(
- self.b1.fun_authors.order_by('first_name').all(), [
+ self.b1.fun_authors.order_by("first_name").all(),
+ [
"Bugs",
],
lambda c: c.first_name,
ordered=False,
)
self.assertQuerysetEqual(
- self.b1.authors(manager='boring_people').all(), [
+ self.b1.authors(manager="boring_people").all(),
+ [
"Droopy",
],
lambda c: c.first_name,
ordered=False,
)
self.assertQuerysetEqual(
- self.b1.authors(manager='fun_people').all(), [
+ self.b1.authors(manager="fun_people").all(),
+ [
"Bugs",
],
lambda c: c.first_name,
@@ -305,21 +357,26 @@ class CustomManagerTests(TestCase):
def test_m2m_related_forward_manager_reused(self):
self.assertIs(self.b1.authors, self.b1.authors)
- self.assertIn('authors', self.b1._state.related_managers_cache)
+ self.assertIn("authors", self.b1._state.related_managers_cache)
def test_m2m_related_revers_manager_reused(self):
- bugs = Person.objects.create(first_name='Bugs', last_name='Bunny')
+ bugs = Person.objects.create(first_name="Bugs", last_name="Bunny")
self.b1.authors.add(bugs)
self.assertIs(bugs.books, bugs.books)
- self.assertIn('books', bugs._state.related_managers_cache)
+ self.assertIn("books", bugs._state.related_managers_cache)
def test_removal_through_default_fk_related_manager(self, bulk=True):
- bugs = FunPerson.objects.create(first_name="Bugs", last_name="Bunny", fun=True, favorite_book=self.b1)
- droopy = FunPerson.objects.create(first_name="Droopy", last_name="Dog", fun=False, favorite_book=self.b1)
+ bugs = FunPerson.objects.create(
+ first_name="Bugs", last_name="Bunny", fun=True, favorite_book=self.b1
+ )
+ droopy = FunPerson.objects.create(
+ first_name="Droopy", last_name="Dog", fun=False, favorite_book=self.b1
+ )
self.b1.fun_people_favorite_books.remove(droopy, bulk=bulk)
self.assertQuerysetEqual(
- FunPerson._base_manager.filter(favorite_book=self.b1), [
+ FunPerson._base_manager.filter(favorite_book=self.b1),
+ [
"Bugs",
"Droopy",
],
@@ -329,7 +386,8 @@ class CustomManagerTests(TestCase):
self.b1.fun_people_favorite_books.remove(bugs, bulk=bulk)
self.assertQuerysetEqual(
- FunPerson._base_manager.filter(favorite_book=self.b1), [
+ FunPerson._base_manager.filter(favorite_book=self.b1),
+ [
"Droopy",
],
lambda c: c.first_name,
@@ -340,7 +398,8 @@ class CustomManagerTests(TestCase):
self.b1.fun_people_favorite_books.clear(bulk=bulk)
self.assertQuerysetEqual(
- FunPerson._base_manager.filter(favorite_book=self.b1), [
+ FunPerson._base_manager.filter(favorite_book=self.b1),
+ [
"Droopy",
],
lambda c: c.first_name,
@@ -351,23 +410,28 @@ class CustomManagerTests(TestCase):
self.test_removal_through_default_fk_related_manager(bulk=False)
def test_removal_through_specified_fk_related_manager(self, bulk=True):
- Person.objects.create(first_name="Bugs", last_name="Bunny", fun=True, favorite_book=self.b1)
- droopy = Person.objects.create(first_name="Droopy", last_name="Dog", fun=False, favorite_book=self.b1)
+ Person.objects.create(
+ first_name="Bugs", last_name="Bunny", fun=True, favorite_book=self.b1
+ )
+ droopy = Person.objects.create(
+ first_name="Droopy", last_name="Dog", fun=False, favorite_book=self.b1
+ )
# The fun manager DOESN'T remove boring people.
- self.b1.favorite_books(manager='fun_people').remove(droopy, bulk=bulk)
+ self.b1.favorite_books(manager="fun_people").remove(droopy, bulk=bulk)
self.assertQuerysetEqual(
- self.b1.favorite_books(manager='boring_people').all(), [
+ self.b1.favorite_books(manager="boring_people").all(),
+ [
"Droopy",
],
lambda c: c.first_name,
ordered=False,
)
# The boring manager DOES remove boring people.
- self.b1.favorite_books(manager='boring_people').remove(droopy, bulk=bulk)
+ self.b1.favorite_books(manager="boring_people").remove(droopy, bulk=bulk)
self.assertQuerysetEqual(
- self.b1.favorite_books(manager='boring_people').all(), [
- ],
+ self.b1.favorite_books(manager="boring_people").all(),
+ [],
lambda c: c.first_name,
ordered=False,
)
@@ -375,17 +439,18 @@ class CustomManagerTests(TestCase):
droopy.save()
# The fun manager ONLY clears fun people.
- self.b1.favorite_books(manager='fun_people').clear(bulk=bulk)
+ self.b1.favorite_books(manager="fun_people").clear(bulk=bulk)
self.assertQuerysetEqual(
- self.b1.favorite_books(manager='boring_people').all(), [
+ self.b1.favorite_books(manager="boring_people").all(),
+ [
"Droopy",
],
lambda c: c.first_name,
ordered=False,
)
self.assertQuerysetEqual(
- self.b1.favorite_books(manager='fun_people').all(), [
- ],
+ self.b1.favorite_books(manager="fun_people").all(),
+ [],
lambda c: c.first_name,
ordered=False,
)
@@ -394,12 +459,19 @@ class CustomManagerTests(TestCase):
self.test_removal_through_specified_fk_related_manager(bulk=False)
def test_removal_through_default_gfk_related_manager(self, bulk=True):
- bugs = FunPerson.objects.create(first_name="Bugs", last_name="Bunny", fun=True, favorite_thing=self.b1)
- droopy = FunPerson.objects.create(first_name="Droopy", last_name="Dog", fun=False, favorite_thing=self.b1)
+ bugs = FunPerson.objects.create(
+ first_name="Bugs", last_name="Bunny", fun=True, favorite_thing=self.b1
+ )
+ droopy = FunPerson.objects.create(
+ first_name="Droopy", last_name="Dog", fun=False, favorite_thing=self.b1
+ )
self.b1.fun_people_favorite_things.remove(droopy, bulk=bulk)
self.assertQuerysetEqual(
- FunPerson._base_manager.order_by('first_name').filter(favorite_thing_id=self.b1.pk), [
+ FunPerson._base_manager.order_by("first_name").filter(
+ favorite_thing_id=self.b1.pk
+ ),
+ [
"Bugs",
"Droopy",
],
@@ -409,7 +481,10 @@ class CustomManagerTests(TestCase):
self.b1.fun_people_favorite_things.remove(bugs, bulk=bulk)
self.assertQuerysetEqual(
- FunPerson._base_manager.order_by('first_name').filter(favorite_thing_id=self.b1.pk), [
+ FunPerson._base_manager.order_by("first_name").filter(
+ favorite_thing_id=self.b1.pk
+ ),
+ [
"Droopy",
],
lambda c: c.first_name,
@@ -420,7 +495,10 @@ class CustomManagerTests(TestCase):
self.b1.fun_people_favorite_things.clear(bulk=bulk)
self.assertQuerysetEqual(
- FunPerson._base_manager.order_by('first_name').filter(favorite_thing_id=self.b1.pk), [
+ FunPerson._base_manager.order_by("first_name").filter(
+ favorite_thing_id=self.b1.pk
+ ),
+ [
"Droopy",
],
lambda c: c.first_name,
@@ -431,13 +509,18 @@ class CustomManagerTests(TestCase):
self.test_removal_through_default_gfk_related_manager(bulk=False)
def test_removal_through_specified_gfk_related_manager(self, bulk=True):
- Person.objects.create(first_name="Bugs", last_name="Bunny", fun=True, favorite_thing=self.b1)
- droopy = Person.objects.create(first_name="Droopy", last_name="Dog", fun=False, favorite_thing=self.b1)
+ Person.objects.create(
+ first_name="Bugs", last_name="Bunny", fun=True, favorite_thing=self.b1
+ )
+ droopy = Person.objects.create(
+ first_name="Droopy", last_name="Dog", fun=False, favorite_thing=self.b1
+ )
# The fun manager DOESN'T remove boring people.
- self.b1.favorite_things(manager='fun_people').remove(droopy, bulk=bulk)
+ self.b1.favorite_things(manager="fun_people").remove(droopy, bulk=bulk)
self.assertQuerysetEqual(
- self.b1.favorite_things(manager='boring_people').all(), [
+ self.b1.favorite_things(manager="boring_people").all(),
+ [
"Droopy",
],
lambda c: c.first_name,
@@ -445,10 +528,10 @@ class CustomManagerTests(TestCase):
)
# The boring manager DOES remove boring people.
- self.b1.favorite_things(manager='boring_people').remove(droopy, bulk=bulk)
+ self.b1.favorite_things(manager="boring_people").remove(droopy, bulk=bulk)
self.assertQuerysetEqual(
- self.b1.favorite_things(manager='boring_people').all(), [
- ],
+ self.b1.favorite_things(manager="boring_people").all(),
+ [],
lambda c: c.first_name,
ordered=False,
)
@@ -456,17 +539,18 @@ class CustomManagerTests(TestCase):
droopy.save()
# The fun manager ONLY clears fun people.
- self.b1.favorite_things(manager='fun_people').clear(bulk=bulk)
+ self.b1.favorite_things(manager="fun_people").clear(bulk=bulk)
self.assertQuerysetEqual(
- self.b1.favorite_things(manager='boring_people').all(), [
+ self.b1.favorite_things(manager="boring_people").all(),
+ [
"Droopy",
],
lambda c: c.first_name,
ordered=False,
)
self.assertQuerysetEqual(
- self.b1.favorite_things(manager='fun_people').all(), [
- ],
+ self.b1.favorite_things(manager="fun_people").all(),
+ [],
lambda c: c.first_name,
ordered=False,
)
@@ -477,12 +561,15 @@ class CustomManagerTests(TestCase):
def test_removal_through_default_m2m_related_manager(self):
bugs = FunPerson.objects.create(first_name="Bugs", last_name="Bunny", fun=True)
self.b1.fun_authors.add(bugs)
- droopy = FunPerson.objects.create(first_name="Droopy", last_name="Dog", fun=False)
+ droopy = FunPerson.objects.create(
+ first_name="Droopy", last_name="Dog", fun=False
+ )
self.b1.fun_authors.add(droopy)
self.b1.fun_authors.remove(droopy)
self.assertQuerysetEqual(
- self.b1.fun_authors.through._default_manager.all(), [
+ self.b1.fun_authors.through._default_manager.all(),
+ [
"Bugs",
"Droopy",
],
@@ -492,7 +579,8 @@ class CustomManagerTests(TestCase):
self.b1.fun_authors.remove(bugs)
self.assertQuerysetEqual(
- self.b1.fun_authors.through._default_manager.all(), [
+ self.b1.fun_authors.through._default_manager.all(),
+ [
"Droopy",
],
lambda c: c.funperson.first_name,
@@ -502,7 +590,8 @@ class CustomManagerTests(TestCase):
self.b1.fun_authors.clear()
self.assertQuerysetEqual(
- self.b1.fun_authors.through._default_manager.all(), [
+ self.b1.fun_authors.through._default_manager.all(),
+ [
"Droopy",
],
lambda c: c.funperson.first_name,
@@ -516,9 +605,10 @@ class CustomManagerTests(TestCase):
self.b1.authors.add(droopy)
# The fun manager DOESN'T remove boring people.
- self.b1.authors(manager='fun_people').remove(droopy)
+ self.b1.authors(manager="fun_people").remove(droopy)
self.assertQuerysetEqual(
- self.b1.authors(manager='boring_people').all(), [
+ self.b1.authors(manager="boring_people").all(),
+ [
"Droopy",
],
lambda c: c.first_name,
@@ -526,27 +616,28 @@ class CustomManagerTests(TestCase):
)
# The boring manager DOES remove boring people.
- self.b1.authors(manager='boring_people').remove(droopy)
+ self.b1.authors(manager="boring_people").remove(droopy)
self.assertQuerysetEqual(
- self.b1.authors(manager='boring_people').all(), [
- ],
+ self.b1.authors(manager="boring_people").all(),
+ [],
lambda c: c.first_name,
ordered=False,
)
self.b1.authors.add(droopy)
# The fun manager ONLY clears fun people.
- self.b1.authors(manager='fun_people').clear()
+ self.b1.authors(manager="fun_people").clear()
self.assertQuerysetEqual(
- self.b1.authors(manager='boring_people').all(), [
+ self.b1.authors(manager="boring_people").all(),
+ [
"Droopy",
],
lambda c: c.first_name,
ordered=False,
)
self.assertQuerysetEqual(
- self.b1.authors(manager='fun_people').all(), [
- ],
+ self.b1.authors(manager="fun_people").all(),
+ [],
lambda c: c.first_name,
ordered=False,
)
@@ -555,7 +646,7 @@ class CustomManagerTests(TestCase):
mgr = models.Manager()
as_manager, mgr_path, qs_path, args, kwargs = mgr.deconstruct()
self.assertFalse(as_manager)
- self.assertEqual(mgr_path, 'django.db.models.manager.Manager')
+ self.assertEqual(mgr_path, "django.db.models.manager.Manager")
self.assertEqual(args, ())
self.assertEqual(kwargs, {})
@@ -563,29 +654,47 @@ class CustomManagerTests(TestCase):
mgr = CustomQuerySet.as_manager()
as_manager, mgr_path, qs_path, args, kwargs = mgr.deconstruct()
self.assertTrue(as_manager)
- self.assertEqual(qs_path, 'custom_managers.models.CustomQuerySet')
+ self.assertEqual(qs_path, "custom_managers.models.CustomQuerySet")
def test_deconstruct_from_queryset(self):
- mgr = DeconstructibleCustomManager('a', 'b')
+ mgr = DeconstructibleCustomManager("a", "b")
as_manager, mgr_path, qs_path, args, kwargs = mgr.deconstruct()
self.assertFalse(as_manager)
- self.assertEqual(mgr_path, 'custom_managers.models.DeconstructibleCustomManager')
- self.assertEqual(args, ('a', 'b',))
+ self.assertEqual(
+ mgr_path, "custom_managers.models.DeconstructibleCustomManager"
+ )
+ self.assertEqual(
+ args,
+ (
+ "a",
+ "b",
+ ),
+ )
self.assertEqual(kwargs, {})
- mgr = DeconstructibleCustomManager('x', 'y', c=3, d=4)
+ mgr = DeconstructibleCustomManager("x", "y", c=3, d=4)
as_manager, mgr_path, qs_path, args, kwargs = mgr.deconstruct()
self.assertFalse(as_manager)
- self.assertEqual(mgr_path, 'custom_managers.models.DeconstructibleCustomManager')
- self.assertEqual(args, ('x', 'y',))
- self.assertEqual(kwargs, {'c': 3, 'd': 4})
+ self.assertEqual(
+ mgr_path, "custom_managers.models.DeconstructibleCustomManager"
+ )
+ self.assertEqual(
+ args,
+ (
+ "x",
+ "y",
+ ),
+ )
+ self.assertEqual(kwargs, {"c": 3, "d": 4})
def test_deconstruct_from_queryset_failing(self):
- mgr = CustomManager('arg')
- msg = ("Could not find manager BaseCustomManagerFromCustomQuerySet in "
- "django.db.models.manager.\n"
- "Please note that you need to inherit from managers you "
- "dynamically generated with 'from_queryset()'.")
+ mgr = CustomManager("arg")
+ msg = (
+ "Could not find manager BaseCustomManagerFromCustomQuerySet in "
+ "django.db.models.manager.\n"
+ "Please note that you need to inherit from managers you "
+ "dynamically generated with 'from_queryset()'."
+ )
with self.assertRaisesMessage(ValueError, msg):
mgr.deconstruct()
@@ -594,15 +703,15 @@ class CustomManagerTests(TestCase):
A custom manager may be defined on an abstract model.
It will be inherited by the abstract model's children.
"""
- PersonFromAbstract.abstract_persons.create(objects='Test')
+ PersonFromAbstract.abstract_persons.create(objects="Test")
self.assertQuerysetEqual(
- PersonFromAbstract.abstract_persons.all(), ["Test"],
+ PersonFromAbstract.abstract_persons.all(),
+ ["Test"],
lambda c: c.objects,
)
class TestCars(TestCase):
-
def test_managers(self):
# Each model class gets a "_default_manager" attribute, which is a
# reference to the first manager defined in the class.
@@ -610,53 +719,60 @@ class TestCars(TestCase):
Car.cars.create(name="Neon", mileage=31, top_speed=100)
self.assertQuerysetEqual(
- Car._default_manager.order_by("name"), [
+ Car._default_manager.order_by("name"),
+ [
"Corvette",
"Neon",
],
- lambda c: c.name
+ lambda c: c.name,
)
self.assertQuerysetEqual(
- Car.cars.order_by("name"), [
+ Car.cars.order_by("name"),
+ [
"Corvette",
"Neon",
],
- lambda c: c.name
+ lambda c: c.name,
)
# alternate manager
self.assertQuerysetEqual(
- Car.fast_cars.all(), [
+ Car.fast_cars.all(),
+ [
"Corvette",
],
- lambda c: c.name
+ lambda c: c.name,
)
# explicit default manager
self.assertQuerysetEqual(
- FastCarAsDefault.cars.order_by("name"), [
+ FastCarAsDefault.cars.order_by("name"),
+ [
"Corvette",
"Neon",
],
- lambda c: c.name
+ lambda c: c.name,
)
self.assertQuerysetEqual(
- FastCarAsDefault._default_manager.all(), [
+ FastCarAsDefault._default_manager.all(),
+ [
"Corvette",
],
- lambda c: c.name
+ lambda c: c.name,
)
# explicit base manager
self.assertQuerysetEqual(
- FastCarAsBase.cars.order_by("name"), [
+ FastCarAsBase.cars.order_by("name"),
+ [
"Corvette",
"Neon",
],
- lambda c: c.name
+ lambda c: c.name,
)
self.assertQuerysetEqual(
- FastCarAsBase._base_manager.all(), [
+ FastCarAsBase._base_manager.all(),
+ [
"Corvette",
],
- lambda c: c.name
+ lambda c: c.name,
)
@@ -681,25 +797,28 @@ class CustomManagersRegressTestCase(TestCase):
manager.
"""
book = Book._base_manager.create(is_published=False)
- Book._base_manager.filter(pk=book.pk).update(title='Hi')
+ Book._base_manager.filter(pk=book.pk).update(title="Hi")
book.refresh_from_db()
- self.assertEqual(book.title, 'Hi')
+ self.assertEqual(book.title, "Hi")
def test_save_clears_annotations_from_base_manager(self):
"""Model.save() clears annotations from the base manager."""
- self.assertEqual(Book._meta.base_manager.name, 'annotated_objects')
- book = Book.annotated_objects.create(title='Hunting')
+ self.assertEqual(Book._meta.base_manager.name, "annotated_objects")
+ book = Book.annotated_objects.create(title="Hunting")
Person.objects.create(
- first_name='Bugs', last_name='Bunny', fun=True,
- favorite_book=book, favorite_thing_id=1,
+ first_name="Bugs",
+ last_name="Bunny",
+ fun=True,
+ favorite_book=book,
+ favorite_thing_id=1,
)
book = Book.annotated_objects.first()
self.assertEqual(book.favorite_avg, 1) # Annotation from the manager.
- book.title = 'New Hunting'
+ book.title = "New Hunting"
# save() fails if annotations that involve related fields aren't
# cleared before the update query.
book.save()
- self.assertEqual(Book.annotated_objects.first().title, 'New Hunting')
+ self.assertEqual(Book.annotated_objects.first().title, "New Hunting")
def test_delete_related_on_filtered_manager(self):
"""Deleting related objects should also not be distracted by a
@@ -707,7 +826,7 @@ class CustomManagersRegressTestCase(TestCase):
test for #2698."""
related = RelatedModel.objects.create(name="xyzzy")
- for name, public in (('one', True), ('two', False), ('three', False)):
+ for name, public in (("one", True), ("two", False), ("three", False)):
RestrictedModel.objects.create(name=name, is_public=public, related=related)
obj = RelatedModel.objects.get(name="xyzzy")