django-localized-fields/tests/test_bulk.py

from django.db import models
from django.test import TestCase

from localized_fields.fields import LocalizedField, LocalizedUniqueSlugField

from .fake_model import get_fake_model


class LocalizedBulkTestCase(TestCase):
    """Tests bulk operations with data structures provided
    by the django-localized-fields library."""

    @staticmethod
    def test_localized_bulk_insert():
        """Tests whether bulk inserts work properly when using
        a :see:LocalizedUniqueSlugField in the model."""

        model = get_fake_model(
            'BulkSlugInsertModel',
            {
                'name': LocalizedField(),
                'slug': LocalizedUniqueSlugField(populate_from='name', include_time=True),
                'score': models.IntegerField()
            }
        )

        to_create = [
            model(name={'en': 'english name 1', 'ro': 'romanian name 1'}, score=1),
            model(name={'en': 'english name 2', 'ro': 'romanian name 2'}, score=2),
            model(name={'en': 'english name 3', 'ro': 'romanian name 3'}, score=3)
        ]

        model.objects.bulk_create(to_create)
        assert model.objects.all().count() == 3

        for obj in to_create:
            obj_db = model.objects.filter(
                name__en=obj.name.en,
                name__ro=obj.name.ro,
                score=obj.score
            ).first()

            assert obj_db
            assert len(obj_db.slug.en) >= len(obj_db.name.en)
            assert len(obj_db.slug.ro) >= len(obj_db.name.ro)