File tree 2 files changed +7
-0
lines changed
2 files changed +7
-0
lines changed Original file line number Diff line number Diff line change @@ -103,6 +103,8 @@ def args(self, **kwargs):
103
103
added_args = []
104
104
for name , value in kwargs .items ():
105
105
arg = self .field .args .get (name )
106
+ if not arg :
107
+ raise KeyError (f"Argument { name } does not exist in { self .field } ." )
106
108
arg_type_serializer = get_arg_serializer (arg .type )
107
109
serialized_value = arg_type_serializer (value )
108
110
added_args .append (
Original file line number Diff line number Diff line change @@ -144,3 +144,8 @@ def test_create_review_mutation_result(ds):
144
144
)
145
145
expected = {"createReview" : {"stars" : 5 , "commentary" : "This is a great movie!" }}
146
146
assert result == expected
147
+
148
+
149
+ def test_invalid_arg (ds ):
150
+ with pytest .raises (KeyError , match = "Argument invalid_arg does not exist in Field: Character." ):
151
+ ds .query (ds .Query .hero .args (invalid_arg = 5 ).select (ds .Character .name ))
You can’t perform that action at this time.
0 commit comments