mirror of
https://github.com/jazzband/django-eav2.git
synced 2026-03-16 22:40:26 +00:00
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
from django.test import TestCase
|
|
|
|
import eav
|
|
from eav.models import Attribute, EnumGroup, EnumValue, Value
|
|
from main_app.models import Patient
|
|
|
|
|
|
class MiscModels(TestCase):
|
|
def test_enumgroup_str(self):
|
|
name = 'Yes / No'
|
|
e = EnumGroup.objects.create(name=name)
|
|
self.assertEqual('<EnumGroup Yes / No>', str(e))
|
|
|
|
def test_attribute_help_text(self):
|
|
desc = 'Patient Age'
|
|
a = Attribute.objects.create(
|
|
name='age', description=desc, datatype=Attribute.TYPE_INT
|
|
)
|
|
self.assertEqual(a.help_text, desc)
|
|
|
|
def test_setting_to_none_deletes_value(self):
|
|
eav.register(Patient)
|
|
Attribute.objects.create(name='age', datatype=Attribute.TYPE_INT)
|
|
p = Patient.objects.create(name='Bob', eav__age=5)
|
|
self.assertEqual(Value.objects.count(), 1)
|
|
p.eav.age = None
|
|
p.save()
|
|
self.assertEqual(Value.objects.count(), 0)
|
|
|
|
def test_string_enum_value_assignment(self):
|
|
yes = EnumValue.objects.create(value='yes')
|
|
no = EnumValue.objects.create(value='no')
|
|
ynu = EnumGroup.objects.create(name='Yes / No / Unknown')
|
|
ynu.values.add(yes)
|
|
ynu.values.add(no)
|
|
Attribute.objects.create(
|
|
name='is_patient', datatype=Attribute.TYPE_ENUM, enum_group=ynu
|
|
)
|
|
eav.register(Patient)
|
|
p = Patient.objects.create(name='Joe')
|
|
p.eav.is_patient = 'yes'
|
|
p.save()
|
|
p = Patient.objects.get(name='Joe') # get from DB again
|
|
self.assertEqual(p.eav.is_patient, yes)
|