mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
QueryService: Use types from sdk (#84029)
This commit is contained in:
@@ -4,9 +4,10 @@ import (
|
||||
"encoding/json"
|
||||
"testing"
|
||||
|
||||
data "github.com/grafana/grafana-plugin-sdk-go/experimental/apis/data/v0alpha1"
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"github.com/grafana/grafana/pkg/apis/query/v0alpha1"
|
||||
query "github.com/grafana/grafana/pkg/apis/query/v0alpha1"
|
||||
)
|
||||
|
||||
func TestParseQueriesIntoQueryDataRequest(t *testing.T) {
|
||||
@@ -39,23 +40,23 @@ func TestParseQueriesIntoQueryDataRequest(t *testing.T) {
|
||||
"to": "1692646267389"
|
||||
}`)
|
||||
|
||||
req := &v0alpha1.GenericQueryRequest{}
|
||||
req := &query.QueryDataRequest{}
|
||||
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"])
|
||||
require.Equal(t, "spreadsheetID", req.Queries[0].GetString("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{}
|
||||
query := &data.DataQuery{}
|
||||
err = json.Unmarshal(out, query)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, "spreadsheetID", query.AdditionalProperties()["spreadsheet"])
|
||||
require.Equal(t, "spreadsheetID", query.GetString("spreadsheet"))
|
||||
|
||||
// The second query has an explicit time range, and legacy datasource name
|
||||
out, err = json.MarshalIndent(req.Queries[1], "", " ")
|
||||
|
||||
Reference in New Issue
Block a user