style: simply defer os.RemoveAll

spf13 · Sep 17, 2019 · c35f8b0 · c35f8b0
1 parent 7a0cf95
commit c35f8b0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 15 deletions.
diff --git a/cobra/cmd/add_test.go b/cobra/cmd/add_test.go
@@ -23,15 +23,10 @@ func TestGoldenAddCmd(t *testing.T) {
 			Viper:   true,
 		},
 	}
+	defer os.RemoveAll(command.AbsolutePath)
 
 	// init project first
 	command.Project.Create()
-	defer func() {
-		if _, err := os.Stat(command.AbsolutePath); err == nil {
-			os.RemoveAll(command.AbsolutePath)
-		}
-	}()
-
 	if err := command.Create(); err != nil {
 		t.Fatal(err)
 	}

diff --git a/cobra/cmd/init_test.go b/cobra/cmd/init_test.go
@@ -12,19 +12,17 @@ func TestGoldenInitCmd(t *testing.T) {
 	wd, _ := os.Getwd()
 	project := &Project{
 		AbsolutePath: fmt.Sprintf("%s/testproject", wd),
-		PkgName:      "github.com/spf13/testproject",
 		Legal:        getLicense(),
 		Copyright:    copyrightLine(),
-		Viper:        true,
-		AppName:      "testproject",
-	}
 
-	defer func() {
-		if _, err := os.Stat(project.AbsolutePath); err == nil {
-			os.RemoveAll(project.AbsolutePath)
-		}
-	}()
+		// required to init
+		AppName: "testproject",
+		PkgName: "github.com/spf13/testproject",
+		Viper:   true,
+	}
+	defer os.RemoveAll(project.AbsolutePath)
 
+	// init project first
 	err := project.Create()
 	if err != nil {
 		t.Fatal(err)