113 lines
2.9 KiB
Go
113 lines
2.9 KiB
Go
/*
|
|
Copyright 2017 Heptio Inc.
|
|
|
|
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 test
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"k8s.io/apimachinery/pkg/api/meta"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
)
|
|
|
|
type FakeDiscoveryHelper struct {
|
|
ResourceList []*metav1.APIResourceList
|
|
Mapper meta.RESTMapper
|
|
AutoReturnResource bool
|
|
}
|
|
|
|
func NewFakeDiscoveryHelper(autoReturnResource bool, resources map[schema.GroupVersionResource]schema.GroupVersionResource) *FakeDiscoveryHelper {
|
|
helper := &FakeDiscoveryHelper{
|
|
AutoReturnResource: autoReturnResource,
|
|
Mapper: &FakeMapper{
|
|
Resources: resources,
|
|
},
|
|
}
|
|
|
|
if resources == nil {
|
|
return helper
|
|
}
|
|
|
|
apiResourceMap := make(map[string][]metav1.APIResource)
|
|
|
|
for _, gvr := range resources {
|
|
var gvString string
|
|
if gvr.Version != "" && gvr.Group != "" {
|
|
gvString = fmt.Sprintf("%s/%s", gvr.Group, gvr.Version)
|
|
} else {
|
|
gvString = fmt.Sprintf("%s%s", gvr.Group, gvr.Version)
|
|
}
|
|
|
|
apiResourceMap[gvString] = append(apiResourceMap[gvString], metav1.APIResource{Name: gvr.Resource})
|
|
}
|
|
|
|
for group, resources := range apiResourceMap {
|
|
helper.ResourceList = append(helper.ResourceList, &metav1.APIResourceList{GroupVersion: group, APIResources: resources})
|
|
}
|
|
|
|
return helper
|
|
}
|
|
|
|
func (dh *FakeDiscoveryHelper) Resources() []*metav1.APIResourceList {
|
|
return dh.ResourceList
|
|
}
|
|
|
|
func (dh *FakeDiscoveryHelper) Refresh() error {
|
|
return nil
|
|
}
|
|
|
|
func (dh *FakeDiscoveryHelper) ResourceFor(input schema.GroupVersionResource) (schema.GroupVersionResource, metav1.APIResource, error) {
|
|
if dh.AutoReturnResource {
|
|
return schema.GroupVersionResource{
|
|
Group: input.Group,
|
|
Version: input.Version,
|
|
Resource: input.Resource,
|
|
},
|
|
metav1.APIResource{
|
|
Name: input.Resource,
|
|
},
|
|
nil
|
|
}
|
|
|
|
gvr, err := dh.Mapper.ResourceFor(input)
|
|
if err != nil {
|
|
return schema.GroupVersionResource{}, metav1.APIResource{}, err
|
|
}
|
|
|
|
var gvString string
|
|
if gvr.Version != "" && gvr.Group != "" {
|
|
gvString = fmt.Sprintf("%s/%s", gvr.Group, gvr.Version)
|
|
} else {
|
|
gvString = gvr.Version + gvr.Group
|
|
}
|
|
|
|
for _, gr := range dh.ResourceList {
|
|
if gr.GroupVersion != gvString {
|
|
continue
|
|
}
|
|
|
|
for _, resource := range gr.APIResources {
|
|
if resource.Name == gvr.Resource {
|
|
return gvr, resource, nil
|
|
}
|
|
}
|
|
}
|
|
|
|
return schema.GroupVersionResource{}, metav1.APIResource{}, errors.New("APIResource not found")
|
|
}
|