314 lines
7.9 KiB
Go
314 lines
7.9 KiB
Go
|
// Disabling building of yaml support in cases where golang is 1.0 or 1.1
|
||
|
|
// as the encoding library is not implemented or supported.
|
||
|
|
|
||
|
|
// +build go1.2
|
||
|
|
|
||
|
|
package altsrc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"gopkg.in/urfave/cli.v1"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestCommandYamlFileTest(t *testing.T) {
|
||
|
|
app := cli.NewApp()
|
||
|
|
set := flag.NewFlagSet("test", 0)
|
||
|
|
ioutil.WriteFile("current.yaml", []byte("test: 15"), 0666)
|
||
|
|
defer os.Remove("current.yaml")
|
||
|
|
test := []string{"test-cmd", "--load", "current.yaml"}
|
||
|
|
set.Parse(test)
|
||
|
|
|
||
|
|
c := cli.NewContext(app, set, nil)
|
||
|
|
|
||
|
|
command := &cli.Command{
|
||
|
|
Name: "test-cmd",
|
||
|
|
Aliases: []string{"tc"},
|
||
|
|
Usage: "this is for testing",
|
||
|
|
Description: "testing",
|
||
|
|
Action: func(c *cli.Context) error {
|
||
|
|
val := c.Int("test")
|
||
|
|
expect(t, val, 15)
|
||
|
|
return nil
|
||
|
|
},
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
NewIntFlag(cli.IntFlag{Name: "test"}),
|
||
|
|
cli.StringFlag{Name: "load"}},
|
||
|
|
}
|
||
|
|
command.Before = InitInputSourceWithContext(command.Flags, NewYamlSourceFromFlagFunc("load"))
|
||
|
|
err := command.Run(c)
|
||
|
|
|
||
|
|
expect(t, err, nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCommandYamlFileTestGlobalEnvVarWins(t *testing.T) {
|
||
|
|
app := cli.NewApp()
|
||
|
|
set := flag.NewFlagSet("test", 0)
|
||
|
|
ioutil.WriteFile("current.yaml", []byte("test: 15"), 0666)
|
||
|
|
defer os.Remove("current.yaml")
|
||
|
|
|
||
|
|
os.Setenv("THE_TEST", "10")
|
||
|
|
defer os.Setenv("THE_TEST", "")
|
||
|
|
test := []string{"test-cmd", "--load", "current.yaml"}
|
||
|
|
set.Parse(test)
|
||
|
|
|
||
|
|
c := cli.NewContext(app, set, nil)
|
||
|
|
|
||
|
|
command := &cli.Command{
|
||
|
|
Name: "test-cmd",
|
||
|
|
Aliases: []string{"tc"},
|
||
|
|
Usage: "this is for testing",
|
||
|
|
Description: "testing",
|
||
|
|
Action: func(c *cli.Context) error {
|
||
|
|
val := c.Int("test")
|
||
|
|
expect(t, val, 10)
|
||
|
|
return nil
|
||
|
|
},
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
NewIntFlag(cli.IntFlag{Name: "test", EnvVar: "THE_TEST"}),
|
||
|
|
cli.StringFlag{Name: "load"}},
|
||
|
|
}
|
||
|
|
command.Before = InitInputSourceWithContext(command.Flags, NewYamlSourceFromFlagFunc("load"))
|
||
|
|
|
||
|
|
err := command.Run(c)
|
||
|
|
|
||
|
|
expect(t, err, nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCommandYamlFileTestGlobalEnvVarWinsNested(t *testing.T) {
|
||
|
|
app := cli.NewApp()
|
||
|
|
set := flag.NewFlagSet("test", 0)
|
||
|
|
ioutil.WriteFile("current.yaml", []byte(`top:
|
||
|
|
test: 15`), 0666)
|
||
|
|
defer os.Remove("current.yaml")
|
||
|
|
|
||
|
|
os.Setenv("THE_TEST", "10")
|
||
|
|
defer os.Setenv("THE_TEST", "")
|
||
|
|
test := []string{"test-cmd", "--load", "current.yaml"}
|
||
|
|
set.Parse(test)
|
||
|
|
|
||
|
|
c := cli.NewContext(app, set, nil)
|
||
|
|
|
||
|
|
command := &cli.Command{
|
||
|
|
Name: "test-cmd",
|
||
|
|
Aliases: []string{"tc"},
|
||
|
|
Usage: "this is for testing",
|
||
|
|
Description: "testing",
|
||
|
|
Action: func(c *cli.Context) error {
|
||
|
|
val := c.Int("top.test")
|
||
|
|
expect(t, val, 10)
|
||
|
|
return nil
|
||
|
|
},
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
NewIntFlag(cli.IntFlag{Name: "top.test", EnvVar: "THE_TEST"}),
|
||
|
|
cli.StringFlag{Name: "load"}},
|
||
|
|
}
|
||
|
|
command.Before = InitInputSourceWithContext(command.Flags, NewYamlSourceFromFlagFunc("load"))
|
||
|
|
|
||
|
|
err := command.Run(c)
|
||
|
|
|
||
|
|
expect(t, err, nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCommandYamlFileTestSpecifiedFlagWins(t *testing.T) {
|
||
|
|
app := cli.NewApp()
|
||
|
|
set := flag.NewFlagSet("test", 0)
|
||
|
|
ioutil.WriteFile("current.yaml", []byte("test: 15"), 0666)
|
||
|
|
defer os.Remove("current.yaml")
|
||
|
|
|
||
|
|
test := []string{"test-cmd", "--load", "current.yaml", "--test", "7"}
|
||
|
|
set.Parse(test)
|
||
|
|
|
||
|
|
c := cli.NewContext(app, set, nil)
|
||
|
|
|
||
|
|
command := &cli.Command{
|
||
|
|
Name: "test-cmd",
|
||
|
|
Aliases: []string{"tc"},
|
||
|
|
Usage: "this is for testing",
|
||
|
|
Description: "testing",
|
||
|
|
Action: func(c *cli.Context) error {
|
||
|
|
val := c.Int("test")
|
||
|
|
expect(t, val, 7)
|
||
|
|
return nil
|
||
|
|
},
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
NewIntFlag(cli.IntFlag{Name: "test"}),
|
||
|
|
cli.StringFlag{Name: "load"}},
|
||
|
|
}
|
||
|
|
command.Before = InitInputSourceWithContext(command.Flags, NewYamlSourceFromFlagFunc("load"))
|
||
|
|
|
||
|
|
err := command.Run(c)
|
||
|
|
|
||
|
|
expect(t, err, nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCommandYamlFileTestSpecifiedFlagWinsNested(t *testing.T) {
|
||
|
|
app := cli.NewApp()
|
||
|
|
set := flag.NewFlagSet("test", 0)
|
||
|
|
ioutil.WriteFile("current.yaml", []byte(`top:
|
||
|
|
test: 15`), 0666)
|
||
|
|
defer os.Remove("current.yaml")
|
||
|
|
|
||
|
|
test := []string{"test-cmd", "--load", "current.yaml", "--top.test", "7"}
|
||
|
|
set.Parse(test)
|
||
|
|
|
||
|
|
c := cli.NewContext(app, set, nil)
|
||
|
|
|
||
|
|
command := &cli.Command{
|
||
|
|
Name: "test-cmd",
|
||
|
|
Aliases: []string{"tc"},
|
||
|
|
Usage: "this is for testing",
|
||
|
|
Description: "testing",
|
||
|
|
Action: func(c *cli.Context) error {
|
||
|
|
val := c.Int("top.test")
|
||
|
|
expect(t, val, 7)
|
||
|
|
return nil
|
||
|
|
},
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
NewIntFlag(cli.IntFlag{Name: "top.test"}),
|
||
|
|
cli.StringFlag{Name: "load"}},
|
||
|
|
}
|
||
|
|
command.Before = InitInputSourceWithContext(command.Flags, NewYamlSourceFromFlagFunc("load"))
|
||
|
|
|
||
|
|
err := command.Run(c)
|
||
|
|
|
||
|
|
expect(t, err, nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCommandYamlFileTestDefaultValueFileWins(t *testing.T) {
|
||
|
|
app := cli.NewApp()
|
||
|
|
set := flag.NewFlagSet("test", 0)
|
||
|
|
ioutil.WriteFile("current.yaml", []byte("test: 15"), 0666)
|
||
|
|
defer os.Remove("current.yaml")
|
||
|
|
|
||
|
|
test := []string{"test-cmd", "--load", "current.yaml"}
|
||
|
|
set.Parse(test)
|
||
|
|
|
||
|
|
c := cli.NewContext(app, set, nil)
|
||
|
|
|
||
|
|
command := &cli.Command{
|
||
|
|
Name: "test-cmd",
|
||
|
|
Aliases: []string{"tc"},
|
||
|
|
Usage: "this is for testing",
|
||
|
|
Description: "testing",
|
||
|
|
Action: func(c *cli.Context) error {
|
||
|
|
val := c.Int("test")
|
||
|
|
expect(t, val, 15)
|
||
|
|
return nil
|
||
|
|
},
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
NewIntFlag(cli.IntFlag{Name: "test", Value: 7}),
|
||
|
|
cli.StringFlag{Name: "load"}},
|
||
|
|
}
|
||
|
|
command.Before = InitInputSourceWithContext(command.Flags, NewYamlSourceFromFlagFunc("load"))
|
||
|
|
|
||
|
|
err := command.Run(c)
|
||
|
|
|
||
|
|
expect(t, err, nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCommandYamlFileTestDefaultValueFileWinsNested(t *testing.T) {
|
||
|
|
app := cli.NewApp()
|
||
|
|
set := flag.NewFlagSet("test", 0)
|
||
|
|
ioutil.WriteFile("current.yaml", []byte(`top:
|
||
|
|
test: 15`), 0666)
|
||
|
|
defer os.Remove("current.yaml")
|
||
|
|
|
||
|
|
test := []string{"test-cmd", "--load", "current.yaml"}
|
||
|
|
set.Parse(test)
|
||
|
|
|
||
|
|
c := cli.NewContext(app, set, nil)
|
||
|
|
|
||
|
|
command := &cli.Command{
|
||
|
|
Name: "test-cmd",
|
||
|
|
Aliases: []string{"tc"},
|
||
|
|
Usage: "this is for testing",
|
||
|
|
Description: "testing",
|
||
|
|
Action: func(c *cli.Context) error {
|
||
|
|
val := c.Int("top.test")
|
||
|
|
expect(t, val, 15)
|
||
|
|
return nil
|
||
|
|
},
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
NewIntFlag(cli.IntFlag{Name: "top.test", Value: 7}),
|
||
|
|
cli.StringFlag{Name: "load"}},
|
||
|
|
}
|
||
|
|
command.Before = InitInputSourceWithContext(command.Flags, NewYamlSourceFromFlagFunc("load"))
|
||
|
|
|
||
|
|
err := command.Run(c)
|
||
|
|
|
||
|
|
expect(t, err, nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCommandYamlFileFlagHasDefaultGlobalEnvYamlSetGlobalEnvWins(t *testing.T) {
|
||
|
|
app := cli.NewApp()
|
||
|
|
set := flag.NewFlagSet("test", 0)
|
||
|
|
ioutil.WriteFile("current.yaml", []byte("test: 15"), 0666)
|
||
|
|
defer os.Remove("current.yaml")
|
||
|
|
|
||
|
|
os.Setenv("THE_TEST", "11")
|
||
|
|
defer os.Setenv("THE_TEST", "")
|
||
|
|
|
||
|
|
test := []string{"test-cmd", "--load", "current.yaml"}
|
||
|
|
set.Parse(test)
|
||
|
|
|
||
|
|
c := cli.NewContext(app, set, nil)
|
||
|
|
|
||
|
|
command := &cli.Command{
|
||
|
|
Name: "test-cmd",
|
||
|
|
Aliases: []string{"tc"},
|
||
|
|
Usage: "this is for testing",
|
||
|
|
Description: "testing",
|
||
|
|
Action: func(c *cli.Context) error {
|
||
|
|
val := c.Int("test")
|
||
|
|
expect(t, val, 11)
|
||
|
|
return nil
|
||
|
|
},
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
NewIntFlag(cli.IntFlag{Name: "test", Value: 7, EnvVar: "THE_TEST"}),
|
||
|
|
cli.StringFlag{Name: "load"}},
|
||
|
|
}
|
||
|
|
command.Before = InitInputSourceWithContext(command.Flags, NewYamlSourceFromFlagFunc("load"))
|
||
|
|
err := command.Run(c)
|
||
|
|
|
||
|
|
expect(t, err, nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCommandYamlFileFlagHasDefaultGlobalEnvYamlSetGlobalEnvWinsNested(t *testing.T) {
|
||
|
|
app := cli.NewApp()
|
||
|
|
set := flag.NewFlagSet("test", 0)
|
||
|
|
ioutil.WriteFile("current.yaml", []byte(`top:
|
||
|
|
test: 15`), 0666)
|
||
|
|
defer os.Remove("current.yaml")
|
||
|
|
|
||
|
|
os.Setenv("THE_TEST", "11")
|
||
|
|
defer os.Setenv("THE_TEST", "")
|
||
|
|
|
||
|
|
test := []string{"test-cmd", "--load", "current.yaml"}
|
||
|
|
set.Parse(test)
|
||
|
|
|
||
|
|
c := cli.NewContext(app, set, nil)
|
||
|
|
|
||
|
|
command := &cli.Command{
|
||
|
|
Name: "test-cmd",
|
||
|
|
Aliases: []string{"tc"},
|
||
|
|
Usage: "this is for testing",
|
||
|
|
Description: "testing",
|
||
|
|
Action: func(c *cli.Context) error {
|
||
|
|
val := c.Int("top.test")
|
||
|
|
expect(t, val, 11)
|
||
|
|
return nil
|
||
|
|
},
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
NewIntFlag(cli.IntFlag{Name: "top.test", Value: 7, EnvVar: "THE_TEST"}),
|
||
|
|
cli.StringFlag{Name: "load"}},
|
||
|
|
}
|
||
|
|
command.Before = InitInputSourceWithContext(command.Flags, NewYamlSourceFromFlagFunc("load"))
|
||
|
|
err := command.Run(c)
|
||
|
|
|
||
|
|
expect(t, err, nil)
|
||
|
|
}
|