Ticket #36892: django-lazy-migrations-bug-unit-test.diff

tests/migrations/test_writer.py

@@ -3 +3 @@
 import enum
 import functools
 import math
+import operator
 import os
 import pathlib
 import re
@@ -25 +26 @@
 from django.test import SimpleTestCase, override_settings
 from django.test.utils import extend_sys_path
 from django.utils.deconstruct import deconstructible
-from django.utils.functional import SimpleLazyObject
+from django.utils.functional import lazy, SimpleLazyObject
 from django.utils.timezone import get_default_timezone, get_fixed_timezone
 from django.utils.translation import gettext_lazy as _
 
@@ -496 +497 @@
             DATE_1 = 1969, 7, 20, "First date"
             DATE_2 = 1969, 11, 19, "Second date"
 
+        TUPLE_1 = (("A", "A value"),)
+        TUPLE_2 = (("B", "B value"),)
+        LAZY_TUPLE = lazy(
+            lambda: tuple(sorted(TUPLE_1 + TUPLE_2, key=operator.itemgetter(1))), tuple
+        )()
+
         self.assertSerializedResultEqual(TextChoices.A, ("'A'", set()))
         self.assertSerializedResultEqual(IntegerChoices.A, ("1", set()))
         self.assertSerializedResultEqual(
@@ -509 +516 @@
             "models.CharField(choices=[('A', 'A value'), ('B', 'B value')], "
             "default='B')",
         )
+        field = models.CharField(default=TUPLE_2[0][0], choices=LAZY_TUPLE)
+        string = MigrationWriter.serialize(field)[0]
+        self.assertEqual(
+            string,
+            "models.CharField(choices=[('A', 'A value'), ('B', 'B value')], "
+            "default='B')",
+        )
         field = models.IntegerField(default=IntegerChoices.B, choices=IntegerChoices)
         string = MigrationWriter.serialize(field)[0]
         self.assertEqual(