package cli import ( "encoding/json" "os" "testing" "github.com/micro/cli/v2" "github.com/micro/go-micro/config/cmd" "github.com/micro/go-micro/config/source" ) 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"}, } // 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) }