From 08688bd7dd8dfa218ecff03d6c94a753b1be8e59 Mon Sep 17 00:00:00 2001
From: Willem Van Onsem <vanonsem.willem@gmail.com>
Date: Tue, 16 Aug 2022 07:09:11 +0200
Subject: [PATCH] Refs #33916 -- Added tests for serialization of enum.Flag in
migrations.
---
tests/migrations/test_writer.py | 32 ++++++++++++++++++++++++++++++++
1 file changed, 32 insertions(+)
diff --git a/tests/migrations/test_writer.py b/tests/migrations/test_writer.py
index 6ec6d938b9..bad8775d4e 100644
--- a/tests/migrations/test_writer.py
+++ b/tests/migrations/test_writer.py
@@ -80,6 +80,11 @@ class IntEnum(enum.IntEnum):
B = 2
+class IntFlagEnum(enum.IntFlag):
+ A = 1
+ B = 2
+
+
class OperationWriterTests(SimpleTestCase):
def test_empty_signature(self):
operation = custom_migration_operations.operations.TestOperation()
@@ -382,6 +387,33 @@ class WriterTests(SimpleTestCase):
"default=migrations.test_writer.IntEnum['A'])",
)
+ def test_serialize_enum_flags(self):
+ self.assertSerializedResultEqual(
+ IntFlagEnum.A,
+ (
+ "migrations.test_writer.IntFlagEnum['A']",
+ {"import migrations.test_writer"},
+ ),
+ )
+ self.assertSerializedResultEqual(
+ IntFlagEnum.B,
+ (
+ "migrations.test_writer.IntFlagEnum['B']",
+ {"import migrations.test_writer"},
+ ),
+ )
+ field = models.IntegerField(
+ default=IntFlagEnum.A, choices=[(m.value, m) for m in IntFlagEnum]
+ )
+ string = MigrationWriter.serialize(field)[0]
+ self.assertEqual(
+ string,
+ "models.IntegerField(choices=["
+ "(1, migrations.test_writer.IntFlagEnum['A']), "
+ "(2, migrations.test_writer.IntFlagEnum['B'])], "
+ "default=migrations.test_writer.IntFlagEnum['A'])",
+ )
+
def test_serialize_choices(self):
class TextChoices(models.TextChoices):
A = "A", "A value"