Fixed #22447 -- Make sure custom model bases can be migrated.
Thanks to cdestigter for the report.
This commit is contained in:
parent
a2340ac6d6
commit
390f888745
|
@ -3,9 +3,18 @@ from __future__ import unicode_literals
|
||||||
|
|
||||||
from django.apps.registry import Apps
|
from django.apps.registry import Apps
|
||||||
from django.db import models
|
from django.db import models
|
||||||
|
from django.utils import six
|
||||||
from django.utils.encoding import python_2_unicode_compatible
|
from django.utils.encoding import python_2_unicode_compatible
|
||||||
|
|
||||||
|
|
||||||
|
class CustomModelBase(models.base.ModelBase):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class ModelWithCustomBase(six.with_metaclass(CustomModelBase, models.Model)):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
@python_2_unicode_compatible
|
@python_2_unicode_compatible
|
||||||
class UnicodeModel(models.Model):
|
class UnicodeModel(models.Model):
|
||||||
title = models.CharField('ÚÑÍ¢ÓÐÉ', max_length=20, default='“Ðjáñgó”')
|
title = models.CharField('ÚÑÍ¢ÓÐÉ', max_length=20, default='“Ðjáñgó”')
|
||||||
|
|
|
@ -3,6 +3,8 @@ from django.db import models
|
||||||
from django.db.migrations.state import ProjectState, ModelState, InvalidBasesError
|
from django.db.migrations.state import ProjectState, ModelState, InvalidBasesError
|
||||||
from django.test import TestCase
|
from django.test import TestCase
|
||||||
|
|
||||||
|
from .models import ModelWithCustomBase
|
||||||
|
|
||||||
|
|
||||||
class StateTests(TestCase):
|
class StateTests(TestCase):
|
||||||
"""
|
"""
|
||||||
|
@ -335,3 +337,9 @@ class StateTests(TestCase):
|
||||||
project_state.add_model_state(ModelState.from_model(Magazine))
|
project_state.add_model_state(ModelState.from_model(Magazine))
|
||||||
with self.assertRaises(ValueError):
|
with self.assertRaises(ValueError):
|
||||||
rendered_state = project_state.render()
|
rendered_state = project_state.render()
|
||||||
|
|
||||||
|
|
||||||
|
class ModelStateTests(TestCase):
|
||||||
|
def test_custom_model_base(self):
|
||||||
|
state = ModelState.from_model(ModelWithCustomBase)
|
||||||
|
self.assertEqual(state.bases, (models.Model,))
|
||||||
|
|
Loading…
Reference in New Issue