mirror of
https://github.com/kyverno/kyverno.git
synced 2024-12-15 17:51:20 +00:00
111 lines
3.4 KiB
Go
111 lines
3.4 KiB
Go
package client
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
openapi_v2 "github.com/googleapis/gnostic/OpenAPIv2"
|
|
|
|
meta "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
"k8s.io/apimachinery/pkg/version"
|
|
"k8s.io/client-go/dynamic/fake"
|
|
kubernetesfake "k8s.io/client-go/kubernetes/fake"
|
|
)
|
|
|
|
const (
|
|
// CSRs CertificateSigningRequest
|
|
CSRs string = "CertificateSigningRequest"
|
|
// Secrets Secret
|
|
Secrets string = "Secret"
|
|
// ConfigMaps ConfigMap
|
|
ConfigMaps string = "ConfigMap"
|
|
// Namespaces Namespace
|
|
Namespaces string = "Namespace"
|
|
)
|
|
|
|
//NewMockClient ---testing utilities
|
|
func NewMockClient(scheme *runtime.Scheme, objects ...runtime.Object) (*Client, error) {
|
|
client := fake.NewSimpleDynamicClient(scheme, objects...)
|
|
// the typed and dynamic client are initialized with similar resources
|
|
kclient := kubernetesfake.NewSimpleClientset(objects...)
|
|
return &Client{
|
|
client: client,
|
|
kclient: kclient,
|
|
}, nil
|
|
|
|
}
|
|
|
|
// NewFakeDiscoveryClient returns a fakediscovery client
|
|
func NewFakeDiscoveryClient(registeredResouces []schema.GroupVersionResource) *fakeDiscoveryClient {
|
|
// Load some-preregistd resources
|
|
res := []schema.GroupVersionResource{
|
|
{Version: "v1", Resource: "configmaps"},
|
|
{Version: "v1", Resource: "endpoints"},
|
|
{Version: "v1", Resource: "namespaces"},
|
|
{Version: "v1", Resource: "resourcequotas"},
|
|
{Version: "v1", Resource: "secrets"},
|
|
{Version: "v1", Resource: "serviceaccounts"},
|
|
{Group: "apps", Version: "v1", Resource: "daemonsets"},
|
|
{Group: "apps", Version: "v1", Resource: "deployments"},
|
|
{Group: "apps", Version: "v1", Resource: "statefulsets"},
|
|
}
|
|
registeredResouces = append(registeredResouces, res...)
|
|
return &fakeDiscoveryClient{registeredResouces: registeredResouces}
|
|
}
|
|
|
|
type fakeDiscoveryClient struct {
|
|
registeredResouces []schema.GroupVersionResource
|
|
}
|
|
|
|
func (c *fakeDiscoveryClient) getGVR(resource string) schema.GroupVersionResource {
|
|
for _, gvr := range c.registeredResouces {
|
|
if gvr.Resource == resource {
|
|
return gvr
|
|
}
|
|
}
|
|
return schema.GroupVersionResource{}
|
|
}
|
|
|
|
func (c *fakeDiscoveryClient) GetServerVersion() (*version.Info, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (c *fakeDiscoveryClient) GetGVRFromKind(kind string) schema.GroupVersionResource {
|
|
resource := strings.ToLower(kind) + "s"
|
|
return c.getGVR(resource)
|
|
}
|
|
|
|
func (c *fakeDiscoveryClient) GetGVRFromAPIVersionKind(apiVersion string, kind string) schema.GroupVersionResource {
|
|
resource := strings.ToLower(kind) + "s"
|
|
return c.getGVR(resource)
|
|
}
|
|
|
|
func (c *fakeDiscoveryClient) FindResource(apiVersion string, kind string) (*meta.APIResource, schema.GroupVersionResource, error) {
|
|
return nil, schema.GroupVersionResource{}, fmt.Errorf("Not implemented")
|
|
}
|
|
|
|
func (c *fakeDiscoveryClient) OpenAPISchema() (*openapi_v2.Document, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func newUnstructured(apiVersion, kind, namespace, name string) *unstructured.Unstructured {
|
|
return &unstructured.Unstructured{
|
|
Object: map[string]interface{}{
|
|
"apiVersion": apiVersion,
|
|
"kind": kind,
|
|
"metadata": map[string]interface{}{
|
|
"namespace": namespace,
|
|
"name": name,
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func newUnstructuredWithSpec(apiVersion, kind, namespace, name string, spec map[string]interface{}) *unstructured.Unstructured {
|
|
u := newUnstructured(apiVersion, kind, namespace, name)
|
|
u.Object["spec"] = spec
|
|
return u
|
|
}
|