fix: add some token tests

Author: dnitsch

main.go

@@ -13,11 +13,8 @@ import (
 	"github.com/hashicorp/go-plugin"
 )
 
-type implIface interface {
-	Value(token string, metadata []byte) (string, error)
-}
 type TokenStorePlugin struct {
-	impl implIface
+	impl tokenstore.TokenStore
 }
 
 func (ts TokenStorePlugin) Value(key string, metadata []byte) (string, error) {
@@ -29,20 +26,28 @@ var (
 	Revision string = "1111aaaa"
 )
 
-func main() {
-
-	versionFlag := flag.Bool("version", false, "plugin version")
-	flag.Parse()
+func ShowFlag(osArgs []string) bool {
+	fs := flag.NewFlagSet("plugin", flag.ContinueOnError)
+	vf := fs.Bool("version", false, "plugin version")
+	if err := fs.Parse(osArgs); err != nil {
+		return false
+	}
 
-	if *versionFlag {
+	if *vf {
 		fmt.Printf("Version: %s-%s\n", Version, Revision)
-		os.Exit(0)
+		return true
 	}
+	return false
+}
+
+func PluginSetup() (*impl.ParamStore, error) {
 
 	// log set up
 	log := hclog.New(hclog.DefaultOptions)
+
 	log.SetLevel(hclog.LevelFromString("error"))
 
+	os.Environ()
 	if val, ok := os.LookupEnv(config.CONFIGMANAGER_LOG); ok && len(val) > 0 {
 		if logLevel := hclog.LevelFromString(val); logLevel != hclog.NoLevel {
 			log.SetLevel(logLevel)
@@ -52,7 +57,22 @@ func main() {
 	// initialize the implementation
 	i, err := impl.NewParamStore(context.Background(), log)
 	if err != nil {
-		log.Error("error", err)
+		log.Error("implementation init error", err, "impl", "awsparamstr")
+		return nil, err
+	}
+
+	return i, nil
+}
+
+// main func should only be used for process terminantion
+// Inits and serves the plugin
+func main() {
+	if ShowFlag(os.Args[1:]) {
+		os.Exit(0)
+	}
+
+	i, err := PluginSetup()
+	if err != nil {
 		os.Exit(1)
 	}
 
@@ -64,6 +84,7 @@ func main() {
 		Plugins: map[string]plugin.Plugin{
 			"configmanager_token_store": &tokenstore.GRPCPlugin{Impl: ts},
 		},
+		//
 		VersionedPlugins: map[int]plugin.PluginSet{
 			1: {
 				"configmanager_token_store": &tokenstore.GRPCPlugin{Impl: ts},

main_test.go

@@ -0,0 +1,31 @@
+package main_test
+
+import (
+	"testing"
+
+	main "github.com/DevLabFoundry/configmanager-plugin-awsparamstr"
+)
+
+func Test_PluginInit(t *testing.T) {
+	t.Run("init plugin", func(t *testing.T) {
+		_, err := main.PluginSetup()
+		if err != nil {
+			t.Fatal(err)
+		}
+	})
+}
+
+func Test_ShowVersion(t *testing.T) {
+	t.Run("show Version", func(t *testing.T) {
+		// os.Args = append(os.Args[:1], "--version")
+		if !main.ShowFlag([]string{"--version"}) {
+			t.Error("should show flag")
+		}
+	})
+
+	t.Run("do not ShowVersion", func(t *testing.T) {
+		if main.ShowFlag([]string{"foo"}) {
+			t.Error("should show flag")
+		}
+	})
+}