fix missing GraphQLAppliedDirectiveArgument: type child was not considered

andimarek · andimarek · commit 431affc8aae4 · 2022-08-24T15:59:11.000+10:00
diff --git a/src/main/java/graphql/schema/GraphQLAppliedDirectiveArgument.java b/src/main/java/graphql/schema/GraphQLAppliedDirectiveArgument.java
@@ -11,6 +11,7 @@
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
+import java.util.ArrayList;
 import java.util.List;
 import java.util.function.Consumer;
 
@@ -34,6 +35,8 @@ public class GraphQLAppliedDirectiveArgument implements GraphQLNamedSchemaElemen
     private final Argument definition;
 
 
+    public static final String CHILD_TYPE = "type";
+
     private GraphQLAppliedDirectiveArgument(String name,
                                             InputValueWithState value,
                                             GraphQLInputType type,
@@ -104,19 +107,23 @@ public Argument getDefinition() {
 
     @Override
     public List<GraphQLSchemaElement> getChildren() {
-        return ImmutableKit.emptyList();
+        List<GraphQLSchemaElement> children = new ArrayList<>();
+        children.add(getType());
+        return children;
     }
 
-
     @Override
     public SchemaElementChildrenContainer getChildrenWithTypeReferences() {
         return SchemaElementChildrenContainer.newSchemaElementChildrenContainer()
+                .child(CHILD_TYPE, originalType)
                 .build();
     }
 
     @Override
     public GraphQLAppliedDirectiveArgument withNewChildren(SchemaElementChildrenContainer newChildren) {
-        return this;
+            return transform(builder ->
+                    builder.type(newChildren.getChildOrNull(CHILD_TYPE))
+            );
     }
 
     @Override
