Merge branch 'chrusty-imports_for_inline_structs'

fixtures/inline_struct_params.go

@@ -0,0 +1,18 @@
+package fixtures
+
+import (
+	"context"
+	"net/http"
+	"time"
+)
+
+//counterfeiter:generate . InlineStructParams
+type InlineStructParams interface {
+	DoSomething(ctx context.Context, body struct {
+		SomeString        string
+		SomeStringPointer *string
+		SomeTime          time.Time
+		SomeTimePointer   *time.Time
+		HTTPRequest       http.Request
+	}) error
+}

generator/loader.go

@@ -131,6 +131,10 @@ func (f *Fake) addImportsFor(typ types.Type) {
 		return
 	case *types.Signature:
 		f.addTypesForMethod(t)
+	case *types.Struct:
+		for i := 0; i < t.NumFields(); i++ {
+			f.addImportsFor(t.Field(i).Type())
+		}
 	default:
 		log.Printf("!!! WARNING: Missing case for type %s\n", reflect.TypeOf(typ).String())
 	}

integration/roundtrip_test.go

@@ -192,6 +192,7 @@ func runTests(useGopath bool, t *testing.T, when spec.G, it spec.S) {
 		t("ImportsGoHyphenPackage", "imports_go_hyphen_package.go", "", filepath.Join("go-hyphenpackage", "fixture.go"))
 		t("FirstInterface", "multiple_interfaces.go", "")
 		t("SecondInterface", "multiple_interfaces.go", "")
+		t("InlineStructParams", "inline_struct_params.go", "")
 		t("RequestFactory", "request_factory.go", "")
 		t("ReusesArgTypes", "reuses_arg_types.go", "")
 		t("SomethingWithForeignInterface", "something_remote.go", "", filepath.Join("aliased_package", "in_aliased_package.go"))