django-eav2/tests/basics.py

from datetime import datetime

from django.test import TestCase

from ..models import *
from ..utils import EavRegistry, EavConfig
from .models import Patient


class EavBasicTests(TestCase):

    """
        Testing basics such as registration, printing and object creation
    """


    def setUp(self):
    
        EavRegistry.unregister(Patient)
        EavRegistry.register(Patient)

        self.attribute = EavAttribute.objects.create(datatype=EavAttribute.TYPE_TEXT,
                                                name='City', help_text='The City', slug='city')
                                                
        self.entity = Patient.objects.create(name="Doe")

        self.value = EavValue.objects.create(entity=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', slug='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(entity=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_types(self):
        _text = EavAttribute.objects.create(datatype=EavAttribute.TYPE_TEXT,
                                            name='Text', slug='text',
                                            help_text='The text')
        val = EavValue.objects.create(entity=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', slug='float',
                                             help_text='The float')
        val = EavValue.objects.create(entity=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', slug='int',
                                           help_text='The int')
        val = EavValue.objects.create(entity=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', slug='date',
                                            help_text='The date')
        val = EavValue.objects.create(entity=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', slug='bool',
                                            help_text='The bool')
        val = EavValue.objects.create(entity=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):
        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):

            @classmethod
            def get_eav_attributes(self):
                return EavAttribute.objects.all()

        EavRegistry.register(Patient, PatientEav)
        
        p = Patient()

        EavRegistry.unregister(Patient)
        
    
    def test_eavregistry_accept_a_settings_class_with_field_names(self):
        
        p = Patient()
        registered_manager = Patient.objects        
        EavRegistry.unregister(Patient)

        class PatientEav(EavConfig):

            proxy_field_name = 'my_eav'
            manager_field_name ='my_objects'

        EavRegistry.register(Patient, PatientEav)
        
        p2 = Patient()
        self.assertEqual(type(p.eav), type(p2.my_eav))
        self.assertEqual(type(registered_manager), type(Patient.my_objects))
                         
        bak_registered_manager = Patient.objects 

        EavRegistry.unregister(Patient)
        
        self.assertEqual(type(Patient.objects), type(bak_registered_manager))