from django.test import TestCase

from localized_fields.fields import LocalizedField

from .fake_model import get_fake_model


class LocalizedQuerySetTestCase(TestCase):
    """Tests query sets with models containing :see:LocalizedField."""

    Model = None

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

        super(LocalizedQuerySetTestCase, cls).setUpClass()

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

    @classmethod
    def test_assign_raw_dict(cls):
        inst = cls.Model()
        inst.title = dict(en='Bread', ro='Paine')
        inst.save()

        inst = cls.Model.objects.get(pk=inst.pk)
        assert inst.title.en == 'Bread'
        assert inst.title.ro == 'Paine'

    @classmethod
    def test_assign_raw_dict_update(cls):
        inst = cls.Model.objects.create(
            title=dict(en='Bread', ro='Paine'))

        cls.Model.objects.update(
            title=dict(en='Beer', ro='Bere'))

        inst = cls.Model.objects.get(pk=inst.pk)
        assert inst.title.en == 'Beer'
        assert inst.title.ro == 'Bere'