mirror of
https://github.com/jazzband/django-eav2.git
synced 2026-03-16 22:40:26 +00:00
146 lines
4.8 KiB
Python
146 lines
4.8 KiB
Python
from datetime import datetime
|
|
|
|
from django.test import TestCase
|
|
|
|
from ..models import *
|
|
from ..utils import EavRegistry, EavConfig
|
|
from .models import Patient
|
|
|
|
|
|
class EavBasicTests(TestCase):
|
|
|
|
|
|
def setUp(self):
|
|
|
|
EavRegistry.unregister(Patient)
|
|
EavRegistry.register(Patient)
|
|
|
|
self.attribute = EavAttribute.objects.create(datatype=EavAttribute.TYPE_TEXT,
|
|
name='City',
|
|
help_text='The City')
|
|
self.entity = Patient.objects.create(name="Doe")
|
|
|
|
self.value = EavValue.objects.create(object=self.entity,
|
|
attribute=self.attribute,
|
|
value_text='Denver')
|
|
|
|
|
|
def tearDown(self):
|
|
EavRegistry.unregister(Patient)
|
|
|
|
|
|
def test_can_create_attribute(self):
|
|
EavAttribute.objects.create(datatype=EavAttribute.TYPE_TEXT,
|
|
name='My text test',
|
|
help_text='My help text')
|
|
|
|
def test_attribute_unicode(self):
|
|
self.assertEqual(unicode(self.attribute), "City (Text)")
|
|
|
|
|
|
def test_can_eaventity_children_give_you_all_attributes_by_default(self):
|
|
qs = Patient.eav.get_eav_attributes()
|
|
self.assertEqual(list(qs), list(EavAttribute.objects.all()))
|
|
|
|
|
|
def test_value_creation(self):
|
|
EavValue.objects.create(object=self.entity,
|
|
attribute=self.attribute,
|
|
value_float=1.2)
|
|
|
|
def test_value_unicode(self):
|
|
self.assertEqual(unicode(self.value), "Doe - City: \"Denver\"")
|
|
|
|
|
|
def test_value_unicode(self):
|
|
self.assertEqual(unicode(self.value), "Doe - City: \"Denver\"")
|
|
|
|
|
|
def test_value_types(self):
|
|
_text = EavAttribute.objects.create(datatype=EavAttribute.TYPE_TEXT,
|
|
name='Text',
|
|
help_text='The text')
|
|
val = EavValue.objects.create(object=self.entity,
|
|
attribute = _text)
|
|
value = "Test text"
|
|
val.value = value
|
|
val.save()
|
|
self.assertEqual(val.value, value)
|
|
|
|
_float = EavAttribute.objects.create(datatype=EavAttribute.TYPE_FLOAT,
|
|
name='Float',
|
|
help_text='The float')
|
|
val = EavValue.objects.create(object=self.entity,
|
|
attribute = _float)
|
|
value = 1.22
|
|
val.value = value
|
|
val.save()
|
|
self.assertEqual(val.value, value)
|
|
|
|
|
|
_int = EavAttribute.objects.create(datatype=EavAttribute.TYPE_INT,
|
|
name='Int',
|
|
help_text='The int')
|
|
val = EavValue.objects.create(object=self.entity,
|
|
attribute = _int)
|
|
value = 7
|
|
val.value = value
|
|
val.save()
|
|
self.assertEqual(val.value, value)
|
|
|
|
_date = EavAttribute.objects.create(datatype=EavAttribute.TYPE_DATE,
|
|
name='Date',
|
|
help_text='The date')
|
|
val = EavValue.objects.create(object=self.entity,
|
|
attribute = _date)
|
|
value = datetime.now()
|
|
val.value = value
|
|
val.save()
|
|
self.assertEqual(val.value, value)
|
|
|
|
_bool = EavAttribute.objects.create(datatype=EavAttribute.TYPE_BOOLEAN,
|
|
name='Bool',
|
|
help_text='The bool')
|
|
val = EavValue.objects.create(object=self.entity,
|
|
attribute = _bool)
|
|
value = False
|
|
val.value = value
|
|
val.save()
|
|
self.assertEqual(val.value, value)
|
|
|
|
|
|
def test_eavregistry_ataches_and_detaches_eav_attribute(self):
|
|
EavRegistry.unregister(Patient)
|
|
p = Patient()
|
|
self.assertFalse(hasattr(p, 'eav'))
|
|
|
|
EavRegistry.register(Patient)
|
|
p2 = Patient()
|
|
self.assertTrue(p2.eav)
|
|
|
|
|
|
def test_eavregistry_ataches_and_detaches_eav_attribute(self):
|
|
from ..utils import EavRegistry
|
|
EavRegistry.unregister(Patient)
|
|
p = Patient()
|
|
self.assertFalse(hasattr(p, 'eav'))
|
|
|
|
EavRegistry.register(Patient)
|
|
p2 = Patient()
|
|
self.assertTrue(p2.eav)
|
|
|
|
|
|
def test_eavregistry_accept_a_settings_class_with_get_queryset(self):
|
|
EavRegistry.unregister(Patient)
|
|
|
|
class PatientEav(EavConfig):
|
|
|
|
def get_eav_attributes(self):
|
|
return EavAttribute.objects.all()
|
|
|
|
EavRegistry.register(Patient, PatientEav)
|
|
|
|
p = Patient()
|
|
|
|
EavRegistry.unregister(Patient)
|
|
|