summaryrefslogtreecommitdiff
path: root/tests/many_to_many
diff options
context:
space:
mode:
authorTobias Kunze <rixx@cutebit.de>2019-04-15 11:03:53 +0200
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2019-04-15 12:02:31 +0200
commit9ac8520fcde29840a1345be19d80dbda53aa6d03 (patch)
tree7c614ed793e27eedb2b76d3319489efeb447cc76 /tests/many_to_many
parent5f7991c42cff73b6278106d499d719b726f85ead (diff)
downloaddjango-9ac8520fcde29840a1345be19d80dbda53aa6d03.tar.gz
Refs #30325 -- Added tests for using count()/exists() with custom managers and reverse M2M relations.
Diffstat (limited to 'tests/many_to_many')
-rw-r--r--tests/many_to_many/models.py7
-rw-r--r--tests/many_to_many/tests.py6
2 files changed, 13 insertions, 0 deletions
diff --git a/tests/many_to_many/models.py b/tests/many_to_many/models.py
index 28d23f2a82..7b46d2484e 100644
--- a/tests/many_to_many/models.py
+++ b/tests/many_to_many/models.py
@@ -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',)
diff --git a/tests/many_to_many/tests.py b/tests/many_to_many/tests.py
index 9dc53307a1..7434d18e02 100644
--- a/tests/many_to_many/tests.py
+++ b/tests/many_to_many/tests.py
@@ -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())