diff options
| author | Florian Apolloner <florian@apolloner.eu> | 2013-02-26 09:53:47 +0100 |
|---|---|---|
| committer | Florian Apolloner <florian@apolloner.eu> | 2013-02-26 14:36:57 +0100 |
| commit | 89f40e36246100df6a11316c31a76712ebc6c501 (patch) | |
| tree | 6e65639683ddaf2027908d1ecb1739e0e2ff853b /tests/managers_regress/models.py | |
| parent | b3d2ccb5bfbaf6e7fe1f98843baaa48c35a70950 (diff) | |
Merged regressiontests and modeltests into the test root.
Diffstat (limited to 'tests/managers_regress/models.py')
| -rw-r--r-- | tests/managers_regress/models.py | 121 |
1 files changed, 121 insertions, 0 deletions
diff --git a/tests/managers_regress/models.py b/tests/managers_regress/models.py new file mode 100644 index 0000000000..d72970d86e --- /dev/null +++ b/tests/managers_regress/models.py @@ -0,0 +1,121 @@ +""" +Various edge-cases for model managers. +""" + +from django.db import models +from django.utils.encoding import python_2_unicode_compatible + + +class OnlyFred(models.Manager): + def get_query_set(self): + return super(OnlyFred, self).get_query_set().filter(name='fred') + + +class OnlyBarney(models.Manager): + def get_query_set(self): + return super(OnlyBarney, self).get_query_set().filter(name='barney') + + +class Value42(models.Manager): + def get_query_set(self): + return super(Value42, self).get_query_set().filter(value=42) + + +class AbstractBase1(models.Model): + name = models.CharField(max_length=50) + + class Meta: + abstract = True + + # Custom managers + manager1 = OnlyFred() + manager2 = OnlyBarney() + objects = models.Manager() + + +class AbstractBase2(models.Model): + value = models.IntegerField() + + class Meta: + abstract = True + + # Custom manager + restricted = Value42() + + +# No custom manager on this class to make sure the default case doesn't break. +class AbstractBase3(models.Model): + comment = models.CharField(max_length=50) + + class Meta: + abstract = True + + +@python_2_unicode_compatible +class Parent(models.Model): + name = models.CharField(max_length=50) + + manager = OnlyFred() + + def __str__(self): + return self.name + + +# Managers from base classes are inherited and, if no manager is specified +# *and* the parent has a manager specified, the first one (in the MRO) will +# become the default. +@python_2_unicode_compatible +class Child1(AbstractBase1): + data = models.CharField(max_length=25) + + def __str__(self): + return self.data + + +@python_2_unicode_compatible +class Child2(AbstractBase1, AbstractBase2): + data = models.CharField(max_length=25) + + def __str__(self): + return self.data + + +@python_2_unicode_compatible +class Child3(AbstractBase1, AbstractBase3): + data = models.CharField(max_length=25) + + def __str__(self): + return self.data + + +@python_2_unicode_compatible +class Child4(AbstractBase1): + data = models.CharField(max_length=25) + + # Should be the default manager, although the parent managers are + # inherited. + default = models.Manager() + + def __str__(self): + return self.data + + +@python_2_unicode_compatible +class Child5(AbstractBase3): + name = models.CharField(max_length=25) + + default = OnlyFred() + objects = models.Manager() + + def __str__(self): + return self.name + + +# Will inherit managers from AbstractBase1, but not Child4. +class Child6(Child4): + value = models.IntegerField() + + +# Will not inherit default manager from parent. +class Child7(Parent): + pass |
