Clean tests a bit (#3)

This commit is contained in:
Siegmeyer of Catarina 2018-06-07 00:28:11 +02:00 committed by Iwo Herka
parent 591e4e11b1
commit f260060b60
2 changed files with 5 additions and 9 deletions

View file

@ -35,7 +35,7 @@ class Queries(TestCase):
eav.unregister(Patient)
def test_get_or_create_with_eav(self):
p = Patient.objects.get_or_create(name='Bob', eav__age=5)
Patient.objects.get_or_create(name='Bob', eav__age=5)
self.assertEqual(Patient.objects.count(), 1)
self.assertEqual(Value.objects.count(), 1)
p = Patient.objects.get_or_create(name='Bob', eav__age=5)
@ -49,7 +49,7 @@ class Queries(TestCase):
p1, _ = Patient.objects.get_or_create(name='Bob', eav__age=6)
self.assertEqual(Patient.objects.get(eav__age=6), p1)
p2, _ = Patient.objects.get_or_create(name='Fred', eav__age=6)
Patient.objects.create(name='Fred', eav__age=6)
self.assertRaises(MultipleObjectsReturned, lambda: Patient.objects.get(eav__age=6))
def test_filtering_on_normal_and_eav_fields(self):

View file

@ -3,11 +3,10 @@ from django.test import TestCase
import eav
from eav.registry import EavConfig
from .models import Patient, Encounter, ExampleModel
from .models import Encounter, ExampleModel, Patient
class RegistryTests(TestCase):
def setUp(self):
pass
@ -33,10 +32,8 @@ class RegistryTests(TestCase):
self.assertEqual(Patient._eav_config_cls.manager_attr, 'objects')
self.assertFalse(Patient._eav_config_cls.manager_only)
self.assertEqual(Patient._eav_config_cls.eav_attr, 'eav')
self.assertEqual(Patient._eav_config_cls.generic_relation_attr,
'eav_values')
self.assertEqual(Patient._eav_config_cls.generic_relation_related_name,
None)
self.assertEqual(Patient._eav_config_cls.generic_relation_attr, 'eav_values')
self.assertEqual(Patient._eav_config_cls.generic_relation_related_name, None)
eav.unregister(Patient)
def test_registering_overriding_defaults(self):
@ -70,7 +67,6 @@ class RegistryTests(TestCase):
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):