mirror of
https://github.com/jazzband/django-eav2.git
synced 2026-03-16 22:40:26 +00:00
Adding decorator tests
This commit is contained in:
parent
3d94eb3ec2
commit
36bfc2e0e8
2 changed files with 7 additions and 1 deletions
|
|
@ -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'
|
||||
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue