update check for "all" resources

This commit is contained in:
juanvallejo 2017-01-17 18:13:23 -05:00
parent 56b1082deb
commit 1b8f4e0b7d
2 changed files with 13 additions and 4 deletions

View File

@ -806,12 +806,13 @@ func HasNames(args []string) (bool, error) {
// MultipleTypesRequested returns true if the provided args contain multiple resource kinds
func MultipleTypesRequested(args []string) bool {
if len(args) == 1 && args[0] == "all" {
return true
}
args = normalizeMultipleResourcesArgs(args)
rKinds := sets.NewString()
for _, arg := range args {
if arg == "all" {
return true
}
rTuple, found, err := splitResourceTypeName(arg)
if err != nil {
continue

View File

@ -1254,6 +1254,14 @@ func TestMultipleTypesRequested(t *testing.T) {
args: []string{"rc"},
expectedMultipleTypes: false,
},
{
args: []string{"pod,all"},
expectedMultipleTypes: true,
},
{
args: []string{"all,rc,pod"},
expectedMultipleTypes: true,
},
{
args: []string{"rc,pod,svc"},
expectedMultipleTypes: true,
@ -1286,7 +1294,7 @@ func TestMultipleTypesRequested(t *testing.T) {
for _, test := range tests {
hasMultipleTypes := MultipleTypesRequested(test.args)
if hasMultipleTypes != test.expectedMultipleTypes {
t.Errorf("expected HasName to return %v for %s", test.expectedMultipleTypes, test.args)
t.Errorf("expected MultipleTypesRequested to return %v for %s", test.expectedMultipleTypes, test.args)
}
}
}