From 990c4c4e9d6f274771748d16a237f9821a846feb Mon Sep 17 00:00:00 2001 From: allen-munsch Date: Mon, 24 Oct 2022 11:22:41 -0500 Subject: [PATCH] fix test --- graphene/types/schema.py | 2 +- graphene/types/tests/test_schema.py | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/graphene/types/schema.py b/graphene/types/schema.py index fd9e988c..41e15f74 100644 --- a/graphene/types/schema.py +++ b/graphene/types/schema.py @@ -453,7 +453,7 @@ class Schema: return _type def get_type(self, name): - return self.graphql_schema.type_map.get(name).graphene_type + return self.graphql_schema.type_map.get(name) def lazy(self, _type): return lambda: self.get_type(_type) diff --git a/graphene/types/tests/test_schema.py b/graphene/types/tests/test_schema.py index 2ca7468d..21bdb8d1 100644 --- a/graphene/types/tests/test_schema.py +++ b/graphene/types/tests/test_schema.py @@ -32,7 +32,10 @@ def test_schema_get_type(): schema = Schema(Query) assert schema.Query == Query assert schema.MyOtherType == MyOtherType - assert schema.get_type("MyOtherType") == MyOtherType + name = "MyOtherType" + assert isinstance(schema.get_type(name), GraphQLObjectType) + assert schema.get_type(name).name == name + assert schema.get_type(name).graphene_type == MyOtherType def test_schema_get_type_error():