Method TestStringWithAllowedValues

benchmark/flags/flags_test.go:38–65

(t *testing.T)

}

func (s) TestStringWithAllowedValues(t *testing.T) {
	const defaultVal = "default"
	tests := []struct {
		args    string
		allowed []string
		wantVal string
		wantErr bool
	}{
		{"-workloads=all", []string{"unary", "streaming", "all"}, "all", false},
		{"-workloads=disallowed", []string{"unary", "streaming", "all"}, defaultVal, true},
	}

	for _, test := range tests {
		flag.CommandLine = flag.NewFlagSet("test", flag.ContinueOnError)
		var w = StringWithAllowedValues("workloads", defaultVal, "usage", test.allowed)
		err := flag.CommandLine.Parse([]string{test.args})
		switch {
		case !test.wantErr && err != nil:
			t.Errorf("failed to parse command line args {%v}: %v", test.args, err)
		case test.wantErr && err == nil:
			t.Errorf("flag.Parse(%v) = nil, want non-nil error", test.args)
		default:
			if *w != test.wantVal {
				t.Errorf("flag value is %v, want %v", *w, test.wantVal)
			}
		}
	}
}

func (s) TestDurationSlice(t *testing.T) {
	defaultVal := []time.Duration{time.Second, time.Nanosecond}