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

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
}