Remove usage of testing's Cleanup

csueiras · Feb 18, 2021 · 474eaf8 · 474eaf8
1 parent 0cfb2c0
commit 474eaf8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/internal/generator/generator_test.go b/internal/generator/generator_test.go
@@ -517,7 +517,7 @@ func loadInterface(t *testing.T, filesCode map[string]input) []*generator.FileCo
 		},
 	}
 	exported := packagestest.Export(t, packagestest.GOPATH, mods)
-	t.Cleanup(exported.Cleanup)
+	defer exported.Cleanup()
 
 	l := loader.NewLoader(func(cfg *packages.Config, patterns ...string) ([]*packages.Package, error) {
 		exported.Config.Mode = cfg.Mode

diff --git a/internal/loader/loader_test.go b/internal/loader/loader_test.go
@@ -22,7 +22,7 @@ type Service interface {
 }
 `,
 		}}})
-	t.Cleanup(exported.Cleanup)
+	defer exported.Cleanup()
 
 	l := loader.NewLoader(func(cfg *packages.Config, patterns ...string) ([]*packages.Package, error) {
 		exported.Config.Mode = cfg.Mode
@@ -60,7 +60,7 @@ type NotAnInterface struct {
 }
 `,
 		}}})
-	t.Cleanup(exported.Cleanup)
+	defer exported.Cleanup()
 
 	t.Run("RegEx", func(t *testing.T) {
 		l := loader.NewLoader(func(cfg *packages.Config, patterns ...string) ([]*packages.Package, error) {