django-localized-fields/tests/test_model.py

from django.test import TestCase

from localized_fields.fields import LocalizedField
from localized_fields.value import LocalizedValue

from .fake_model import get_fake_model


class LocalizedModelTestCase(TestCase):
    """Tests whether the :see:LocalizedModel class."""

    TestModel = None

    @classmethod
    def setUpClass(cls):
        """Creates the test model in the database."""

        super(LocalizedModelTestCase, cls).setUpClass()

        cls.TestModel = get_fake_model(
            {
                'title': LocalizedField()
            }
        )

    @classmethod
    def test_defaults(cls):
        """Tests whether all :see:LocalizedField
        fields are assigned an empty :see:LocalizedValue
        instance when the model is instanitiated."""

        obj = cls.TestModel()

        assert isinstance(obj.title, LocalizedValue)

    @classmethod
    def test_model_init_kwargs(cls):
        """Tests whether all :see:LocalizedField
        fields are assigned an empty :see:LocalizedValue
        instance when the model is instanitiated."""
        data = {
            'title': {
                'en': 'english_title',
                'ro': 'romanian_title',
                'nl': 'dutch_title'
            }
        }
        obj = cls.TestModel(**data)

        assert isinstance(obj.title, LocalizedValue)
        assert obj.title.en == 'english_title'
        assert obj.title.ro == 'romanian_title'
        assert obj.title.nl == 'dutch_title'