Ensures we add our own package to the imports

ryanmoran · Sep 4, 2024 · b14b23b · b14b23b
1 parent dd7d6f7
commit b14b23b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/parsing/parse.go b/parsing/parse.go
@@ -51,7 +51,7 @@ func Parse(path, name string) (Fake, error) {
 		return Fake{}, fmt.Errorf("failed to parse scope file: %w", err)
 	}
 
-	var imports []Import
+	imports := []Import{{Path: pkg.Path()}}
 	for _, i := range file.Imports {
 		buffer := bytes.NewBuffer(nil)
 

diff --git a/parsing/parse_test.go b/parsing/parse_test.go
@@ -16,6 +16,7 @@ var _ = Describe("Parse", func() {
 		Expect(err).NotTo(HaveOccurred())
 		Expect(fake).To(Equal(parsing.Fake{
 			Imports: []parsing.Import{
+				{Name: "", Path: "io"},
 				{Name: "", Path: "errors"},
 				{Name: "", Path: "sync"},
 			},
@@ -52,6 +53,7 @@ var _ = Describe("Parse", func() {
 			Expect(err).NotTo(HaveOccurred())
 			Expect(fake).To(Equal(parsing.Fake{
 				Imports: []parsing.Import{
+					{Name: "", Path: "io"},
 					{Name: "", Path: "errors"},
 					{Name: "", Path: "sync"},
 				},