fix: missing import path when receiver and methods are defined in dif…

gotests_test.go

@@ -869,6 +869,12 @@ func TestGenerateTests(t *testing.T) {
 				template: "testify",
 			},
 			want: mustReadAndFormatGoFile(t, "testdata/named/named_on_subtests_on_parallel_on_template_testify.go"),
+		}, {
+			name: "receiver and methods are defined in different file",
+			args: args{
+				srcPath: "testdata/separate_definition/method.go",
+			},
+			want: mustReadAndFormatGoFile(t, "testdata/goldens/receiver_and_methods_in_different_file.go"),
 		},
 	}
 	tmp, err := ioutil.TempDir("", "gotests_test")

internal/goparser/goparser.go

@@ -48,15 +48,19 @@ func (p *Parser) Parse(srcPath string, files []models.Path) (*Result, error) {
 	if err != nil {
 		return nil, err
 	}
-	return &Result{
+	result := &Result{
 		Header: &models.Header{
 			Comments: parsePkgComment(f, f.Package),
 			Package:  f.Name.String(),
 			Imports:  parseImports(f.Imports),
 			Code:     goCode(b, f),
 		},
 		Funcs: p.parseFunctions(fset, f, fs),
-	}, nil
+	}
+	for _, v := range fs {
+		result.Header.Imports = append(result.Header.Imports, parseImports(v.Imports)...)
+	}
+	return result, nil
 }
 
 func (p *Parser) readFile(srcPath string) ([]byte, error) {

testdata/goldens/receiver_and_methods_in_different_file.go

@@ -0,0 +1,34 @@
+package separate_definition
+
+import (
+	"testing"
+
+	"github.com/cweill/gotests/testdata/separate_definition/field"
+)
+
+func TestFoo_Foo(t *testing.T) {
+	type fields struct {
+		Bar       string
+		TestFiled field.Test
+	}
+	type args struct {
+		s string
+	}
+	tests := []struct {
+		name    string
+		fields  fields
+		args    args
+		wantErr bool
+	}{
+		// TODO: Add test cases.
+	}
+	for _, tt := range tests {
+		f := &Foo{
+			Bar:       tt.fields.Bar,
+			TestFiled: tt.fields.TestFiled,
+		}
+		if err := f.Foo(tt.args.s); (err != nil) != tt.wantErr {
+			t.Errorf("%q. Foo.Foo() error = %v, wantErr %v", tt.name, err, tt.wantErr)
+		}
+	}
+}

testdata/separate_definition/field/field.go

@@ -0,0 +1,4 @@
+package field
+
+type Test struct {
+}

testdata/separate_definition/method.go

@@ -0,0 +1,5 @@
+package separate_definition
+
+func (*Foo) Foo(s string) error {
+	return nil
+}

testdata/separate_definition/receiver.go

@@ -0,0 +1,8 @@
+package separate_definition
+
+import "github.com/cweill/gotests/testdata/separate_definition/field"
+
+type Foo struct {
+	Bar       string
+	TestFiled field.Test
+}