Fix test failures on moto

torchbox · Jun 14, 2024 · b71e895 · b71e895
1 parent ed83be6
commit b71e895
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wagtail_storages/tests/test_wagtail_hooks.py b/wagtail_storages/tests/test_wagtail_hooks.py
@@ -2,14 +2,14 @@
 
 from django.test import RequestFactory, TestCase, override_settings
 
-from moto import mock_s3
+from moto import mock_aws
 
 from wagtail_storages.factories import CollectionViewRestrictionFactory, DocumentFactory
 from wagtail_storages.tests.base import CreateBucket
 from wagtail_storages.wagtail_hooks import serve_document_from_s3
 
 
-@mock_s3
+@mock_aws
 class TestWagtailHooks(CreateBucket, TestCase):
     @override_settings(
         DEFAULT_FILE_STORAGE="django.core.files.storage.FileSystemStorage"