Add context tests

Author: syrusakbary

Diff for: flask_graphql/graphqlview.py

@@ -48,7 +48,9 @@ def get_root_value(self, request):
         return self.root_value
 
     def get_context(self, request):
-        return self.context or request
+        if self.context is not None:
+            return self.context
+        return request
 
     def get_middleware(self, request):
         return self.middleware

Diff for: tests/schema.py

@@ -13,6 +13,8 @@ def resolve_raises(*_):
         'thrower': GraphQLField(GraphQLNonNull(GraphQLString), resolver=resolve_raises),
         'request': GraphQLField(GraphQLNonNull(GraphQLString),
                                 resolver=lambda obj, args, context, info: context.args.get('q')),
+        'context': GraphQLField(GraphQLNonNull(GraphQLString),
+                                resolver=lambda obj, args, context, info: context),
         'test': GraphQLField(
             type=GraphQLString,
             args={

Diff for: tests/test_graphqlview.py

@@ -426,6 +426,20 @@ def test_passes_request_into_request_context(client):
         }
     }
 
+
+@pytest.mark.parametrize('app', [create_app(context="CUSTOM CONTEXT")])
+def test_supports_pretty_printing(client):
+    response = client.get(url_string(query='{context}'))
+
+
+    assert response.status_code == 200
+    assert response_json(response) == {
+        'data': {
+            'context': 'CUSTOM CONTEXT'
+        }
+    }
+
+
 def test_post_multipart_data(client):
     query = 'mutation TestMutation { writeTest { test } }'
     response = client.post(