Ensure we call get_query_set on django 1.5-

Supersedes https://github.com/carljm/django-model-utils/pull/106

I can't actually see why this works, but it does: tests pass
after this that failed before.
This commit is contained in:
Matthew Schinckel 2014-03-20 13:11:00 +10:30
parent a49936c267
commit 93500bb381
2 changed files with 4 additions and 1 deletions

View file

@ -240,6 +240,8 @@ class PassThroughManagerMixin(object):
def __getattr__(self, name):
if name in self._deny_methods:
raise AttributeError(name)
if django.VERSION < (1, 6, 0):
return getattr(self.get_query_set(), name)
return getattr(self.get_queryset(), name)
def get_queryset(self):

View file

@ -1238,7 +1238,7 @@ class PassThroughManagerTests(TestCase):
class CreatePassThroughManagerTests(TestCase):
def setUp(self):
self.dude = Dude.objects.create(name='El Duderino')
self.other_dude = Dude.objects.create(name='Das D\xc3de')
self.other_dude = Dude.objects.create(name='Das Dude')
def test_reverse_manager(self):
Spot.objects.create(
@ -1249,6 +1249,7 @@ class CreatePassThroughManagerTests(TestCase):
name='The Crux', owner=self.other_dude, closed=True, secure=True,
secret=False
)
self.assertEqual(self.dude.spots_owned.closed().all().count(), 1)
self.assertEqual(self.dude.spots_owned.closed().count(), 1)
def test_related_queryset_pickling(self):