From 36bfc2e0e87a4d6de3f4cf4739373e510ae8a0f0 Mon Sep 17 00:00:00 2001 From: JP White Date: Sun, 16 Aug 2015 00:30:42 -0400 Subject: [PATCH] Adding decorator tests --- eav/tests/models.py | 2 +- eav/tests/registry.py | 6 ++++++ 2 files changed, 7 insertions(+), 1 deletion(-) 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)