diff options
| -rw-r--r-- | django/db/migrations/serializer.py | 2 | ||||
| -rw-r--r-- | tests/migrations/test_writer.py | 7 |
2 files changed, 8 insertions, 1 deletions
diff --git a/django/db/migrations/serializer.py b/django/db/migrations/serializer.py index e19c881cda..6e78462e95 100644 --- a/django/db/migrations/serializer.py +++ b/django/db/migrations/serializer.py @@ -273,7 +273,7 @@ class TupleSerializer(BaseSequenceSerializer): class TypeSerializer(BaseSerializer): def serialize(self): special_cases = [ - (models.Model, "models.Model", []), + (models.Model, "models.Model", ['from django.db import models']), (type(None), 'type(None)', []), ] for case, string, imports in special_cases: diff --git a/tests/migrations/test_writer.py b/tests/migrations/test_writer.py index a590ff4398..471194ffc2 100644 --- a/tests/migrations/test_writer.py +++ b/tests/migrations/test_writer.py @@ -658,6 +658,13 @@ class WriterTests(SimpleTestCase): def test_serialize_type_none(self): self.assertSerializedEqual(type(None)) + def test_serialize_type_model(self): + self.assertSerializedEqual(models.Model) + self.assertSerializedResultEqual( + MigrationWriter.serialize(models.Model), + ("('models.Model', {'from django.db import models'})", set()), + ) + def test_simple_migration(self): """ Tests serializing a simple migration. |