package cli import ( "encoding/json" "os" "testing" "github.com/micro/cli/v2" "github.com/micro/go-micro/v2" "github.com/micro/go-micro/v2/config" "github.com/micro/go-micro/v2/config/cmd" "github.com/micro/go-micro/v2/config/source" ) func TestCliSourceDefault(t *testing.T) { const expVal string = "flagvalue" service := micro.NewService( micro.Flags( // to be able to run insude go test &cli.StringFlag{ Name: "test.timeout", }, &cli.BoolFlag{ Name: "test.v", }, &cli.StringFlag{ Name: "test.run", }, &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 fval := config.Get("flag").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) }