Adding decorator tests

Author: jpwhite3

eav/tests/models.py

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

eav/tests/registry.py

@@ -70,6 +70,12 @@ def test_unregistering(self):
         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)