|
- // 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)
- }
|