clean up testproject files after test executes

spf13 · Jun 7, 2019 · 984374f · 984374f
1 parent 11aa612
commit 984374f
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/cobra/cmd/add_test.go b/cobra/cmd/add_test.go
@@ -16,9 +16,22 @@ func TestGoldenAddCmd(t *testing.T) {
 			AbsolutePath: fmt.Sprintf("%s/testproject", wd),
 			Legal:        getLicense(),
 			Copyright:    copyrightLine(),
+
+			// required to init
+			AppName: "testproject",
+			PkgName: "github.com/spf13/testproject",
+			Viper:   true,
 		},
 	}
 
+	// 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
@@ -8,6 +8,7 @@ import (
 )
 
 func TestGoldenInitCmd(t *testing.T) {
+
 	wd, _ := os.Getwd()
 	project := &Project{
 		AbsolutePath: fmt.Sprintf("%s/testproject", wd),
@@ -23,6 +24,12 @@ func TestGoldenInitCmd(t *testing.T) {
 		t.Fatal(err)
 	}
 
+	defer func() {
+		if _, err := os.Stat(project.AbsolutePath); err == nil {
+			os.RemoveAll(project.AbsolutePath)
+		}
+	}()
+
 	expectedFiles := []string{"LICENSE", "main.go", "cmd/root.go"}
 	for _, f := range expectedFiles {
 		generatedFile := fmt.Sprintf("%s/%s", project.AbsolutePath, f)