K8s: Add subresource to the example apiserver (#78030)

2023-11-13 19:51:58 -08:00
parent a2a6f9a6d8
commit 1be1432926
16 changed files with 844 additions and 168 deletions
@@ -0,0 +1,167 @@
+package playlist
+
+import (
+	"context"
+	"encoding/json"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/runtime/schema"
+
+	"github.com/grafana/grafana/pkg/services/featuremgmt"
+	"github.com/grafana/grafana/pkg/tests/apis"
+	"github.com/grafana/grafana/pkg/tests/testinfra"
+)
+
+func TestExampleApp(t *testing.T) {
+	if testing.Short() {
+		t.Skip("skipping integration test")
+	}
+	helper := apis.NewK8sTestHelper(t, testinfra.GrafanaOpts{
+		AppModeProduction: true, // do not start extra port 6443
+		DisableAnonymous:  true,
+		EnableFeatureToggles: []string{
+			featuremgmt.FlagGrafanaAPIServer,
+			featuremgmt.FlagGrafanaAPIServerWithExperimentalAPIs, // Required to start the example service
+		},
+	})
+
+	t.Run("Check runtime info resource", func(t *testing.T) {
+		// Resource is not namespaced!
+		client := helper.Org1.Admin.Client.Resource(schema.GroupVersionResource{
+			Group:    "example.grafana.app",
+			Version:  "v0alpha1",
+			Resource: "runtime",
+		})
+		rsp, err := client.List(context.Background(), metav1.ListOptions{})
+		require.NoError(t, err)
+
+		v, ok := rsp.Object["startupTime"].(int64)
+		require.True(t, ok)
+		require.Greater(t, v, time.Now().Add(-1*time.Hour).UnixMilli()) // should be within the last hour
+	})
+
+	t.Run("Check discovery client", func(t *testing.T) {
+		disco := helper.NewDiscoveryClient()
+		resources, err := disco.ServerResourcesForGroupVersion("example.grafana.app/v0alpha1")
+		require.NoError(t, err)
+
+		v1Disco, err := json.MarshalIndent(resources, "", "  ")
+		require.NoError(t, err)
+		//	fmt.Printf("%s", string(v1Disco))
+
+		require.JSONEq(t, `{
+			"kind": "APIResourceList",
+			"apiVersion": "v1",
+			"groupVersion": "example.grafana.app/v0alpha1",
+			"resources": [
+			  {
+				"name": "dummy",
+				"singularName": "dummy",
+				"namespaced": true,
+				"kind": "DummyResource",
+				"verbs": [
+				  "get",
+				  "list"
+				]
+			  },
+			  {
+				"name": "dummy/sub",
+				"singularName": "",
+				"namespaced": true,
+				"kind": "DummySubresource",
+				"verbs": [
+				  "get"
+				]
+			  },
+			  {
+				"name": "runtime",
+				"singularName": "runtime",
+				"namespaced": false,
+				"kind": "RuntimeInfo",
+				"verbs": [
+				  "list"
+				]
+			  }
+			]
+		  }`, string(v1Disco))
+
+		//fmt.Printf("%s", string(v1Disco))
+		require.JSONEq(t, `[
+			{
+			  "version": "v0alpha1",
+			  "freshness": "Current",
+			  "resources": [
+				{
+				  "resource": "dummy",
+				  "responseKind": {
+					"group": "",
+					"kind": "DummyResource",
+					"version": ""
+				  },
+				  "scope": "Namespaced",
+				  "singularResource": "dummy",
+				  "subresources": [
+					{
+					  "responseKind": {
+						"group": "",
+						"kind": "DummySubresource",
+						"version": ""
+					  },
+					  "subresource": "sub",
+					  "verbs": [
+						"get"
+					  ]
+					}
+				  ],
+				  "verbs": [
+					"get",
+					"list"
+				  ]
+				},
+				{
+				  "resource": "runtime",
+				  "responseKind": {
+					"group": "",
+					"kind": "RuntimeInfo",
+					"version": ""
+				  },
+				  "scope": "Cluster",
+				  "singularResource": "runtime",
+				  "verbs": [
+					"list"
+				  ]
+				}
+			  ]
+			}
+		  ]`, helper.GetGroupVersionInfoJSON("example.grafana.app"))
+	})
+
+	t.Run("Check dummy with subresource", func(t *testing.T) {
+		client := helper.Org1.Viewer.Client.Resource(schema.GroupVersionResource{
+			Group:    "example.grafana.app",
+			Version:  "v0alpha1",
+			Resource: "dummy",
+		}).Namespace("default")
+		rsp, err := client.Get(context.Background(), "test2", metav1.GetOptions{})
+		require.NoError(t, err)
+
+		require.Equal(t, "dummy: test2", rsp.Object["spec"])
+		require.Equal(t, "DummyResource", rsp.GetObjectKind().GroupVersionKind().Kind)
+
+		// Now a sub-resource
+		rsp, err = client.Get(context.Background(), "test2", metav1.GetOptions{}, "sub")
+		require.NoError(t, err)
+
+		raw, err := json.MarshalIndent(rsp, "", "  ")
+		require.NoError(t, err)
+		//fmt.Printf("%s", string(raw))
+		require.JSONEq(t, `{
+			"apiVersion": "example.grafana.app/v0alpha1",
+			"kind": "DummySubresource",
+			"info": "default/viewer-1"
+		  }`, string(raw))
+	})
+}