K8s: Add basic query service (#80325)

2024-01-31 10:36:51 -08:00
parent d1b938ba15
commit e013cd427c
38 changed files with 2143 additions and 203 deletions
@@ -0,0 +1,98 @@
+package dashboards
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+	"k8s.io/apimachinery/pkg/runtime/schema"
+
+	"github.com/grafana/grafana-plugin-sdk-go/backend"
+
+	query "github.com/grafana/grafana/pkg/apis/query/v0alpha1"
+	"github.com/grafana/grafana/pkg/services/datasources"
+	"github.com/grafana/grafana/pkg/services/featuremgmt"
+	"github.com/grafana/grafana/pkg/tests/apis"
+	"github.com/grafana/grafana/pkg/tests/testinfra"
+)
+
+func TestSimpleQuery(t *testing.T) {
+	if testing.Short() {
+		t.Skip("skipping integration test")
+	}
+	helper := apis.NewK8sTestHelper(t, testinfra.GrafanaOpts{
+		AppModeProduction: false, // dev mode required for datasource connections
+		EnableFeatureToggles: []string{
+			featuremgmt.FlagGrafanaAPIServerWithExperimentalAPIs, // Required to start the example service
+		},
+	})
+
+	// Create a single datasource
+	ds := helper.CreateDS(&datasources.AddDataSourceCommand{
+		Name:  "test",
+		Type:  datasources.DS_TESTDATA,
+		UID:   "test",
+		OrgID: int64(1),
+	})
+	require.Equal(t, "test", ds.UID)
+
+	t.Run("Call query", func(t *testing.T) {
+		client := helper.Org1.Admin.RESTClient(t, &schema.GroupVersion{
+			Group:   "query.grafana.app",
+			Version: "v0alpha1",
+		})
+
+		q := query.GenericDataQuery{
+			Datasource: &query.DataSourceRef{
+				Type: "grafana-testdata-datasource",
+				UID:  ds.UID,
+			},
+		}
+		q.AdditionalProperties()["csvContent"] = "a,b,c\n1,hello,true"
+		q.AdditionalProperties()["scenarioId"] = "csv_content"
+		body, err := json.Marshal(&query.GenericQueryRequest{Queries: []query.GenericDataQuery{q}})
+		require.NoError(t, err)
+
+		result := client.Post().
+			Namespace("default").
+			Suffix("query").
+			SetHeader("Content-type", "application/json").
+			Body(body).
+			Do(context.Background())
+
+		require.NoError(t, result.Error())
+
+		body, err = result.Raw()
+		require.NoError(t, err)
+		fmt.Printf("OUT: %s", string(body))
+
+		rsp := &backend.QueryDataResponse{}
+		err = json.Unmarshal(body, rsp)
+		require.NoError(t, err)
+		require.Equal(t, 1, len(rsp.Responses))
+
+		frame := rsp.Responses["A"].Frames[0]
+		disp, err := frame.StringTable(100, 10)
+		require.NoError(t, err)
+		fmt.Printf("%s\n", disp)
+
+		type expect struct {
+			idx  int
+			name string
+			val  any
+		}
+		for _, check := range []expect{
+			{0, "a", int64(1)},
+			{1, "b", "hello"},
+			{2, "c", true},
+		} {
+			field := frame.Fields[check.idx]
+			require.Equal(t, check.name, field.Name)
+
+			v, _ := field.ConcreteAt(0)
+			require.Equal(t, check.val, v)
+		}
+	})
+}