Fixed #27688 -- Made messages' add_message() request check use ducktyping.

1 files changed, 21 insertions, 0 deletions

diff --git a/tests/messages_tests/test_api.py b/tests/messages_tests/test_api.py
index 0cb17c1656..8baff57c70 100644
--- a/tests/messages_tests/test_api.py
+++ b/tests/messages_tests/test_api.py
@@ -42,3 +42,24 @@ class ApiTests(SimpleTestCase):
     def test_middleware_missing_silently(self):
         messages.add_message(self.request, messages.DEBUG, 'some message', fail_silently=True)
         self.assertEqual(self.storage.store, [])
+
+
+class CustomRequest(object):
+    def __init__(self, request):
+        self._request = request
+
+    def __getattribute__(self, attr):
+        try:
+            return super(CustomRequest, self).__getattribute__(attr)
+        except AttributeError:
+            return getattr(self._request, attr)
+
+
+class CustomRequestApiTests(ApiTests):
+    """
+    add_message() should use ducktyping to allow request wrappers such as the
+    one in Django REST framework.
+    """
+    def setUp(self):
+        super(CustomRequestApiTests, self).setUp()
+        self.request = CustomRequest(self.request)