Adding decorator tests

This commit is contained in:
JP White 2015-08-16 00:30:42 -04:00
parent 3d94eb3ec2
commit 36bfc2e0e8
2 changed files with 7 additions and 1 deletions

View file

@ -21,7 +21,7 @@ class Encounter(models.Model):
return '%s: encounter num %d' % (self.patient, self.num)
@register_eav()
class ExampleModel(models.Model)
class ExampleModel(models.Model):
class Meta:
app_label = 'eav'

View file

@ -70,6 +70,12 @@ class RegistryTests(TestCase):
self.assertEqual(Patient.objects, old_mgr)
self.assertFalse(hasattr(Patient, '_eav_config_cls'))
def test_unregistering_via_decorator(self):
self.assertTrue(ExampleModel.objects.__class__.__name__ == 'EntityManager')
eav.unregister(ExampleModel)
e = ExampleModel()
self.assertFalse(ExampleModel.objects.__class__.__name__ == 'EntityManager')
def test_unregistering_unregistered_model_proceeds_silently(self):
eav.unregister(Patient)