Improved backend tests

Author: syrusakbary

graphql/backend/compiled.py

@@ -1,3 +1,4 @@
+from six import string_types
 from .base import GraphQLDocument
 
 
@@ -7,6 +8,9 @@ def from_code(cls, schema, code, uptodate=None, extra_namespace=None):
  """Creates a GraphQLDocument object from compiled code and the globals. This
  is used by the loaders and schema to create a document object.
  """
+ if isinstance(code, string_types):
+ filename = '<document>'
+ code = compile(code, filename, 'exec')
  namespace = {"__file__": code.co_filename}
  exec(code, namespace)
  if extra_namespace:

graphql/backend/tests/test_base.py

@@ -0,0 +1,23 @@
+import pytest
+from .. import get_default_backend, set_default_backend, GraphQLCoreBackend
+
+
+def test_get_default_backend_returns_core_by_default():
+ backend = get_default_backend()
+ assert isinstance(backend, GraphQLCoreBackend)
+
+
+def test_set_default_backend():
+ default_backend = get_default_backend()
+ new_backend = GraphQLCoreBackend()
+ assert new_backend != default_backend
+ set_default_backend(new_backend)
+ assert get_default_backend() == new_backend
+
+
+def test_set_default_backend_fails_if_invalid_backend():
+ default_backend = get_default_backend()
+ with pytest.raises(Exception) as exc_info:
+ set_default_backend(object())
+ assert str(exc_info.value) == 'backend must be an instance of GraphQLBackend.'
+ assert get_default_backend() == default_backend

graphql/backend/tests/test_cache.py

@@ -10,8 +10,15 @@
 from .schema import schema
 
 
-def test_backend_is_cached_when_needed():
+def test_cached_backend():
  cached_backend = GraphQLCachedBackend(GraphQLCoreBackend())
  document1 = cached_backend.document_from_string(schema, "{ hello }")
  document2 = cached_backend.document_from_string(schema, "{ hello }")
  assert document1 == document2
+
+
+def test_cached_backend_with_use_consistent_hash():
+ cached_backend = GraphQLCachedBackend(GraphQLCoreBackend(), use_consistent_hash=True)
+ document1 = cached_backend.document_from_string(schema, "{ hello }")
+ document2 = cached_backend.document_from_string(schema, "{ hello }")
+ assert document1 == document2

graphql/backend/tests/test_compileddocument.py

@@ -0,0 +1,56 @@
+from ...language.base import parse
+from ...utils.ast_to_code import ast_to_code
+from ..compiled import GraphQLCompiledDocument
+from .schema import schema
+
+
+def test_compileddocument_from_module_dict():
+ document_string = '{ hello }'
+ document_ast = parse(document_string)
+ document = GraphQLCompiledDocument.from_module_dict(schema, {
+ 'document_string': document_string,
+ 'document_ast': document_ast,
+ 'execute': lambda *_: True
+ })
+ assert document.operations_map == {
+ None: 'query'
+ }
+ assert document.document_string == document_string
+ assert document.document_ast == document_ast
+ assert document.schema == schema
+ assert document.execute()
+
+
+def test_compileddocument_from_code():
+ document_string = '{ hello }'
+ document_ast = parse(document_string)
+ code = '''
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from graphql.language import ast
+from graphql.language.parser import Loc
+from graphql.language.source import Source
+
+
+schema = None
+document_string = """{document_string}"""
+source = Source(document_string)
+
+
+def loc(start, end):
+ return Loc(start, end, source)
+
+document_ast = {document_ast}
+
+def execute(*_):
+ return True
+'''.format(document_string=document_string, document_ast=ast_to_code(document_ast))
+ document = GraphQLCompiledDocument.from_code(schema, code)
+ assert document.operations_map == {
+ None: 'query'
+ }
+ assert document.document_string == document_string
+ assert document.document_ast == document_ast
+ assert document.schema == schema
+ assert document.execute()

setup.cfg

@@ -2,5 +2,11 @@
 exclude = tests,scripts,setup.py,docs,graphql/execution/executors/asyncio_utils.py
 max-line-length = 160
 
+[coverage:run]
+omit=graphql/backend/quiver_cloud.py,tests,*/tests/*
+
+[coverage:report]
+omit=graphql/backend/quiver_cloud.py,tests,*/tests/*
+
 [bdist_wheel]
 universal=1