From 1d4613e807c81f51720152714ccd3089f5a70f53 Mon Sep 17 00:00:00 2001 From: Carl Meyer Date: Fri, 24 Sep 2010 10:30:58 -0400 Subject: [PATCH] make tests compatible with pre-1.2 --- model_utils/tests/models.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/model_utils/tests/models.py b/model_utils/tests/models.py index cbe0048..fa7ff9d 100644 --- a/model_utils/tests/models.py +++ b/model_utils/tests/models.py @@ -88,7 +88,10 @@ class ByAuthorQuerySet(models.query.QuerySet, AuthorMixin): class FeaturedManager(models.Manager): def get_query_set(self): - return ByAuthorQuerySet(self.model, using=self._db).filter(feature=True) + kwargs = {} + if hasattr(self, '_db'): + kwargs['using'] = self._db + return ByAuthorQuerySet(self.model, **kwargs).filter(feature=True) class Entry(models.Model): author = models.CharField(max_length=20)