Fix support of get_full_details() for Throttled exceptions

ticosax · ticosax · commit 15f85d6be65d · 2016-10-26T15:49:36.000+02:00
Since `str` objects are immutable, appending to existing `str` creates
in fact a new `str` instance.

Thus `ErrorDetail.detail.code` attribute is lost after `str` concatenation operation.
diff --git a/rest_framework/exceptions.py b/rest_framework/exceptions.py
@@ -210,14 +210,15 @@ class Throttled(APIException):
     default_code = 'throttled'
 
     def __init__(self, wait=None, detail=None, code=None):
-        super(Throttled, self).__init__(detail, code)
 
         if wait is None:
             self.wait = None
         else:
             self.wait = math.ceil(wait)
-            self.detail += ' ' + force_text(ungettext(
-                self.extra_detail_singular.format(wait=self.wait),
-                self.extra_detail_plural.format(wait=self.wait),
-                self.wait
-            ))
+            if detail is None:
+                detail = ' '.join((
+                    force_text(self.default_detail),
+                    force_text(ungettext(self.extra_detail_singular.format(wait=self.wait),
+                                         self.extra_detail_plural.format(wait=self.wait),
+                                         self.wait))))
+        super(Throttled, self).__init__(detail, code)
diff --git a/tests/test_exceptions.py b/tests/test_exceptions.py
@@ -1,9 +1,12 @@
 from __future__ import unicode_literals
 
 from django.test import TestCase
+from django.utils import six
 from django.utils.translation import ugettext_lazy as _
 
-from rest_framework.exceptions import ErrorDetail, _get_error_details
+from rest_framework.exceptions import (
+    ErrorDetail, Throttled, _get_error_details
+)
 
 
 class ExceptionTestCase(TestCase):
@@ -39,3 +42,13 @@ def test_get_error_details(self):
             _get_error_details([[lazy_example]])[0][0],
             ErrorDetail
         )
+
+    def test_get_full_details_with_throttling(self):
+        exception = Throttled()
+        assert exception.get_full_details() == {
+            'message': 'Request was throttled.', 'code': 'throttled'}
+
+        exception = Throttled(wait=2)
+        assert exception.get_full_details() == {
+            'message': 'Request was throttled. Expected available in {} seconds.'.format(2 if six.PY3 else 2.),
+            'code': 'throttled'}
