diff options
author | Harrison Healey <harrisonmhealey@gmail.com> | 2018-06-21 10:30:20 -0400 |
---|---|---|
committer | Jesús Espino <jespinog@gmail.com> | 2018-06-21 16:30:20 +0200 |
commit | 07c785e294e70494ae6016d59749a71af3f74920 (patch) | |
tree | 9b5e8deb15a182aa6a890261bb77b2e6faebbaea /vendor/github.com/mattermost/viper/flags_test.go | |
parent | 1f65f0e3d6d142b859f8dda52da99cb3d4a01c9c (diff) | |
download | chat-07c785e294e70494ae6016d59749a71af3f74920.tar.gz chat-07c785e294e70494ae6016d59749a71af3f74920.tar.bz2 chat-07c785e294e70494ae6016d59749a71af3f74920.zip |
MM-10730 Added support for empty environment variables to viper (#8973)
Diffstat (limited to 'vendor/github.com/mattermost/viper/flags_test.go')
-rw-r--r-- | vendor/github.com/mattermost/viper/flags_test.go | 65 |
1 files changed, 65 insertions, 0 deletions
diff --git a/vendor/github.com/mattermost/viper/flags_test.go b/vendor/github.com/mattermost/viper/flags_test.go new file mode 100644 index 000000000..0b976b605 --- /dev/null +++ b/vendor/github.com/mattermost/viper/flags_test.go @@ -0,0 +1,65 @@ +package viper + +import ( + "testing" + + "github.com/spf13/pflag" + "github.com/stretchr/testify/assert" +) + +func TestBindFlagValueSet(t *testing.T) { + flagSet := pflag.NewFlagSet("test", pflag.ContinueOnError) + + var testValues = map[string]*string{ + "host": nil, + "port": nil, + "endpoint": nil, + } + + var mutatedTestValues = map[string]string{ + "host": "localhost", + "port": "6060", + "endpoint": "/public", + } + + for name := range testValues { + testValues[name] = flagSet.String(name, "", "test") + } + + flagValueSet := pflagValueSet{flagSet} + + err := BindFlagValues(flagValueSet) + if err != nil { + t.Fatalf("error binding flag set, %v", err) + } + + flagSet.VisitAll(func(flag *pflag.Flag) { + flag.Value.Set(mutatedTestValues[flag.Name]) + flag.Changed = true + }) + + for name, expected := range mutatedTestValues { + assert.Equal(t, Get(name), expected) + } +} + +func TestBindFlagValue(t *testing.T) { + var testString = "testing" + var testValue = newStringValue(testString, &testString) + + flag := &pflag.Flag{ + Name: "testflag", + Value: testValue, + Changed: false, + } + + flagValue := pflagValue{flag} + BindFlagValue("testvalue", flagValue) + + assert.Equal(t, testString, Get("testvalue")) + + flag.Value.Set("testing_mutate") + flag.Changed = true //hack for pflag usage + + assert.Equal(t, "testing_mutate", Get("testvalue")) +} |