Adds test around VersionInfo

cmd/root.go

@@ -29,6 +29,15 @@ type VersionInfo struct {
 
 var vInfo VersionInfo
 
+func (v VersionInfo) String() string {
+	return fmt.Sprintf(
+		"ecsrun version info\nVersion: %s\nCommit: %s\nDate Built: %s\nBuilt By: %s\n",
+		v.Version,
+		v.Commit,
+		v.Date,
+		v.BuiltBy)
+}
+
 var cfgFile string
 
 var log = logrus.New()
@@ -138,12 +147,7 @@ func initVerbose() {
 
 func initVersion() {
 	if viper.GetBool("version") {
-		fmt.Printf(
-			"ecsrun version info\nVersion: %s\nCommit: %s\nDate Built: %s\nBuilt By: %s\n",
-			vInfo.Version,
-			vInfo.Commit,
-			vInfo.Date,
-			vInfo.BuiltBy)
+		fmt.Print(vInfo.String())
 		os.Exit(0)
 	}
 }

cmd/root_test.go

@@ -58,7 +58,7 @@ func TestExecute(t *testing.T) {
 	os.Setenv("ECSRUN_TASK", "task")
 	os.Setenv("ECSRUN_VERBOSE", "true")
 
-	Execute(newEcsClientFake)
+	Execute(newEcsClientFake, VersionInfo{})
 
 	var sesh = viper.Get("session")
 	cred, _ := sesh.(*session.Session).Config.Credentials.Get()
@@ -67,6 +67,27 @@ func TestExecute(t *testing.T) {
 	teardown()
 }
 
+func TestVersion(t *testing.T) {
+	setup()
+	assert := assert.New(t)
+
+	vInfo := VersionInfo{
+		Version: "0.1.0",
+		Commit:  "238958943",
+		Date:    "06/19/20",
+		BuiltBy: "MDG",
+	}
+
+	result := vInfo.String()
+
+	assert.Contains(result, "0.1.0")
+	assert.Contains(result, "238958943")
+	assert.Contains(result, "06/19/20")
+	assert.Contains(result, "MDG")
+
+	teardown()
+}
+
 func TestInitAws(t *testing.T) {
 	assert := assert.New(t)
 	setup()