ad688d7761
searching for an alternative name for the resource. Note that the list is ordered and the first match will yield the extended resource's name. One important thing to highlight is that the ShortcutExpander will fall back to PriorityRestMaper to determine the group for the resource. Also this PR introduces a new shortcut namely sc which will resolve to storageclasses within storage.k8s.io group In addition the type of kubectl.ShortForms has been changed to ResourceShortcuts struct. It has got a brand new name, it was also extended and ordered by group.
100 lines
2.8 KiB
Go
100 lines
2.8 KiB
Go
/*
|
|
Copyright 2016 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package util
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"k8s.io/kubernetes/pkg/api/testapi"
|
|
"k8s.io/kubernetes/pkg/runtime/schema"
|
|
)
|
|
|
|
func TestReplaceAliases(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
arg string
|
|
expected string
|
|
}{
|
|
{
|
|
name: "no-replacement",
|
|
arg: "service",
|
|
expected: "service",
|
|
},
|
|
{
|
|
name: "all-replacement",
|
|
arg: "all",
|
|
expected: "pods,replicationcontrollers,services,statefulsets,horizontalpodautoscalers,jobs,deployments,replicasets",
|
|
},
|
|
{
|
|
name: "alias-in-comma-separated-arg",
|
|
arg: "all,secrets",
|
|
expected: "pods,replicationcontrollers,services,statefulsets,horizontalpodautoscalers,jobs,deployments,replicasets,secrets",
|
|
},
|
|
{
|
|
name: "sc-resolves-to-storageclasses",
|
|
arg: "sc",
|
|
expected: "storageclasses",
|
|
},
|
|
{
|
|
name: "storageclasses-no-replacement",
|
|
arg: "storageclasses",
|
|
expected: "storageclasses",
|
|
},
|
|
}
|
|
|
|
mapper := NewShortcutExpander(testapi.Default.RESTMapper(), nil)
|
|
|
|
for _, test := range tests {
|
|
resources := []string{}
|
|
for _, arg := range strings.Split(test.arg, ",") {
|
|
curr, _ := mapper.AliasesForResource(arg)
|
|
resources = append(resources, curr...)
|
|
}
|
|
if strings.Join(resources, ",") != test.expected {
|
|
t.Errorf("%s: unexpected argument: expected %s, got %s", test.name, test.expected, resources)
|
|
}
|
|
}
|
|
}
|
|
func TestKindFor(t *testing.T) {
|
|
tests := []struct {
|
|
in schema.GroupVersionResource
|
|
expected schema.GroupVersionKind
|
|
}{
|
|
{
|
|
in: schema.GroupVersionResource{Group: "storage.k8s.io", Version: "", Resource: "sc"},
|
|
expected: schema.GroupVersionKind{Group: "storage.k8s.io", Version: "v1beta1", Kind: "StorageClass"},
|
|
},
|
|
{
|
|
in: schema.GroupVersionResource{Group: "", Version: "", Resource: "sc"},
|
|
expected: schema.GroupVersionKind{Group: "storage.k8s.io", Version: "v1beta1", Kind: "StorageClass"},
|
|
},
|
|
}
|
|
|
|
mapper := NewShortcutExpander(testapi.Default.RESTMapper(), nil)
|
|
|
|
for i, test := range tests {
|
|
ret, err := mapper.KindFor(test.in)
|
|
if err != nil {
|
|
t.Errorf("%d: unexpected error returned %s", i, err.Error())
|
|
}
|
|
if ret != test.expected {
|
|
t.Errorf("%d: unexpected data returned %#v, expected %#v", i, ret, test.expected)
|
|
}
|
|
}
|
|
}
|