Plugins: Support Admission validation hooks (#87718)

pkg/tsdb/grafana-testdata-datasource/admission_handler_test.go

@@ -0,0 +1,68 @@
+package testdatasource
+
+import (
+	"context"
+	"encoding/json"
+	"testing"
+
+	"github.com/grafana/grafana-plugin-sdk-go/backend"
+	"github.com/stretchr/testify/require"
+)
+
+func TestSettingsHandler(t *testing.T) {
+	svc := &Service{}
+	require.NotNil(t, svc)
+
+	// Check missing datasource
+	s, err := svc.MutateAdmission(context.Background(), &backend.AdmissionRequest{
+		PluginContext: backend.PluginContext{},
+	})
+	require.NoError(t, err)
+	require.False(t, s.Allowed)
+	require.Equal(t, int32(400), s.Result.Code)
+
+	// Empty is OK
+	s, _ = svc.MutateAdmission(context.Background(),
+		asAdmissionRequest(&backend.DataSourceInstanceSettings{
+			APIVersion: "v0alpha1",
+		}))
+	require.True(t, s.Allowed)
+
+	// Any values should be an error
+	s, err = svc.MutateAdmission(context.Background(),
+		asAdmissionRequest(&backend.DataSourceInstanceSettings{
+			JSONData: json.RawMessage(`{"hello": "world"}`), // Settings must be empty
+		}))
+	require.NoError(t, err)
+	require.False(t, s.Allowed)
+	require.Equal(t, int32(400), s.Result.Code)
+
+	// Any values should be an error
+	s, err = svc.MutateAdmission(context.Background(),
+		asAdmissionRequest(&backend.DataSourceInstanceSettings{
+			DecryptedSecureJSONData: map[string]string{
+				"A": "Value",
+			},
+		}))
+	require.NoError(t, err)
+	require.False(t, s.Allowed)
+	require.Equal(t, int32(400), s.Result.Code)
+
+	// Invalid API Version
+	s, err = svc.MutateAdmission(context.Background(),
+		asAdmissionRequest(&backend.DataSourceInstanceSettings{
+			APIVersion: "v1234",
+		}))
+	require.NoError(t, err)
+	require.False(t, s.Allowed)
+	require.Equal(t, int32(400), s.Result.Code)
+}
+
+func asAdmissionRequest(settings *backend.DataSourceInstanceSettings) *backend.AdmissionRequest {
+	req := &backend.AdmissionRequest{}
+	if settings != nil {
+		req.Kind = settings.GVK()
+		req.ObjectBytes, _ = backend.DataSourceInstanceSettingsToProtoBytes(settings)
+	}
+	return req
+}