diff --git a/eav/tests/models.py b/eav/tests/models.py index 7a0f270..ae85394 100644 --- a/eav/tests/models.py +++ b/eav/tests/models.py @@ -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' diff --git a/eav/tests/registry.py b/eav/tests/registry.py index 0239f24..de3eb07 100644 --- a/eav/tests/registry.py +++ b/eav/tests/registry.py @@ -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)