K8s: Add basic query service (#80325)

2024-01-31 20:36:51 +02:00
parent d1b938ba15
commit e013cd427c
38 changed files with 2143 additions and 203 deletions
@@ -0,0 +1,75 @@
+package v0alpha1_test
+
+import (
+	"encoding/json"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/grafana/grafana/pkg/apis/query/v0alpha1"
+)
+
+func TestParseQueriesIntoQueryDataRequest(t *testing.T) {
+	request := []byte(`{
+		"queries": [
+			{
+				"refId": "A",
+				"datasource": {
+					"type": "grafana-googlesheets-datasource",
+					"uid": "b1808c48-9fc9-4045-82d7-081781f8a553"
+				},
+				"cacheDurationSeconds": 300,
+				"spreadsheet": "spreadsheetID",
+				"datasourceId": 4,
+				"intervalMs": 30000,
+				"maxDataPoints": 794
+			},
+			{
+				"refId": "Z",
+				"datasource": "old",
+				"maxDataPoints": 10,
+				"timeRange": {
+					"from": "100",
+					"to": "200"
+				}
+			}
+		],
+		"from": "1692624667389",
+		"to": "1692646267389"
+	}`)
+
+	req := &v0alpha1.GenericQueryRequest{}
+	err := json.Unmarshal(request, req)
+	require.NoError(t, err)
+
+	require.Len(t, req.Queries, 2)
+	require.Equal(t, "b1808c48-9fc9-4045-82d7-081781f8a553", req.Queries[0].Datasource.UID)
+	require.Equal(t, "spreadsheetID", req.Queries[0].AdditionalProperties()["spreadsheet"])
+
+	// Write the query (with additional spreadsheetID) to JSON
+	out, err := json.MarshalIndent(req.Queries[0], "", "  ")
+	require.NoError(t, err)
+
+	// And read it back with standard JSON marshal functions
+	query := &v0alpha1.GenericDataQuery{}
+	err = json.Unmarshal(out, query)
+	require.NoError(t, err)
+	require.Equal(t, "spreadsheetID", query.AdditionalProperties()["spreadsheet"])
+
+	// The second query has an explicit time range, and legacy datasource name
+	out, err = json.MarshalIndent(req.Queries[1], "", "  ")
+	require.NoError(t, err)
+	// fmt.Printf("%s\n", string(out))
+	require.JSONEq(t, `{
+		"datasource": {
+		  "type": "", ` /* NOTE! this implies legacy naming */ +`
+		  "uid": "old"
+		},
+		"maxDataPoints": 10,
+		"refId": "Z",
+		"timeRange": {
+		  "from": "100",
+		  "to": "200"
+		}
+	  }`, string(out))
+}