feat(main): adding main test to check version

main_test.go

@@ -0,0 +1,22 @@
+package main
+
+import (
+	"os"
+	"sbam/pkg/cmd"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestMain(t *testing.T) {
+	err := cmd.SetVersionInfo("1.0", "abc123", "2022-01-01")
+	assert.NoError(t, err)
+
+	old := os.Args
+	defer func() { os.Args = old }()
+
+	os.Args = []string{"cmd", "--version"}
+
+	err = cmd.Execute()
+	assert.NoError(t, err)
+}

pkg/cmd/root.go

@@ -16,7 +16,7 @@ var rootCmd = &cobra.Command{
 	Initiate parameters from command line, env variables or config.yaml file.`,
 }
 
-func Execute() {
+func Execute() error {
 	err := rootCmd.Execute()
 	if err != nil {
 		os.Exit(1)
@@ -28,6 +28,7 @@ func Execute() {
 			os.Exit(0)
 		}
 	}
+	return nil
 }
 
 func init() {
@@ -43,6 +44,7 @@ func init() {
 	}
 }
 
-func SetVersionInfo(version, commit, date string) {
+func SetVersionInfo(version, commit, date string) error {
 	rootCmd.Version = fmt.Sprintf("%s (Built on %s from Git SHA %s)", version, date, commit)
+	return nil
 }