test_connection: add test that checks generated types

Author: Cito

tests/connection/test_connection.py

@@ -1,7 +1,8 @@
 from typing import List, NamedTuple
 
-from graphql import graphql_sync
-from graphql.type import (
+from graphql import (
+    graphql_sync,
+    print_schema,
     GraphQLField,
     GraphQLInt,
     GraphQLObjectType,
@@ -17,6 +18,8 @@
     forward_connection_args,
 )
 
+from ..utils import dedent
+
 
 class User(NamedTuple):
     name: str
@@ -175,3 +178,72 @@ def works_with_backward_connection_args():
             },
             None,
         )
+
+    def generates_correct_types():
+        assert print_schema(schema).rstrip() == dedent(
+            '''
+            type Query {
+              user: User
+            }
+
+            type User {
+              name: String
+              friends(after: String, first: Int, before: String, last: Int): FriendConnection
+              friendsForward(after: String, first: Int): UserConnection
+              friendsBackward(before: String, last: Int): UserConnection
+            }
+
+            """A connection to a list of items."""
+            type FriendConnection {
+              """Information to aid in pagination."""
+              pageInfo: PageInfo!
+
+              """A list of edges."""
+              edges: [FriendEdge]
+              totalCount: Int
+            }
+
+            """Information about pagination in a connection."""
+            type PageInfo {
+              """When paginating forwards, are there more items?"""
+              hasNextPage: Boolean!
+
+              """When paginating backwards, are there more items?"""
+              hasPreviousPage: Boolean!
+
+              """When paginating backwards, the cursor to continue."""
+              startCursor: String
+
+              """When paginating forwards, the cursor to continue."""
+              endCursor: String
+            }
+
+            """An edge in a connection."""
+            type FriendEdge {
+              """The item at the end of the edge"""
+              node: User
+
+              """A cursor for use in pagination"""
+              cursor: String!
+              friendshipTime: String
+            }
+
+            """A connection to a list of items."""
+            type UserConnection {
+              """Information to aid in pagination."""
+              pageInfo: PageInfo!
+
+              """A list of edges."""
+              edges: [UserEdge]
+            }
+
+            """An edge in a connection."""
+            type UserEdge {
+              """The item at the end of the edge"""
+              node: User
+
+              """A cursor for use in pagination"""
+              cursor: String!
+            }
+            '''  # noqa: E501
+        )