test: adding test in the root package in order to test the run command (

#733)

* adding test in the root package in order to test the run command

Signed-off-by: Fokion Sotiropoulos <[email protected]>

cmd/venom/root/root.go

@@ -18,7 +18,7 @@ func New() *cobra.Command {
 	return rootCmd
 }
 
-//AddCommands adds child commands to the root command rootCmd.
+// AddCommands adds child commands to the root command rootCmd.
 func addCommands(cmd *cobra.Command) {
 	cmd.AddCommand(run.Cmd)
 	cmd.AddCommand(version.Cmd)

cmd/venom/root/root_test.go

@@ -0,0 +1,35 @@
+package root
+
+import (
+	"os/exec"
+	"path/filepath"
+	"strings"
+	"testing"
+
+	"github.com/ovh/venom"
+	"github.com/stretchr/testify/assert"
+)
+
+// getTopLevelFolder returns the top level folder of the project
+func getTopLevelFolder() string {
+	out, err := exec.Command("go", "list", "-m", "-f", "{{.Dir}}").Output()
+	if err != nil {
+		panic(err)
+	}
+	return strings.TrimSpace(string(out))
+}
+
+// TestRunCmd tests the run command
+func TestRunCmd(t *testing.T) {
+	var validArgs []string
+
+	validArgs = append(validArgs, "run", filepath.Join(getTopLevelFolder(), "tests", "assertions"))
+	rootCmd := New()
+	rootCmd.SetArgs(validArgs)
+	venom.IsTest = "test"
+	assert.Equal(t, 3, len(rootCmd.Commands()))
+	err := rootCmd.Execute()
+	assert.NoError(t, err)
+	rootCmd.Execute()
+
+}

cmd/venom/run/cmd.go

@@ -244,6 +244,7 @@ func initFromEnv(environ []string) ([]string, error) {
 		v := strings.Split(os.Getenv("VENOM_VAR"), " ")
 		variables = v
 	}
+
 	if os.Getenv("VENOM_VAR_FROM_FILE") != "" {
 		varFiles = strings.Split(os.Getenv("VENOM_VAR_FROM_FILE"), " ")
 	}

venom.go

@@ -27,6 +27,7 @@ var (
 	IsTest  = ""
 )
 
+// OSExit is a wrapper for os.Exit
 func OSExit(exitCode int) {
 	if IsTest != "" {
 		bincover.ExitCode = exitCode