Fix support of get_full_details() for Throttled exceptions

ticosax · ticosax · commit bbb5dc9a4312 · 2016-10-26T15:40:41.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
@@ -216,8 +216,11 @@ def __init__(self, wait=None, detail=None, code=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
-            ))
+            code = self.detail.code
+            self.detail = ErrorDetail(
+                ' '.join((self.detail, force_text(ungettext(
+                    self.extra_detail_singular.format(wait=self.wait),
+                    self.extra_detail_plural.format(wait=self.wait),
+                    self.wait
+                )))))
+            self.detail.code = 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'}
