diff --git a/internal/generator/fixture/template.go b/internal/generator/fixture/template.go index 836fde5..dc3427b 100644 --- a/internal/generator/fixture/template.go +++ b/internal/generator/fixture/template.go @@ -35,11 +35,11 @@ func setUp(t *testing.T) *fixture { f := &fixture{ ctrl: ctrl_mock.NewController(t), ctx: context.Background(), + } {{ range .Fields }} - {{ .FieldName }}: {{ mockAlias .PkgPath }}.New{{ typeShort .TypeName }}(ctrl_mock.NewController(t)), + f.{{ .FieldName }} = {{ mockAlias .PkgPath }}.New{{ typeShort .TypeName }}(f.ctrl) {{- end }} - } f.{{ .ImplAlias }} = &{{ .StructName }}{ {{- range .Fields }} diff --git a/internal/generator/fixture/testdata/gen.golden b/internal/generator/fixture/testdata/gen.golden index 117e095..24021eb 100644 --- a/internal/generator/fixture/testdata/gen.golden +++ b/internal/generator/fixture/testdata/gen.golden @@ -23,11 +23,11 @@ func setUp(t *testing.T) *fixture { f := &fixture{ ctrl: ctrl_mock.NewController(t), ctx: context.Background(), - - myType1: mypackage1_mock.NewMockMyType1(ctrl_mock.NewController(t)), - myType2: mypackage2_mock.NewMockMyType2(ctrl_mock.NewController(t)), } + f.myType1 = mypackage1_mock.NewMockMyType1(f.ctrl) + f.myType2 = mypackage2_mock.NewMockMyType2(f.ctrl) + f.toGen = &ToGen{ myType1: f.myType1, myType2: f.myType2, diff --git a/testdata/without_external/mypackage2/mock/mock_gen.go b/testdata/without_external/mypackage2/mock/mock_gen.go index d469df6..357e9c4 100644 --- a/testdata/without_external/mypackage2/mock/mock_gen.go +++ b/testdata/without_external/mypackage2/mock/mock_gen.go @@ -19,6 +19,7 @@ import ( type MockMyType2 struct { ctrl *gomock.Controller recorder *MockMyType2MockRecorder + isgomock struct{} } // MockMyType2MockRecorder is the mock recorder for MockMyType2.