chore: import upstream snapshot with attribution

config/source/cli/cli_test.go

@@ -0,0 +1,121 @@
+package cli
+
+import (
+	"encoding/json"
+	"os"
+	"testing"
+
+	"github.com/urfave/cli/v2"
+	"go-micro.dev/v6"
+	"go-micro.dev/v6/cmd"
+	"go-micro.dev/v6/config"
+	"go-micro.dev/v6/config/source"
+)
+
+func TestCliSourceDefault(t *testing.T) {
+	const expVal string = "flagvalue"
+
+	service := micro.NewService("test", micro.Flags(
+		// to be able to run inside go test
+		&cli.StringFlag{
+			Name: "test.timeout",
+		},
+		&cli.StringFlag{
+			Name: "test.bench",
+		},
+		&cli.BoolFlag{
+			Name: "test.v",
+		},
+		&cli.StringFlag{
+			Name: "test.run",
+		},
+		&cli.StringFlag{
+			Name: "test.testlogfile",
+		},
+		&cli.StringFlag{
+			Name: "test.paniconexit0",
+		},
+		&cli.StringFlag{
+			Name:    "flag",
+			Usage:   "It changes something",
+			EnvVars: []string{"flag"},
+			Value:   expVal,
+		},
+	),
+	)
+	var cliSrc source.Source
+	service.Init(
+		// Loads CLI configuration
+		micro.Action(func(c *cli.Context) error {
+			cliSrc = NewSource(
+				Context(c),
+			)
+			return nil
+		}),
+	)
+
+	config.Load(cliSrc)
+	if val, err := config.Get("flag"); err != nil {
+		t.Fatal(err)
+	} else if fval := val.String("default"); fval != expVal {
+		t.Fatalf("default flag value not loaded %v != %v", fval, expVal)
+	}
+}
+
+func test(t *testing.T, withContext bool) {
+	var src source.Source
+
+	// setup app
+	app := cmd.App()
+	app.Name = "testapp"
+	app.Flags = []cli.Flag{
+		&cli.StringFlag{
+			Name:    "db-host",
+			EnvVars: []string{"db-host"},
+			Value:   "myval",
+		},
+	}
+
+	// with context
+	if withContext {
+		// set action
+		app.Action = func(c *cli.Context) error {
+			src = WithContext(c)
+			return nil
+		}
+
+		// run app
+		app.Run([]string{"run", "-db-host", "localhost"})
+		// no context
+	} else {
+		// set args
+		os.Args = []string{"run", "-db-host", "localhost"}
+		src = NewSource()
+	}
+
+	// test config
+	c, err := src.Read()
+	if err != nil {
+		t.Error(err)
+	}
+
+	var actual map[string]interface{}
+	if err := json.Unmarshal(c.Data, &actual); err != nil {
+		t.Error(err)
+	}
+
+	actualDB := actual["db"].(map[string]interface{})
+	if actualDB["host"] != "localhost" {
+		t.Errorf("expected localhost, got %v", actualDB["name"])
+	}
+}
+
+func TestCliSource(t *testing.T) {
+	// without context
+	test(t, false)
+}
+
+func TestCliSourceWithContext(t *testing.T) {
+	// with context
+	test(t, true)
+}