Add back pretty argument to json_encode

Author: Cito

graphql_server/__init__.py

@@ -185,13 +185,15 @@ def load_json_body(data):
         raise HttpQueryError(400, "POST body sent invalid JSON.")
 
 
-def json_encode(data):
-    # type: (Union[Dict,List]) -> str
+def json_encode(data, pretty=False):
+    # type: (Union[Dict,List],Optional[bool]) -> str
     """Serialize the given data(a dictionary or a list) using JSON.
 
     The given data (a dictionary or a list) will be serialized using JSON
     and returned as a string that will be nicely formatted if you set pretty=True.
     """
+    if pretty:
+        return json_encode_pretty(data)
     return json.dumps(data, separators=(",", ":"))
 
 

tests/test_helpers.py

@@ -25,6 +25,13 @@ def test_json_encode_pretty():
     assert result == '{\n  "query": "{test}"\n}'
 
 
+def test_json_encode_with_pretty_argument():
+    result = json_encode({"query": "{test}"}, pretty=False)
+    assert result == '{"query":"{test}"}'
+    result = json_encode({"query": "{test}"}, pretty=True)
+    assert result == '{\n  "query": "{test}"\n}'
+
+
 def test_load_json_body_as_dict():
     result = load_json_body('{"query": "{test}"}')
     assert result == {"query": "{test}"}