diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/migrations/test_writer.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/tests/migrations/test_writer.py b/tests/migrations/test_writer.py index addf10b0f0..6a65e87d5a 100644 --- a/tests/migrations/test_writer.py +++ b/tests/migrations/test_writer.py @@ -193,6 +193,10 @@ class WriterTests(SimpleTestCase): A = 1 B = 2 + class NestedChoices(models.TextChoices): + X = 'X', 'X value' + Y = 'Y', 'Y value' + def safe_exec(self, string, value=None): d = {} try: @@ -388,6 +392,18 @@ class WriterTests(SimpleTestCase): "default=datetime.date(1969, 11, 19))" ) + def test_serialize_nested_class(self): + for nested_cls in [self.NestedEnum, self.NestedChoices]: + cls_name = nested_cls.__name__ + with self.subTest(cls_name): + self.assertSerializedResultEqual( + nested_cls, + ( + "migrations.test_writer.WriterTests.%s" % cls_name, + {'import migrations.test_writer'}, + ), + ) + def test_serialize_uuid(self): self.assertSerializedEqual(uuid.uuid1()) self.assertSerializedEqual(uuid.uuid4()) |