Fixed tests

Author: syrusakbary

examples/starwars/schema.py

@@ -43,9 +43,7 @@ class Input:
     faction = graphene.Field(Faction)
 
     @classmethod
-    def mutate_and_get_payload(cls, info, **input):
-        ship_name = input.get('ship_name')
-        faction_id = input.get('faction_id')
+    def mutate_and_get_payload(cls, root, info, ship_name, faction_id, client_mutation_id=None):
         faction_key = ndb.Key(FactionModel, faction_id)
         ship = create_ship(ship_name, faction_key)
         faction = faction_key.get()

graphene_gae/ndb/types.py

@@ -114,7 +114,7 @@ def is_type_of(cls, root, info):
         return type(root) == cls._meta.model
 
     @classmethod
-    def get_node(cls, urlsafe_key, *_):
+    def get_node(cls, info, urlsafe_key):
         try:
             key = ndb.Key(urlsafe=urlsafe_key)
         except:

tests/_ndb/test_types_relay.py

@@ -89,12 +89,12 @@ class QueryRoot(graphene.ObjectType):
 class TestNDBTypesRelay(BaseTest):
 
     def testNdbNode_getNode_invalidId_shouldReturnNone(self):
-        result = ArticleType.get_node("I'm not a valid NDB encoded key")
+        result = ArticleType.get_node(None, "I'm not a valid NDB encoded key")
         self.assertIsNone(result)
 
     def testNdbNode_getNode_validID_entityDoesntExist_shouldReturnNone(self):
         article_key = ndb.Key('Article', 'invalid_id_thats_not_in_db')
-        result = ArticleType.get_node(article_key.urlsafe())
+        result = ArticleType.get_node(None, article_key.urlsafe())
         self.assertIsNone(result)
 
     def testNdbNode_getNode_validID_entityDoes_shouldReturnEntity(self):
@@ -104,7 +104,7 @@ def testNdbNode_getNode_validID_entityDoes_shouldReturnEntity(self):
             author_key=Author(name="John Dow", email="[email protected]").put(),
         ).put()
 
-        result = ArticleType.get_node(article_key.urlsafe())
+        result = ArticleType.get_node(None, article_key.urlsafe())
         article = article_key.get()
 
         self.assertIsNotNone(result)