mirror of
https://github.com/jazzband/django-eav2.git
synced 2026-03-17 06:50:24 +00:00
c4d7cedeb8
Replace flake8 with ruff and apply consistent linting rules across the entire codebase. Update type annotations, quotation marks, and other style-related changes to comply with the new standards.
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
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)
|