Allow empty custom context in GraphQLView (#106)

* Allow empty custom context in GraphQLView

* chore: black

---------

Co-authored-by: Kien Dang <mail@kien.ai>

Author: leonardwellthy

graphql_server/aiohttp/graphqlview.py

@@ -76,7 +76,7 @@ def get_root_value(self):
     def get_context(self, request):
         context = (
             copy.copy(self.context)
-            if self.context and isinstance(self.context, MutableMapping)
+            if self.context is not None and isinstance(self.context, MutableMapping)
             else {}
         )
         if isinstance(context, MutableMapping) and "request" not in context:

graphql_server/flask/graphqlview.py

@@ -73,7 +73,7 @@ def get_root_value(self):
     def get_context(self):
         context = (
             copy.copy(self.context)
-            if self.context and isinstance(self.context, MutableMapping)
+            if self.context is not None and isinstance(self.context, MutableMapping)
             else {}
         )
         if isinstance(context, MutableMapping) and "request" not in context:

graphql_server/quart/graphqlview.py

@@ -77,7 +77,7 @@ def get_root_value(self):
     def get_context(self):
         context = (
             copy.copy(self.context)
-            if self.context and isinstance(self.context, MutableMapping)
+            if self.context is not None and isinstance(self.context, MutableMapping)
             else {}
         )
         if isinstance(context, MutableMapping) and "request" not in context:

graphql_server/sanic/graphqlview.py

@@ -78,7 +78,7 @@ def get_root_value(self):
     def get_context(self, request):
         context = (
             copy.copy(self.context)
-            if self.context and isinstance(self.context, MutableMapping)
+            if self.context is not None and isinstance(self.context, MutableMapping)
             else {}
         )
         if isinstance(context, MutableMapping) and "request" not in context:

graphql_server/webob/graphqlview.py

@@ -72,7 +72,7 @@ def get_root_value(self):
     def get_context(self, request):
         context = (
             copy.copy(self.context)
-            if self.context and isinstance(self.context, MutableMapping)
+            if self.context is not None and isinstance(self.context, MutableMapping)
             else {}
         )
         if isinstance(context, MutableMapping) and "request" not in context:

tests/aiohttp/schema.py

@@ -35,6 +35,9 @@ def resolve_raises(*_):
                         GraphQLNonNull(GraphQLString),
                         resolve=lambda obj, info: info.context["request"],
                     ),
+                    "property": GraphQLField(
+                        GraphQLString, resolve=lambda obj, info: info.context.property
+                    ),
                 },
             ),
             resolve=lambda obj, info: info.context,

tests/aiohttp/test_graphqlview.py

@@ -539,6 +539,24 @@ async def test_context_remapped_if_not_mapping(app, client):
     assert "Request" in _json["data"]["context"]["request"]
 
 
+class CustomContext(dict):
+    property = "A custom property"
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize("app", [create_app(context=CustomContext())])
+async def test_allow_empty_custom_context(app, client):
+    response = await client.get(url_string(query="{context { property request }}"))
+
+    _json = await response.json()
+    assert response.status == 200
+    assert "data" in _json
+    assert "request" in _json["data"]["context"]
+    assert "property" in _json["data"]["context"]
+    assert "A custom property" == _json["data"]["context"]["property"]
+    assert "Request" in _json["data"]["context"]["request"]
+
+
 @pytest.mark.asyncio
 @pytest.mark.parametrize("app", [create_app(context={"request": "test"})])
 async def test_request_not_replaced(app, client):

tests/flask/schema.py

@@ -29,6 +29,9 @@ def resolve_raises(*_):
                         GraphQLNonNull(GraphQLString),
                         resolve=lambda obj, info: info.context["request"],
                     ),
+                    "property": GraphQLField(
+                        GraphQLString, resolve=lambda obj, info: info.context.property
+                    ),
                 },
             ),
             resolve=lambda obj, info: info.context,

tests/flask/test_graphqlview.py

@@ -499,6 +499,23 @@ def test_context_remapped_if_not_mapping(app, client):
     assert "Request" in res["data"]["context"]["request"]
 
 
+class CustomContext(dict):
+    property = "A custom property"
+
+
+@pytest.mark.parametrize("app", [create_app(context=CustomContext())])
+def test_allow_empty_custom_context(app, client):
+    response = client.get(url_string(app, query="{context { property request }}"))
+
+    assert response.status_code == 200
+    res = response_json(response)
+    assert "data" in res
+    assert "request" in res["data"]["context"]
+    assert "property" in res["data"]["context"]
+    assert "A custom property" == res["data"]["context"]["property"]
+    assert "Request" in res["data"]["context"]["request"]
+
+
 def test_post_multipart_data(app, client):
     query = "mutation TestMutation { writeTest { test } }"
     response = client.post(

tests/quart/schema.py

@@ -32,6 +32,9 @@ def resolve_raises(*_):
                         GraphQLNonNull(GraphQLString),
                         resolve=lambda obj, info: info.context["request"],
                     ),
+                    "property": GraphQLField(
+                        GraphQLString, resolve=lambda obj, info: info.context.property
+                    ),
                 },
             ),
             resolve=lambda obj, info: info.context,