mockkubeapiserver: Add basic CRD handling

When a user creates a CRD, register it as a recognized type.

Author: justinsb

mockkubeapiserver/apiserver.go

@@ -89,7 +89,7 @@ func (s *MockKubeAPIServer) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		}
 	}
 
-	klog.Infof("kubeapiserver request: %s %s", r.Method, r.URL)
+	klog.V(4).Infof("kubeapiserver request: %s %s", r.Method, r.URL)
 
 	path := r.URL.Path
 	tokens := strings.Split(strings.Trim(path, "/"), "/")

mockkubeapiserver/controllers.go

@@ -17,8 +17,13 @@ limitations under the License.
 package mockkubeapiserver
 
 import (
+	"fmt"
+
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
 	"k8s.io/apimachinery/pkg/runtime/schema"
+	"k8s.io/apimachinery/pkg/types"
+	"k8s.io/klog/v2"
 )
 
 func (s *MemoryStorage) objectChanged(u *unstructured.Unstructured) {
@@ -27,6 +32,10 @@ func (s *MemoryStorage) objectChanged(u *unstructured.Unstructured) {
 	switch gvk.GroupKind() {
 	case schema.GroupKind{Kind: "Namespace"}:
 		s.namespaceChanged(u)
+	case schema.GroupKind{Group: "apiextensions.k8s.io", Kind: "CustomResourceDefinition"}:
+		if err := s.crdChanged(u); err != nil {
+			klog.Warningf("crd change was invalid: %v", err)
+		}
 	}
 }
 
@@ -57,3 +66,87 @@ func (s *MemoryStorage) namespaceChanged(u *unstructured.Unstructured) {
 		unstructured.SetNestedSlice(u.Object, finalizers, "spec", "finalizers")
 	}
 }
+
+func (s *MemoryStorage) crdChanged(u *unstructured.Unstructured) error {
+	// TODO: Deleted / changed CRDs
+
+	group, _, _ := unstructured.NestedString(u.Object, "spec", "group")
+	if group == "" {
+		return fmt.Errorf("spec.group not set")
+	}
+
+	kind, _, _ := unstructured.NestedString(u.Object, "spec", "names", "kind")
+	if kind == "" {
+		return fmt.Errorf("spec.names.kind not set")
+	}
+
+	resource, _, _ := unstructured.NestedString(u.Object, "spec", "names", "plural")
+	if resource == "" {
+		return fmt.Errorf("spec.names.plural not set")
+	}
+
+	scope, _, _ := unstructured.NestedString(u.Object, "spec", "scope")
+	if scope == "" {
+		return fmt.Errorf("spec.scope not set")
+	}
+
+	versionsObj, found, _ := unstructured.NestedFieldNoCopy(u.Object, "spec", "versions")
+	if !found {
+		return fmt.Errorf("spec.versions not set")
+	}
+
+	versions, ok := versionsObj.([]interface{})
+	if !ok {
+		return fmt.Errorf("spec.versions not a slice")
+	}
+
+	for _, versionObj := range versions {
+		version, ok := versionObj.(map[string]interface{})
+		if !ok {
+			return fmt.Errorf("spec.versions element not an object")
+		}
+
+		versionName, _, _ := unstructured.NestedString(version, "name")
+		if versionName == "" {
+			return fmt.Errorf("version name not set")
+		}
+		gvk := schema.GroupVersionKind{Group: group, Version: versionName, Kind: kind}
+		gvr := gvk.GroupVersion().WithResource(resource)
+		gr := gvr.GroupResource()
+
+		storage := &resourceStorage{
+			GroupResource: gr,
+			objects:       make(map[types.NamespacedName]*unstructured.Unstructured),
+		}
+
+		// TODO: share storage across different versions
+		s.resourceStorages[gr] = storage
+
+		r := &ResourceInfo{
+			API: metav1.APIResource{
+				Name:    resource,
+				Group:   gvk.Group,
+				Version: gvk.Version,
+				Kind:    gvk.Kind,
+			},
+			GVK:     gvk,
+			GVR:     gvr,
+			storage: storage,
+		}
+		r.ListGVK = gvk.GroupVersion().WithKind(gvk.Kind + "List")
+
+		// TODO: Set r.TypeInfo from schema
+
+		switch scope {
+		case "Namespaced":
+			r.API.Namespaced = true
+		case "Cluster":
+			r.API.Namespaced = false
+		default:
+			return fmt.Errorf("scope %q is not recognized", scope)
+		}
+
+		s.schema.resources = append(s.schema.resources, r)
+	}
+	return nil
+}