add tests

ccbrown · Mar 10, 2024 · 9b98469 · 9b98469
1 parent 0693e61
commit 9b98469
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/graphql/validator/validate_fragments_test.go b/graphql/validator/validate_fragments_test.go
@@ -66,4 +66,9 @@ func TestValidateFragmentSpreads(t *testing.T) {
 		assert.Empty(t, validateSource(t, `{union{... on UnionMember{... on Union{... on UnionObjectA{a}}}}}`))
 		assert.Len(t, validateSource(t, `{union{... on UnionMember{... on Node{id}}}}`), 1)
 	})
+
+	t.Run("Features", func(t *testing.T) {
+		assert.Empty(t, validateSource(t, `{experimentalObject{...a}} fragment a on ExperimentalObject { foo }`, "experimentalobject"))
+		assert.Len(t, validateSource(t, `{experimentalObject{...a}} fragment a on ExperimentalObject { foo }`), 2)
+	})
 }
diff --git a/graphql/validator/validator_test.go b/graphql/validator/validator_test.go
@@ -47,6 +47,16 @@ var objectType = &schema.ObjectType{
 	Name: "Object",
 }
 
+var experimentalObjectType = &schema.ObjectType{
+	Name:             "ExperimentalObject",
+	RequiredFeatures: schema.NewFeatureSet("experimentalobject"),
+	Fields: map[string]*schema.FieldDefinition{
+		"foo": {
+			Type: schema.BooleanType,
+		},
+	},
+}
+
 var complexInputType = &schema.InputObjectType{
 	Name: "ComplexInput",
 	Fields: map[string]*schema.InputValueDefinition{
@@ -92,6 +102,10 @@ func init() {
 			Type: schema.BooleanType,
 			Cost: schema.FieldResolverCost(0),
 		},
+		"experimentalObject": {
+			Type:             experimentalObjectType,
+			RequiredFeatures: schema.NewFeatureSet("experimentalobject"),
+		},
 		"booleanArgField": {
 			Type: schema.BooleanType,
 			Arguments: map[string]*schema.InputValueDefinition{