mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-10-24 22:26:08 +00:00
Code Issues 32 Releases Wiki Activity

Refs #23822 -- Made MigrationOptimizer aware of model managers

Browse Source
This commit is contained in:
Markus Holtermann
2015-01-07 21:14:25 +01:00
committed by Tim Graham
parent 127f9e073d
commit 1f03d2d924
3 changed files with 98 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
django/db/migrations/optimizer.py
View File

@@ -211,6 +211,7 @@ class MigrationOptimizer(object):
fields=operation.fields,
options=operation.options,
bases=operation.bases,
managers=operation.managers,
)
]
@@ -247,6 +248,7 @@ class MigrationOptimizer(object):
fields=operation.fields + [(other.name, other.field)],
options=operation.options,
bases=operation.bases,
managers=operation.managers,
)
]
@@ -261,6 +263,7 @@ class MigrationOptimizer(object):
],
options=operation.options,
bases=operation.bases,
managers=operation.managers,
)
]
@@ -275,6 +278,7 @@ class MigrationOptimizer(object):
],
options=operation.options,
bases=operation.bases,
managers=operation.managers,
)
]
@@ -290,6 +294,7 @@ class MigrationOptimizer(object):
],
options=operation.options,
bases=operation.bases,
managers=operation.managers,
)
]

4
tests/migrations/models.py
View File

@@ -52,6 +52,10 @@ class UnmigratedModel(models.Model):
pass
class EmptyManager(models.Manager):
use_in_migrations = True
class FoodQuerySet(models.query.QuerySet):
pass

97
tests/migrations/test_optimizer.py
View File

@@ -5,6 +5,8 @@ from django.db.migrations.optimizer import MigrationOptimizer
from django.db import migrations
from django.db import models
from .models import CustomModelBase, EmptyManager
class OptimizerTests(TestCase):
"""
@@ -55,13 +57,26 @@ class OptimizerTests(TestCase):
"""
CreateModel should absorb RenameModels.
"""
managers = [('objects', EmptyManager())]
self.assertOptimizesTo(
[
migrations.CreateModel("Foo", [("name", models.CharField(max_length=255))]),
migrations.CreateModel(
name="Foo",
fields=[("name", models.CharField(max_length=255))],
options={'verbose_name': 'Foo'},
bases=(CustomModelBase),
managers=managers,
),
migrations.RenameModel("Foo", "Bar"),
],
[
migrations.CreateModel("Bar", [("name", models.CharField(max_length=255))]),
migrations.CreateModel(
"Bar",
[("name", models.CharField(max_length=255))],
options={'verbose_name': 'Foo'},
bases=(CustomModelBase),
managers=managers,
)
],
)
@@ -159,16 +174,29 @@ class OptimizerTests(TestCase):
"""
AddField should optimize into CreateModel.
"""
managers = [('objects', EmptyManager())]
self.assertOptimizesTo(
[
migrations.CreateModel("Foo", [("name", models.CharField(max_length=255))]),
migrations.CreateModel(
name="Foo",
fields=[("name", models.CharField(max_length=255))],
options={'verbose_name': 'Foo'},
bases=(CustomModelBase),
managers=managers,
),
migrations.AddField("Foo", "age", models.IntegerField()),
],
[
migrations.CreateModel("Foo", [
migrations.CreateModel(
name="Foo",
fields=[
("name", models.CharField(max_length=255)),
("age", models.IntegerField()),
]),
],
options={'verbose_name': 'Foo'},
bases=(CustomModelBase),
managers=managers,
),
],
)
@@ -214,15 +242,28 @@ class OptimizerTests(TestCase):
"""
AlterField should optimize into CreateModel.
"""
managers = [('objects', EmptyManager())]
self.assertOptimizesTo(
[
migrations.CreateModel("Foo", [("name", models.CharField(max_length=255))]),
migrations.CreateModel(
name="Foo",
fields=[("name", models.CharField(max_length=255))],
options={'verbose_name': 'Foo'},
bases=(CustomModelBase),
managers=managers,
),
migrations.AlterField("Foo", "name", models.IntegerField()),
],
[
migrations.CreateModel("Foo", [
migrations.CreateModel(
name="Foo",
fields=[
("name", models.IntegerField()),
]),
],
options={'verbose_name': 'Foo'},
bases=(CustomModelBase),
managers=managers,
),
],
)
@@ -230,15 +271,28 @@ class OptimizerTests(TestCase):
"""
RenameField should optimize into CreateModel.
"""
managers = [('objects', EmptyManager())]
self.assertOptimizesTo(
[
migrations.CreateModel("Foo", [("name", models.CharField(max_length=255))]),
migrations.CreateModel(
name="Foo",
fields=[("name", models.CharField(max_length=255))],
options={'verbose_name': 'Foo'},
bases=(CustomModelBase),
managers=managers,
),
migrations.RenameField("Foo", "name", "title"),
],
[
migrations.CreateModel("Foo", [
migrations.CreateModel(
name="Foo",
fields=[
("title", models.CharField(max_length=255)),
]),
],
options={'verbose_name': 'Foo'},
bases=(CustomModelBase),
managers=managers,
),
],
)
@@ -277,18 +331,31 @@ class OptimizerTests(TestCase):
"""
RemoveField should optimize into CreateModel.
"""
managers = [('objects', EmptyManager())]
self.assertOptimizesTo(
[
migrations.CreateModel("Foo", [
migrations.CreateModel(
name="Foo",
fields=[
("name", models.CharField(max_length=255)),
("age", models.IntegerField()),
]),
],
options={'verbose_name': 'Foo'},
bases=(CustomModelBase),
managers=managers,
),
migrations.RemoveField("Foo", "age"),
],
[
migrations.CreateModel("Foo", [
migrations.CreateModel(
name="Foo",
fields=[
("name", models.CharField(max_length=255)),
]),
],
options={'verbose_name': 'Foo'},
bases=(CustomModelBase),
managers=managers,
),
],
)