fixes case where embeded structs would cause no field to be found

99designs · Aug 24, 2018 · 7d6f8ed · 7d6f8ed
1 parent 40f904a
commit 7d6f8ed
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/codegen/util.go b/codegen/util.go
@@ -134,7 +134,7 @@ func findField(typ *types.Struct, name, structTag string) (*types.Var, error) {
 		if field.Anonymous() {
 			if named, ok := field.Type().(*types.Struct); ok {
 				f, err := findField(named, name, structTag)
-				if err != nil {
+				if err != nil && !strings.HasPrefix(err.Error(), "no field named") {
 					return nil, err
 				}
 				if f != nil && foundField == nil {
@@ -144,7 +144,7 @@ func findField(typ *types.Struct, name, structTag string) (*types.Var, error) {
 
 			if named, ok := field.Type().Underlying().(*types.Struct); ok {
 				f, err := findField(named, name, structTag)
-				if err != nil {
+				if err != nil && !strings.HasPrefix(err.Error(), "no field named") {
 					return nil, err
 				}
 				if f != nil && foundField == nil {

diff --git a/codegen/util_test.go b/codegen/util_test.go
@@ -38,6 +38,10 @@ type Amb struct {
 	Bar string ` + "`" + `gqlgen:"foo"` + "`" + `
 	Foo int    ` + "`" + `gqlgen:"foo"` + "`" + `
 }
+type Embed struct {
+	Std
+	Test string
+}
 `
 	scope, err := parseScope(input, "test")
 	require.NoError(t, err)
@@ -46,6 +50,7 @@ type Amb struct {
 	anon := scope.Lookup("Anon").Type().Underlying().(*types.Struct)
 	tags := scope.Lookup("Tags").Type().Underlying().(*types.Struct)
 	amb := scope.Lookup("Amb").Type().Underlying().(*types.Struct)
+	embed := scope.Lookup("Embed").Type().Underlying().(*types.Struct)
 
 	tests := []struct {
 		Name        string
@@ -61,6 +66,7 @@ type Amb struct {
 		{"Picks field with tag over field name when passed a tag", tags, "foo", "gqlgen", "Bar", false},
 		{"Errors when ambigious", amb, "foo", "gqlgen", "", true},
 		{"Finds a field that is in embedded struct", anon, "bar", "", "Bar", false},
+		{"Finds field that is not in embedded struct", embed, "test", "", "Test", false},
 	}
 
 	for _, tt := range tests {