mirror of
https://github.com/django/django.git
synced 2024-12-22 17:16:24 +00:00
102 lines
3.0 KiB
Python
102 lines
3.0 KiB
Python
from django.core import checks
|
|
from django.db import migrations
|
|
from django.db.migrations.operations.base import Operation
|
|
from django.test import TestCase
|
|
|
|
|
|
class DeprecatedMigrationOperationTests(TestCase):
|
|
def test_default_operation(self):
|
|
class MyOperation(Operation):
|
|
system_check_deprecated_details = {}
|
|
|
|
my_operation = MyOperation()
|
|
|
|
class Migration(migrations.Migration):
|
|
operations = [my_operation]
|
|
|
|
self.assertEqual(
|
|
Migration("name", "app_label").check(),
|
|
[
|
|
checks.Warning(
|
|
msg="MyOperation has been deprecated.",
|
|
obj=my_operation,
|
|
id="migrations.WXXX",
|
|
)
|
|
],
|
|
)
|
|
|
|
def test_user_specified_details(self):
|
|
class MyOperation(Operation):
|
|
system_check_deprecated_details = {
|
|
"msg": "This operation is deprecated and will be removed soon.",
|
|
"hint": "Use something else.",
|
|
"id": "migrations.W999",
|
|
}
|
|
|
|
my_operation = MyOperation()
|
|
|
|
class Migration(migrations.Migration):
|
|
operations = [my_operation]
|
|
|
|
self.assertEqual(
|
|
Migration("name", "app_label").check(),
|
|
[
|
|
checks.Warning(
|
|
msg="This operation is deprecated and will be removed soon.",
|
|
obj=my_operation,
|
|
hint="Use something else.",
|
|
id="migrations.W999",
|
|
)
|
|
],
|
|
)
|
|
|
|
|
|
class RemovedMigrationOperationTests(TestCase):
|
|
def test_default_operation(self):
|
|
class MyOperation(Operation):
|
|
system_check_removed_details = {}
|
|
|
|
my_operation = MyOperation()
|
|
|
|
class Migration(migrations.Migration):
|
|
operations = [my_operation]
|
|
|
|
self.assertEqual(
|
|
Migration("name", "app_label").check(),
|
|
[
|
|
checks.Error(
|
|
msg=(
|
|
"MyOperation has been removed except for support in historical "
|
|
"migrations."
|
|
),
|
|
obj=my_operation,
|
|
id="migrations.EXXX",
|
|
)
|
|
],
|
|
)
|
|
|
|
def test_user_specified_details(self):
|
|
class MyOperation(Operation):
|
|
system_check_removed_details = {
|
|
"msg": "Support for this operation is gone.",
|
|
"hint": "Use something else.",
|
|
"id": "migrations.E999",
|
|
}
|
|
|
|
my_operation = MyOperation()
|
|
|
|
class Migration(migrations.Migration):
|
|
operations = [my_operation]
|
|
|
|
self.assertEqual(
|
|
Migration("name", "app_label").check(),
|
|
[
|
|
checks.Error(
|
|
msg="Support for this operation is gone.",
|
|
obj=my_operation,
|
|
hint="Use something else.",
|
|
id="migrations.E999",
|
|
)
|
|
],
|
|
)
|