1 files changed, 3 insertions, 14 deletions
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):