Test byte decoding of _raise_for_5xx.

Author: ab

tests/test_oauth2_session.py

@@ -535,6 +535,34 @@ def fake_send(r, **kwargs):
                 sess.fetch_token(url)
             self.assertTrue(sess.authorized)
 
+    def test_raise_for_5xx(self):
+        for reason_bytes in [
+            b"\xa1An error occurred!",  # iso-8859-i
+            b"\xc2\xa1An error occurred!",  # utf-8
+        ]:
+            fake_resp = mock.MagicMock()
+            fake_resp.status_code = 504
+            fake_resp.reason = reason_bytes
+            reason_unicode = "¡An error occurred!"
+            fake_resp.url = "https://example.com/token"
+            expected = (
+                "504 Server Error: " + reason_unicode + " for url: " + fake_resp.url
+            )
+
+            # Verify that our sample is unicode in both Python2 and Python3
+            try:
+                unicode("")
+            except NameError:
+                unicode = str
+            self.assertIsInstance(reason_unicode, unicode)
+
+            sess = OAuth2Session("test-id")
+
+            with self.assertRaises(HTTPError) as cm:
+                sess._raise_for_5xx(fake_resp)
+
+            self.assertEqual(cm.exception.args[0], expected)
+
 
 class OAuth2SessionNetrcTest(OAuth2SessionTest):
     """Ensure that there is no magic auth handling.