class BaseManagerTestCase(unittest.TestCase): def setUp(self): from django.db import models self.moderator = GenericModerator(UserProfile) class CustomManager(models.Manager): pass class ModelClass(models.Model): pass self.custom_manager = CustomManager self.model_class = ModelClass def test_get_base_manager(self): self.model_class.add_to_class('objects', self.custom_manager()) base_manager = self.moderator._get_base_manager( self.model_class, 'objects') self.assertEqual(base_manager, self.custom_manager) delattr(self.model_class, 'objects') def test_get_base_manager_default_manager(self): base_manager = self.moderator._get_base_manager( self.model_class, 'objects') self.assertEqual(base_manager, Manager)
class BaseManagerTestCase(unittest.TestCase): def setUp(self): from django.db import models self.moderator = GenericModerator(UserProfile) class CustomManager(models.Manager): pass class ModelClass(models.Model): pass self.custom_manager = CustomManager self.model_class = ModelClass def test_get_base_manager(self): self.model_class.add_to_class('objects', self.custom_manager()) base_manager = self.moderator._get_base_manager(self.model_class, 'objects') self.assertEqual(base_manager, self.custom_manager) delattr(self.model_class, 'objects') def test_get_base_manager_default_manager(self): base_manager = self.moderator._get_base_manager(self.model_class, 'objects') self.assertEqual(base_manager, Manager)