File: test_model_many_to_many.py

package info (click to toggle)
python-django-parler 2.3-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 1,032 kB
  • sloc: python: 4,293; makefile: 164; sh: 6
file content (88 lines) | stat: -rw-r--r-- 2,965 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
from .testapp.models import (
    ManyToManyAndOtherFieldsTranslationModel,
    ManyToManyOnlyFieldsTranslationModel,
    RegularModel,
)
from .utils import AppTestCase


class ModelManyToManyTestCase(AppTestCase):
    @classmethod
    def setUpTestData(cls):
        super().setUpTestData()
        cls.regular_one = RegularModel.objects.create(original_field="One")
        cls.regular_two = RegularModel.objects.create(original_field="Two")

    def test_save_many_to_many_only(self):
        """Test a model that has *only* translated many to many fields."""
        obj = ManyToManyOnlyFieldsTranslationModel.objects.create(shared="One")

        # Set many to many for English
        obj.set_current_language("en")
        obj.create_translation("en")
        obj.translated_many_to_many.set([self.regular_one])
        obj.save()
        self.assertEqual(
            ManyToManyOnlyFieldsTranslationModel.objects.language("en")
            .first()
            .translated_many_to_many.all()[0],
            self.regular_one,
        )

        # Set many to many for French
        obj.set_current_language("fr")
        obj.create_translation("fr")
        obj.translated_many_to_many.set([self.regular_two])
        obj.save()
        self.assertEqual(
            ManyToManyOnlyFieldsTranslationModel.objects.language("fr")
            .first()
            .translated_many_to_many.all()[0],
            self.regular_two,
        )

        # Check fallback
        self.assertEqual(
            ManyToManyOnlyFieldsTranslationModel.objects.language("nl")
            .first()
            .translated_many_to_many.all()[0],
            self.regular_one,
        )

    def test_save_many_to_many_and_other_fields(self):
        """Test a model that has *only* translated many to many fields."""
        obj = ManyToManyAndOtherFieldsTranslationModel.objects.create(shared="One")

        # Set many to many for English
        obj.set_current_language("en")
        obj.tr_title = "English"
        obj.save()
        obj.translated_many_to_many.set([self.regular_one])
        obj.save()
        self.assertEqual(
            ManyToManyAndOtherFieldsTranslationModel.objects.language("en")
            .first()
            .translated_many_to_many.all()[0],
            self.regular_one,
        )

        # Set many to many for French
        obj.set_current_language("fr")
        obj.tr_title = "French"
        obj.save()
        obj.translated_many_to_many.set([self.regular_two])
        obj.save()
        self.assertEqual(
            ManyToManyAndOtherFieldsTranslationModel.objects.language("fr")
            .first()
            .translated_many_to_many.all()[0],
            self.regular_two,
        )

        # Check fallback
        self.assertEqual(
            ManyToManyAndOtherFieldsTranslationModel.objects.language("nl")
            .first()
            .translated_many_to_many.all()[0],
            self.regular_one,
        )