summaryrefslogtreecommitdiff
path: root/tests/custom_managers_regress/models.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/custom_managers_regress/models.py')
-rw-r--r--tests/custom_managers_regress/models.py45
1 files changed, 45 insertions, 0 deletions
diff --git a/tests/custom_managers_regress/models.py b/tests/custom_managers_regress/models.py
new file mode 100644
index 0000000000..71073f0fe7
--- /dev/null
+++ b/tests/custom_managers_regress/models.py
@@ -0,0 +1,45 @@
+"""
+Regression tests for custom manager classes.
+"""
+
+from django.db import models
+from django.utils.encoding import python_2_unicode_compatible
+
+
+class RestrictedManager(models.Manager):
+ """
+ A manager that filters out non-public instances.
+ """
+ def get_query_set(self):
+ return super(RestrictedManager, self).get_query_set().filter(is_public=True)
+
+@python_2_unicode_compatible
+class RelatedModel(models.Model):
+ name = models.CharField(max_length=50)
+
+ def __str__(self):
+ return self.name
+
+@python_2_unicode_compatible
+class RestrictedModel(models.Model):
+ name = models.CharField(max_length=50)
+ is_public = models.BooleanField(default=False)
+ related = models.ForeignKey(RelatedModel)
+
+ objects = RestrictedManager()
+ plain_manager = models.Manager()
+
+ def __str__(self):
+ return self.name
+
+@python_2_unicode_compatible
+class OneToOneRestrictedModel(models.Model):
+ name = models.CharField(max_length=50)
+ is_public = models.BooleanField(default=False)
+ related = models.OneToOneField(RelatedModel)
+
+ objects = RestrictedManager()
+ plain_manager = models.Manager()
+
+ def __str__(self):
+ return self.name