micro/config/source/cli/cli_test.go

67 lines
1.1 KiB
Go
Raw Normal View History

2019-05-30 23:11:13 +01:00
package cli
import (
"encoding/json"
"os"
"testing"
2020-01-29 22:40:43 +00:00
"github.com/micro/cli/v2"
"github.com/micro/go-micro/v2/config/cmd"
"github.com/micro/go-micro/v2/config/source"
2019-05-30 23:11:13 +01:00
)
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"},
2019-05-30 23:11:13 +01:00
}
// with context
if withContext {
// set action
app.Action = func(c *cli.Context) error {
2019-05-30 23:11:13 +01:00
src = WithContext(c)
return nil
2019-05-30 23:11:13 +01:00
}
// 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)
}