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) {