Vasiliy Tolstov
67acd9288b
* config/source/cli: fix tests * skip mdns test in travis Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
116 lines
2.1 KiB
Go
116 lines
2.1 KiB
Go
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 inside go test
|
|
&cli.StringFlag{
|
|
Name: "test.timeout",
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "test.v",
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "test.run",
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "test.testlogfile",
|
|
},
|
|
&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)
|
|
}
|