from django.test import TestCase

import eav
from eav.models import Attribute, EnumGroup, EnumValue, Value
from test_project.models import Patient


class ModelTest(TestCase):
    def setUp(self):
        eav.register(Patient)
        Attribute.objects.create(name="age", datatype=Attribute.TYPE_INT)
        Attribute.objects.create(name="height", datatype=Attribute.TYPE_FLOAT)
        Attribute.objects.create(name="weight", datatype=Attribute.TYPE_FLOAT)
        Attribute.objects.create(name="color", datatype=Attribute.TYPE_TEXT)

        EnumGroup.objects.create(name="Yes / No")
        EnumValue.objects.create(value="yes")
        EnumValue.objects.create(value="no")
        EnumValue.objects.create(value="unknown")

    def test_attr_natural_keys(self):
        attr = Attribute.objects.get(name="age")
        attr_natural_key = attr.natural_key()
        attr_retrieved_model = Attribute.objects.get_by_natural_key(*attr_natural_key)
        self.assertEqual(attr_retrieved_model, attr)

    def test_value_natural_keys(self):
        p = Patient.objects.create(name="Jon")
        p.eav.age = 5
        p.save()

        val = p.eav_values.first()

        value_natural_key = val.natural_key()
        value_retrieved_model = Value.objects.get_by_natural_key(*value_natural_key)
        self.assertEqual(value_retrieved_model, val)

    def test_enum_group_natural_keys(self):
        enum_group = EnumGroup.objects.first()
        enum_group_natural_key = enum_group.natural_key()
        enum_group_retrieved_model = EnumGroup.objects.get_by_natural_key(
            *enum_group_natural_key,
        )
        self.assertEqual(enum_group_retrieved_model, enum_group)

    def test_enum_value_natural_keys(self):
        enum_value = EnumValue.objects.first()
        enum_value_natural_key = enum_value.natural_key()
        enum_value_retrieved_model = EnumValue.objects.get_by_natural_key(
            *enum_value_natural_key,
        )
        self.assertEqual(enum_value_retrieved_model, enum_value)