Refs #34730 -- Made Message importable from django.contrib.messages.

4 files changed, 4 insertions, 8 deletions

diff --git a/tests/messages_tests/base.py b/tests/messages_tests/base.py
index d89e186278..6fe8892ac1 100644
--- a/tests/messages_tests/base.py
+++ b/tests/messages_tests/base.py
@@ -1,8 +1,7 @@
-from django.contrib.messages import constants, get_level, set_level
+from django.contrib.messages import Message, constants, get_level, set_level
 from django.contrib.messages.api import MessageFailure
 from django.contrib.messages.constants import DEFAULT_LEVELS
 from django.contrib.messages.storage import default_storage
-from django.contrib.messages.storage.base import Message
 from django.http import HttpRequest, HttpResponse
 from django.test import modify_settings, override_settings
 from django.urls import reverse
diff --git a/tests/messages_tests/test_cookie.py b/tests/messages_tests/test_cookie.py
index 344df96886..8fd6fa3d61 100644
--- a/tests/messages_tests/test_cookie.py
+++ b/tests/messages_tests/test_cookie.py
@@ -3,8 +3,7 @@ import random
 from unittest import TestCase
 
 from django.conf import settings
-from django.contrib.messages import constants
-from django.contrib.messages.storage.base import Message
+from django.contrib.messages import Message, constants
 from django.contrib.messages.storage.cookie import (
     CookieStorage,
     MessageDecoder,
diff --git a/tests/messages_tests/test_session.py b/tests/messages_tests/test_session.py
index 07be6063f7..cfc7d3e8c5 100644
--- a/tests/messages_tests/test_session.py
+++ b/tests/messages_tests/test_session.py
@@ -1,5 +1,4 @@
-from django.contrib.messages import constants
-from django.contrib.messages.storage.base import Message
+from django.contrib.messages import Message, constants
 from django.contrib.messages.storage.session import SessionStorage
 from django.core.exceptions import ImproperlyConfigured
 from django.http import HttpRequest
diff --git a/tests/messages_tests/tests.py b/tests/messages_tests/tests.py
index 2aa8935474..1ebb6aaba2 100644
--- a/tests/messages_tests/tests.py
+++ b/tests/messages_tests/tests.py
@@ -1,8 +1,7 @@
 from unittest import mock
 
-from django.contrib.messages import constants
+from django.contrib.messages import Message, constants
 from django.contrib.messages.storage import base
-from django.contrib.messages.storage.base import Message
 from django.test import SimpleTestCase, override_settings