From 04c848e8c729e190f12cc1bd943de9d28fbecf1d Mon Sep 17 00:00:00 2001
From: Paul McLanahan <paul@mclanahan.net>
Date: Wed, 9 Mar 2011 14:44:10 -0500
Subject: [PATCH] Added tests for PassThroughManager.

---
 model_utils/managers.py     |  2 +-
 model_utils/tests/models.py | 33 ++++++++++++++++++++++++++++-
 model_utils/tests/tests.py  | 42 ++++++++++++++++++++++++++++++++++++-
 3 files changed, 74 insertions(+), 3 deletions(-)

diff --git a/model_utils/managers.py b/model_utils/managers.py
index 47e7fbb..1f51d9d 100644
--- a/model_utils/managers.py
+++ b/model_utils/managers.py
@@ -104,7 +104,7 @@ class PassThroughManager(models.Manager):
     
     def __init__(self, queryset_cls=None):
         self._queryset_cls = queryset_cls
-        super(PassthroughManager, self).__init__()
+        super(PassThroughManager, self).__init__()
     
     def __getattr__(self, name):
         if name in self._deny_methods:
diff --git a/model_utils/tests/models.py b/model_utils/tests/models.py
index 5e63597..a84e1cc 100644
--- a/model_utils/tests/models.py
+++ b/model_utils/tests/models.py
@@ -2,7 +2,7 @@ from django.db import models
 from django.utils.translation import ugettext_lazy as _
 
 from model_utils.models import InheritanceCastModel, TimeStampedModel, StatusModel, TimeFramedModel
-from model_utils.managers import QueryManager, manager_from, InheritanceManager
+from model_utils.managers import QueryManager, manager_from, InheritanceManager, PassThroughManager
 from model_utils.fields import SplitField, MonitorField
 from model_utils import Choices
 
@@ -115,3 +115,34 @@ class Entry(models.Model):
     featured = manager_from(PublishedMixin,
                             manager_cls=FeaturedManager,
                             queryset_cls=ByAuthorQuerySet)
+
+class DudeQuerySet(models.query.QuerySet):
+    def abiding(self):
+        return self.filter(abides=True)
+    
+    def rug_positive(self):
+        return self.filter(has_rug=True)
+    
+    def rug_negative(self):
+        return self.filter(has_rug=False)
+    
+    def by_name(self, name):
+        return self.filter(name__iexact=name)
+
+class AbidingManager(PassThroughManager):
+    def get_query_set(self):
+        return DudeQuerySet(self.model, using=self._db).abiding()
+    
+    def get_stats(self):
+        return {
+            'abiding_count': self.count(),
+            'rug_count': self.rug_positive().count(),
+        }
+
+class Dude(models.Model):
+    abides = models.BooleanField(default=True)
+    name = models.CharField(max_length=20)
+    has_rug = models.BooleanField()
+    
+    objects = PassThroughManager(DudeQuerySet)
+    abiders = AbidingManager()
diff --git a/model_utils/tests/tests.py b/model_utils/tests/tests.py
index d073f64..d2733f3 100644
--- a/model_utils/tests/tests.py
+++ b/model_utils/tests/tests.py
@@ -1,3 +1,8 @@
+try:
+    import cPickle as pickle
+except ImportError:
+    import pickle
+
 from datetime import datetime, timedelta
 
 from django.test import TestCase
@@ -16,7 +21,7 @@ from model_utils.tests.models import (
     InheritParent, InheritChild, InheritChild2, InheritanceManagerTestParent,
     InheritanceManagerTestChild1, InheritanceManagerTestChild2,
     TimeStamp, Post, Article, Status, StatusPlainTuple, TimeFrame, Monitored,
-    StatusManagerAdded, TimeFrameManagerAdded,  Entry)
+    StatusManagerAdded, TimeFrameManagerAdded, Entry, Dude)
 
 
 class GetExcerptTests(TestCase):
@@ -408,6 +413,7 @@ if 'south' in settings.INSTALLED_APPS:
                               NoRendered._meta.get_field,
                               '_body_excerpt')
 
+
 class ManagerFromTests(TestCase):
     def setUp(self):
         Entry.objects.create(author='George', published=True)
@@ -429,3 +435,37 @@ class ManagerFromTests(TestCase):
 
     def test_cant_reconcile_qs_class(self):
         self.assertRaises(TypeError, Entry.broken.all)
+    
+    def test_queryset_pickling_fails(self):
+        qs = Entry.objects.all()
+        def dump_load():
+            pqs = pickle.dumps(qs)
+            upqs = pickle.loads(pqs)
+        self.assertRaises(pickle.PicklingError, dump_load)
+
+
+class PassThroughManagerTests(TestCase):
+    def setUp(self):
+        Dude.objects.create(name='The Dude', abides=True, has_rug=False)
+        Dude.objects.create(name='His Dudeness', abides=False, has_rug=True)
+        Dude.objects.create(name='Duder', abides=False, has_rug=False)
+        Dude.objects.create(name='El Duderino', abides=True, has_rug=True)
+
+    def test_chaining(self):
+        self.assertEqual(Dude.objects.by_name('Duder').count(), 1)
+        self.assertEqual(Dude.objects.all().by_name('Duder').count(), 1)
+        self.assertEqual(Dude.abiders.rug_positive().count(), 1)
+        self.assertEqual(Dude.abiders.all().rug_positive().count(), 1)
+    
+    def test_manager_only_methods(self):
+        stats = Dude.abiders.get_stats()
+        self.assertEqual(stats['rug_count'], 1)
+        def notonqs():
+            Dude.abiders.all().get_stats()
+        self.assertRaises(AttributeError, notonqs)
+    
+    def test_queryset_pickling(self):
+        qs = Dude.objects.all()
+        saltyqs = pickle.dumps(qs)
+        unqs = pickle.loads(saltyqs)
+        self.assertEqual(unqs.by_name('The Dude').count(), 1)