graphql-python/graphene-django
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene-django.git synced 2025-07-13 01:32:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add headers arg to GraphQLTestCase.query

Browse Source
This commit is contained in:
st. chibie 2019-12-14 18:04:44 +01:00
parent 968002f155
commit bc20a38936
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
graphene_django/utils/testing.py
View File

@ -24,7 +24,7 @@ class GraphQLTestCase(TestCase):
cls._client = Client()
def query(self, query, op_name=None, input_data=None, variables=None):
def query(self, query, op_name=None, input_data=None, variables=None, headers=None):
"""
Args:
query (string) - GraphQL query to run
@ -36,7 +36,9 @@ class GraphQLTestCase(TestCase):
are provided, the ``input`` field in the ``variables``
dict will be overwritten with this value.
variables (dict) - If provided, the "variables" field in GraphQL will be
set to this value.
set to this value.
headers (dict) - If provided, the headers in POST request to GRAPHQL_URL
will be set to this value.
Returns:
Response object from client
@ -53,7 +55,7 @@ class GraphQLTestCase(TestCase):
body["variables"] = {"input": input_data}
resp = self._client.post(
self.GRAPHQL_URL, json.dumps(body), content_type="application/json"
self.GRAPHQL_URL, json.dumps(body), content_type="application/json", **headers
)
return resp