Tracing tests

Author: ekampf

graphql/execution/tests/test_tracing.py

@@ -0,0 +1,167 @@
+
+from graphql.execution import execute
+from graphql.language.parser import parse
+from graphql.type import (GraphQLArgument, GraphQLBoolean, GraphQLField,
+                          GraphQLID, GraphQLInt, GraphQLList, GraphQLNonNull,
+                          GraphQLObjectType, GraphQLSchema, GraphQLString)
+
+current_time = -1
+
+
+def test_tracing_result(mocker):
+
+    def get_time():
+        global current_time
+        current_time += 1
+        return current_time
+
+    time_mock = mocker.patch('time.time')
+    time_mock.side_effect = get_time
+
+    BlogImage = GraphQLObjectType('BlogImage', {
+        'url': GraphQLField(GraphQLString),
+        'width': GraphQLField(GraphQLInt),
+        'height': GraphQLField(GraphQLInt),
+    })
+
+    BlogAuthor = GraphQLObjectType('Author', lambda: {
+        'id': GraphQLField(GraphQLString),
+        'name': GraphQLField(GraphQLString),
+        'pic': GraphQLField(BlogImage,
+                            args={
+                                'width': GraphQLArgument(GraphQLInt),
+                                'height': GraphQLArgument(GraphQLInt),
+                            },
+                            resolver=lambda obj, info, **args:
+                            obj.pic(args['width'], args['height'])
+                            ),
+        'recentArticle': GraphQLField(BlogArticle),
+    })
+
+    BlogArticle = GraphQLObjectType('Article', {
+        'id': GraphQLField(GraphQLNonNull(GraphQLString)),
+        'isPublished': GraphQLField(GraphQLBoolean),
+        'author': GraphQLField(BlogAuthor),
+        'title': GraphQLField(GraphQLString),
+        'body': GraphQLField(GraphQLString),
+        'keywords': GraphQLField(GraphQLList(GraphQLString)),
+    })
+
+    BlogQuery = GraphQLObjectType('Query', {
+        'article': GraphQLField(
+            BlogArticle,
+            args={'id': GraphQLArgument(GraphQLID)},
+            resolver=lambda obj, info, **args: Article(args['id'])),
+        'feed': GraphQLField(
+            GraphQLList(BlogArticle),
+            resolver=lambda *_: map(Article, range(1, 2 + 1))),
+    })
+
+    BlogSchema = GraphQLSchema(BlogQuery)
+
+    class Article(object):
+
+        def __init__(self, id):
+            self.id = id
+            self.isPublished = True
+            self.author = Author()
+            self.title = 'My Article {}'.format(id)
+            self.body = 'This is a post'
+            self.hidden = 'This data is not exposed in the schema'
+            self.keywords = ['foo', 'bar', 1, True, None]
+
+    class Author(object):
+        id = 123
+        name = 'John Smith'
+
+        def pic(self, width, height):
+            return Pic(123, width, height)
+
+        @property
+        def recentArticle(self): return Article(1)
+
+    class Pic(object):
+
+        def __init__(self, uid, width, height):
+            self.url = 'cdn://{}'.format(uid)
+            self.width = str(width)
+            self.height = str(height)
+
+    request = '''
+    {
+        feed {
+          id
+          title
+        }
+    }
+    '''
+
+    # Note: this is intentionally not validating to ensure appropriate
+    # behavior occurs when executing an invalid query.
+    result = execute(BlogSchema, parse(request), tracing=True)
+    assert not result.errors
+    assert result.data == \
+        {
+            "feed": [
+                {
+                    "id": "1",
+                    "title": "My Article 1"
+                },
+                {
+                    "id": "2",
+                    "title": "My Article 2"
+                },
+            ],
+        }
+
+    assert result.extensions['tracing'] == {
+        'version': 1,
+        'startTime': '1970-01-01T00:00:00.fZ',
+        'endTime': '1970-01-01T04:26:40.fZ',
+        'duration': 16000,
+        'execution': {
+            'resolvers': [
+                {
+                    'path': ['feed'],
+                    'parentType': 'Query',
+                    'fieldName': 'feed',
+                    'returnType': '[Article]',
+                    'startOffset': 3000,
+                    'duration': 1000
+                },
+                {
+                    'path': ['feed', 0, 'id'],
+                    'parentType': 'Article',
+                    'fieldName': 'id',
+                    'returnType': 'String!',
+                    'startOffset': 6000,
+                    'duration': 1000
+                },
+                {
+                    'path': ['feed', 0, 'title'],
+                    'parentType': 'Article',
+                    'fieldName': 'title',
+                    'returnType': 'String',
+                    'startOffset': 9000,
+                    'duration': 1000
+                },
+                {
+                    'path': ['feed', 1, 'id'],
+                    'parentType': 'Article',
+                    'fieldName': 'id',
+                    'returnType': 'String!',
+                    'startOffset': 12000,
+                    'duration': 1000
+                },
+                {
+                    'path': ['feed', 1, 'title'],
+                    'parentType': 'Article',
+                    'fieldName': 'title',
+                    'returnType': 'String',
+                    'startOffset': 15000,
+                    'duration': 1000
+                }
+            ]
+        }
+    }
+