Add file checking for some test cases of app support

Author: desaroger

command/clean_test.go

@@ -6,9 +6,11 @@ package command
 
 import (
 	"os"
+	"path/filepath"
 	"strings"
 	"testing"
 
+	"github.com/git-time-metric/gtm/project"
 	"github.com/git-time-metric/gtm/util"
 	"github.com/mitchellh/cli"
 )
@@ -57,6 +59,10 @@ func TestAppOnly(t *testing.T) {
 	repo.Seed()
 	os.Chdir(repo.Workdir())
 
+	repo.SaveFile("browser.app", project.GTMDir, "")
+	repo.SaveFile("1458496803.event", project.GTMDir, filepath.Join("event", "event.go"))
+	repo.SaveFile("1458497804.event", project.GTMDir, filepath.Join(project.GTMDir, "browser.app"))
+
 	(InitCmd{UI: new(cli.MockUi)}).Run([]string{})
 
 	ui := new(cli.MockUi)
@@ -68,6 +74,14 @@ func TestAppOnly(t *testing.T) {
 	if rc != 0 {
 		t.Errorf("gtm clean(%+v), want 0 got %d, %s", args, rc, ui.ErrorWriter.String())
 	}
+
+	if !repo.FileExists("1458496803.event", project.GTMDir) {
+		t.Errorf("gtm clean(%+v), want non-app event to not be deleted, but was deleted", args)
+	}
+
+	if repo.FileExists("1458497804.event", project.GTMDir) {
+		t.Errorf("gtm clean(%+v), want app event to be deleted, but was found", args)
+	}
 }
 
 func TestCleanInvalidOption(t *testing.T) {

util/test.go

@@ -139,6 +139,12 @@ func (t TestRepo) SaveFile(filename, subdir, content string) {
 	CheckFatal(t.test, err)
 }
 
+// FileExists Checks if a file exists in the repo folder
+func (t TestRepo) FileExists(filename, subdir string) bool {
+	_, err := os.Stat(filepath.Join(subdir, filename))
+	return !os.IsNotExist(err)
+}
+
 // Clone creates a clone of this repo
 func (t TestRepo) Clone() TestRepo {
 	path, err := ioutil.TempDir("", "gtm")