diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/check_framework/test_migrations.py | 101 | ||||
| -rw-r--r-- | tests/db_functions/migrations/0001_setup_extensions.py | 17 | ||||
| -rw-r--r-- | tests/postgres_tests/migrations/0001_setup_extensions.py | 33 |
3 files changed, 14 insertions, 137 deletions
diff --git a/tests/check_framework/test_migrations.py b/tests/check_framework/test_migrations.py deleted file mode 100644 index 0b00690e77..0000000000 --- a/tests/check_framework/test_migrations.py +++ /dev/null @@ -1,101 +0,0 @@ -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", - ) - ], - ) diff --git a/tests/db_functions/migrations/0001_setup_extensions.py b/tests/db_functions/migrations/0001_setup_extensions.py index 5909a96eb8..0289055499 100644 --- a/tests/db_functions/migrations/0001_setup_extensions.py +++ b/tests/db_functions/migrations/0001_setup_extensions.py @@ -1,22 +1,11 @@ -from django.db import migrations -from django.db.migrations.operations.base import Operation - - -class DummyOperation(Operation): - def state_forwards(self, app_label, state): - pass - - def database_forwards(self, app_label, schema_editor, from_state, to_state): - pass - - def database_backwards(self, app_label, schema_editor, from_state, to_state): - pass +from unittest import mock +from django.db import migrations try: from django.contrib.postgres.operations import CryptoExtension except ImportError: - CryptoExtension = DummyOperation + CryptoExtension = mock.Mock() class Migration(migrations.Migration): diff --git a/tests/postgres_tests/migrations/0001_setup_extensions.py b/tests/postgres_tests/migrations/0001_setup_extensions.py index 3edeff7b82..090abf9649 100644 --- a/tests/postgres_tests/migrations/0001_setup_extensions.py +++ b/tests/postgres_tests/migrations/0001_setup_extensions.py @@ -1,17 +1,6 @@ -from django.db import connection, migrations -from django.db.migrations.operations.base import Operation - - -class DummyOperation(Operation): - def state_forwards(self, app_label, state): - pass - - def database_forwards(self, app_label, schema_editor, from_state, to_state): - pass - - def database_backwards(self, app_label, schema_editor, from_state, to_state): - pass +from unittest import mock +from django.db import connection, migrations try: from django.contrib.postgres.operations import ( @@ -26,14 +15,14 @@ try: UnaccentExtension, ) except ImportError: - BloomExtension = DummyOperation - BtreeGinExtension = DummyOperation - BtreeGistExtension = DummyOperation - CITextExtension = DummyOperation - CreateExtension = DummyOperation - HStoreExtension = DummyOperation - TrigramExtension = DummyOperation - UnaccentExtension = DummyOperation + BloomExtension = mock.Mock() + BtreeGinExtension = mock.Mock() + BtreeGistExtension = mock.Mock() + CITextExtension = mock.Mock() + CreateExtension = mock.Mock() + HStoreExtension = mock.Mock() + TrigramExtension = mock.Mock() + UnaccentExtension = mock.Mock() needs_crypto_extension = False else: needs_crypto_extension = ( @@ -52,7 +41,7 @@ class Migration(migrations.Migration): # dash in its name. CreateExtension("uuid-ossp"), # CryptoExtension is required for RandomUUID() on PostgreSQL < 13. - CryptoExtension() if needs_crypto_extension else DummyOperation(), + CryptoExtension() if needs_crypto_extension else mock.Mock(), HStoreExtension(), TrigramExtension(), UnaccentExtension(), |