Merge pull request #332 from robdaemon/add_inputobject_field

Adds an AddFieldConfig method to InputObjects
graphql-go · Jul 19, 2018 · 086b3bf · 086b3bf
2 parents 83f0349 + a05d5bb
commit 086b3bf
Showing 2 changed files with 37 additions and 0 deletions.
diff --git a/definition.go b/definition.go
@@ -1149,6 +1149,18 @@ func (gt *InputObject) defineFieldMap() InputObjectFieldMap {
 	return resultFieldMap
 }
 
+func (gt *InputObject) AddFieldConfig(fieldName string, fieldConfig *InputObjectFieldConfig) {
+	if fieldName == "" || fieldConfig == nil {
+		return
+	}
+	fieldMap, ok := gt.typeConfig.Fields.(InputObjectConfigFieldMap)
+	if gt.err = invariant(ok, "Cannot add field to a thunk"); gt.err != nil {
+		return
+	}
+	fieldMap[fieldName] = fieldConfig
+	gt.fields = gt.defineFieldMap()
+}
+
 func (gt *InputObject) Fields() InputObjectFieldMap {
 	if !gt.init {
 		gt.fields = gt.defineFieldMap()

diff --git a/definition_test.go b/definition_test.go
@@ -641,3 +641,28 @@ func TestTypeSystem_DefinitionExampe_AllowsCyclicFieldTypes(t *testing.T) {
 	}
 
 }
+
+func TestTypeSystem_DefinitionExample_CanAddInputObjectField(t *testing.T) {
+	io := graphql.NewInputObject(graphql.InputObjectConfig{
+		Name: "inputObject",
+		Fields: graphql.InputObjectConfigFieldMap{
+			"value": &graphql.InputObjectFieldConfig{
+				Type: graphql.String,
+			},
+		},
+	})
+	io.AddFieldConfig("newValue", &graphql.InputObjectFieldConfig{
+		Type: graphql.Int,
+	})
+	fieldMap := io.Fields()
+
+	if len(fieldMap) < 2 {
+		t.Fatalf("Unexpected result, inputObject should have two fields, has %d", len(fieldMap))
+	}
+	if _, ok := fieldMap["value"]; !ok {
+		t.Fatal("Unexpected result, inputObject should have a field named 'value'")
+	}
+	if _, ok := fieldMap["newValue"]; !ok {
+		t.Fatal("Unexpected result, inputObject should have a field named 'newValue'")
+	}
+}