django-eav2/tests/test_misc_models.py

import pytest
from django.test import TestCase

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


@pytest.fixture
def enumgroup(db):
    """Sample `EnumGroup` object for testing."""
    test_group = EnumGroup.objects.create(name='Yes / No')
    value_yes = EnumValue.objects.create(value='Yes')
    value_no = EnumValue.objects.create(value='No')
    test_group.values.add(value_yes)
    test_group.values.add(value_no)
    return test_group


def test_enumgroup_display(enumgroup):
    """Test repr() and str() of EnumGroup."""
    assert f'<EnumGroup {enumgroup.name}>' == repr(enumgroup)
    assert str(enumgroup) == str(enumgroup.name)


def test_enumvalue_display(enumgroup):
    """Test repr() and str() of EnumValue."""
    test_value = enumgroup.values.first()
    assert f'<EnumValue {test_value.value}>' == repr(test_value)
    assert str(test_value) == test_value.value


class MiscModels(TestCase):
    """Miscellaneous tests on models."""

    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)