mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-10-31 09:41:08 +00:00
Code Issues 32 Releases Wiki Activity

Refs #30325 -- Added tests for using count()/exists() with custom managers and reverse M2M relations.

Browse Source
This commit is contained in:
Tobias Kunze
2019-04-15 11:03:53 +02:00
committed by Mariusz Felisiak
parent 5f7991c42c
commit 9ac8520fcd
2 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/many_to_many/models.py
View File

@@ -27,6 +27,11 @@ class Tag(models.Model):
return self.name
class NoDeletedArticleManager(models.Manager):
def get_queryset(self):
return super().get_queryset().exclude(headline='deleted')
class Article(models.Model):
headline = models.CharField(max_length=100)
# Assign a string as name to make sure the intermediary model is
@@ -34,6 +39,8 @@ class Article(models.Model):
publications = models.ManyToManyField(Publication, name='publications')
tags = models.ManyToManyField(Tag, related_name='tags')
objects = NoDeletedArticleManager()
class Meta:
ordering = ('headline',)

6
tests/many_to_many/tests.py
View File

@@ -580,3 +580,9 @@ class ManyToManyTests(TestCase):
]
)
self.assertQuerysetEqual(b.publications.all(), ['<Publication: Science Weekly>'])
def test_custom_default_manager_exists_count(self):
a5 = Article.objects.create(headline='deleted')
a5.publications.add(self.p2)
self.assertEqual(self.p2.article_set.count(), self.p2.article_set.all().count())
self.assertEqual(self.p3.article_set.exists(), self.p3.article_set.all().exists())