diff -r 5ff1fc726848 -r c6bca38c1cbf parts/django/tests/regressiontests/managers_regress/models.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/parts/django/tests/regressiontests/managers_regress/models.py Sat Jan 08 11:20:57 2011 +0530 @@ -0,0 +1,100 @@ +""" +Various edge-cases for model managers. +""" + +from django.db import models + +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 + +class Parent(models.Model): + name = models.CharField(max_length=50) + + manager = OnlyFred() + + def __unicode__(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. +class Child1(AbstractBase1): + data = models.CharField(max_length=25) + + def __unicode__(self): + return self.data + +class Child2(AbstractBase1, AbstractBase2): + data = models.CharField(max_length=25) + + def __unicode__(self): + return self.data + +class Child3(AbstractBase1, AbstractBase3): + data = models.CharField(max_length=25) + + def __unicode__(self): + return self.data + +class Child4(AbstractBase1): + data = models.CharField(max_length=25) + + # Should be the default manager, although the parent managers are + # inherited. + default = models.Manager() + + def __unicode__(self): + return self.data + +class Child5(AbstractBase3): + name = models.CharField(max_length=25) + + default = OnlyFred() + objects = models.Manager() + + def __unicode__(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