STACKITGit/cli
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Reset bool flags so they don't leak between tests

Browse source
This commit is contained in:
Mislav Marohnić 2019-12-04 16:47:11 +01:00
parent 5ac4b1e6d6
commit fb8a7e26e6
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
command/pr_test.go
View file

@ -35,10 +35,15 @@ func RunCommand(cmd *cobra.Command, args string) (string, error) {
outBuf := bytes.Buffer{}
cmd.SetOut(&outBuf)
// Reset flag slice values so they don't leak between tests
cmd.Flags().Visit(func(f *pflag.Flag) {
if v, ok := f.Value.(pflag.SliceValue); ok {
// Reset flag values so they don't leak between tests
cmd.Flags().VisitAll(func(f *pflag.Flag) {
switch v := f.Value.(type) {
case pflag.SliceValue:
v.Replace([]string{})
default:
if v.Type() == "bool" {
v.Set(f.DefValue)
}
}
})