add decode error

Author: vvcheremushkin

star_resty/exceptions.py

@@ -2,6 +2,8 @@
 
 from marshmallow.exceptions import ValidationError
 
+__all__ = ('StarRestError', 'DumpError', 'DecodeError')
+
 
 class StarRestError(Exception):
     pass
@@ -18,3 +20,9 @@ def normalized_messages(self) -> Dict:
             return self.orig_exc.normalized_messages()
 
         return {'_schema': 'Error dump content'}
+
+
+class DecodeError(StarRestError):
+
+    def normalized_messages(self) -> Dict:
+        return {'_body': str(self)}

star_resty/types/json.py

@@ -1,9 +1,11 @@
 import types
 from typing import Mapping, Type, TypeVar, Union
 
+import ujson
 from marshmallow import EXCLUDE, Schema
 from starlette.requests import Request
 
+from star_resty.exceptions import DecodeError
 from .parser import Parser, set_parser
 
 __all__ = ('json_schema', 'json_payload')
@@ -32,8 +34,13 @@ def media_type(self):
         return 'application/json'
 
     async def parse(self, request: Request):
-        body = await request.json()
+        body = await request.body()
         if body is None:
-            body = {}
-
-        return self.schema.load(body, unknown=self.unknown)
+            data = {}
+        else:
+            try:
+                data = ujson.loads(body)
+            except (TypeError, ValueError) as e:
+                raise DecodeError('Invalid json body') from e
+
+        return self.schema.load(data, unknown=self.unknown)

tests/test_method.py

@@ -15,7 +15,7 @@
 async def test_create_user():
     request = mock.MagicMock(spec_set=Request)
     request.path_params = {'id': 1}
-    request.json.return_value = {'name': 'Name', 'email': '[email protected]'}
+    request.body.return_value = json.dumps({'name': 'Name', 'email': '[email protected]'}).encode('utf8')
 
     endpoint = CreateUser.as_endpoint()
     resp = await endpoint(request)