diff --git a/examples/starwars_relay/tests/test_objectidentification.py b/examples/starwars_relay/tests/test_objectidentification.py index 87394421..327d5b0c 100644 --- a/examples/starwars_relay/tests/test_objectidentification.py +++ b/examples/starwars_relay/tests/test_objectidentification.py @@ -56,7 +56,7 @@ type Ship implements Node { type ShipConnection { pageInfo: PageInfo! - edges: [ShipEdge] + edges: [ShipEdge]! } type ShipEdge { diff --git a/graphene/relay/tests/test_connection.py b/graphene/relay/tests/test_connection.py index 18d890c1..87f937ae 100644 --- a/graphene/relay/tests/test_connection.py +++ b/graphene/relay/tests/test_connection.py @@ -28,8 +28,9 @@ def test_connection(): pageinfo_field = fields['page_info'] assert isinstance(edge_field, Field) - assert isinstance(edge_field.type, List) - assert edge_field.type.of_type == MyObjectConnection.Edge + assert isinstance(edge_field.type, NonNull) + assert isinstance(edge_field.type.of_type, List) + assert edge_field.type.of_type.of_type == MyObjectConnection.Edge assert isinstance(pageinfo_field, Field) assert isinstance(pageinfo_field.type, NonNull)