From f16619818708e004cb48c2e9b724de3c2901b5ea Mon Sep 17 00:00:00 2001
From: katdom13 <katdom13@gmail.com>
Date: Fri, 14 Jun 2024 18:11:45 +0800
Subject: [PATCH] Fix test failures on moto

---
 wagtail_storages/tests/base.py                     |  4 ++--
 wagtail_storages/tests/test_protected_documents.py |  4 ++--
 wagtail_storages/tests/test_signal_handlers.py     | 10 +++++-----
 wagtail_storages/tests/test_wagtail_hooks.py       |  4 ++--
 4 files changed, 11 insertions(+), 11 deletions(-)

diff --git a/wagtail_storages/tests/base.py b/wagtail_storages/tests/base.py
index b9491f6..222d8b3 100644
--- a/wagtail_storages/tests/base.py
+++ b/wagtail_storages/tests/base.py
@@ -1,10 +1,10 @@
 from django.conf import settings
 
 import boto3
-from moto import mock_s3
+from moto import mock_aws
 
 
-@mock_s3
+@mock_aws
 class CreateBucket:
     def setUp(self):
         bucket_name = settings.AWS_STORAGE_BUCKET_NAME
diff --git a/wagtail_storages/tests/test_protected_documents.py b/wagtail_storages/tests/test_protected_documents.py
index 66034ad..12c6fc2 100644
--- a/wagtail_storages/tests/test_protected_documents.py
+++ b/wagtail_storages/tests/test_protected_documents.py
@@ -3,14 +3,14 @@
 from django.test import TestCase
 from django.urls import reverse
 
-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.tests.utils import is_s3_object_is_public
 
 
-@mock_s3
+@mock_aws
 class AmazonS3DocumentTests(CreateBucket, TestCase):
     def check_s3_url(self, url):
         return "s3.amazonaws.com" in url or "media.torchbox.com" in url
diff --git a/wagtail_storages/tests/test_signal_handlers.py b/wagtail_storages/tests/test_signal_handlers.py
index 93d8afc..1343dcb 100644
--- a/wagtail_storages/tests/test_signal_handlers.py
+++ b/wagtail_storages/tests/test_signal_handlers.py
@@ -5,7 +5,7 @@
 from django.test import TestCase, override_settings
 
 import factory
-from moto import mock_s3
+from moto import mock_aws
 
 from wagtail_storages.factories import (
     CollectionFactory,
@@ -39,7 +39,7 @@ def test_skipping_s3_storage_decorator_with_s3_storage(self):
         mock.assert_called_once()
 
 
-@mock_s3
+@mock_aws
 class TestUpdateDocumentAclsWhenCollectionSaved(CreateBucket, TestCase):
     @factory.django.mute_signals(post_save)
     def test_s3_object_acl_set_to_public(self):
@@ -60,7 +60,7 @@ def test_s3_object_acl_set_to_private(self):
         self.assertFalse(is_s3_object_is_public(document.file.file.obj))
 
 
-@mock_s3
+@mock_aws
 class TestUpdateDocumentAclsWhenDocumentSaved(CreateBucket, TestCase):
     @factory.django.mute_signals(post_save)
     def test_s3_object_acl_set_to_public(self):
@@ -80,7 +80,7 @@ def test_s3_object_acl_set_to_private(self):
         self.assertFalse(is_s3_object_is_public(document.file.file.obj))
 
 
-@mock_s3
+@mock_aws
 class TestPurgeDocumentsWhenCollectionSavedWithRestrictions(CreateBucket, TestCase):
     @override_settings(
         WAGTAILFRONTENDCACHE={
@@ -135,7 +135,7 @@ def test_cache_not_purged_for_public_collection(self):
         urlopen_mock.assert_not_called()
 
 
-@mock_s3
+@mock_aws
 class TestPurgeDocumentFromCacheWhenSaved(CreateBucket, TestCase):
     @override_settings(
         WAGTAILFRONTENDCACHE={
diff --git a/wagtail_storages/tests/test_wagtail_hooks.py b/wagtail_storages/tests/test_wagtail_hooks.py
index 6459ed7..1e83306 100644
--- 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"