mirror of https://github.com/grafana/grafana.git
65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
package pyroscope
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// This is where the tests for the datasource backend live.
|
|
func Test_QueryData(t *testing.T) {
|
|
ds := PyroscopeDatasource{}
|
|
|
|
resp, err := ds.QueryData(
|
|
context.Background(),
|
|
&backend.QueryDataRequest{
|
|
Queries: []backend.DataQuery{
|
|
{RefID: "A"},
|
|
},
|
|
},
|
|
)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if len(resp.Responses) != 1 {
|
|
t.Fatal("QueryData must return a response")
|
|
}
|
|
}
|
|
|
|
func Test_CallResource(t *testing.T) {
|
|
ds := &PyroscopeDatasource{
|
|
client: &FakeClient{},
|
|
}
|
|
|
|
t.Run("series resource", func(t *testing.T) {
|
|
sender := &FakeSender{}
|
|
err := ds.CallResource(
|
|
context.Background(),
|
|
&backend.CallResourceRequest{
|
|
PluginContext: backend.PluginContext{},
|
|
Path: "profileTypes",
|
|
Method: "GET",
|
|
URL: "profileTypes",
|
|
Headers: nil,
|
|
Body: nil,
|
|
},
|
|
sender,
|
|
)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 200, sender.Resp.Status)
|
|
require.Equal(t, `[{"id":"type:1","label":"cpu"},{"id":"type:2","label":"memory"}]`, string(sender.Resp.Body))
|
|
})
|
|
}
|
|
|
|
type FakeSender struct {
|
|
Resp *backend.CallResourceResponse
|
|
}
|
|
|
|
func (fs *FakeSender) Send(resp *backend.CallResourceResponse) error {
|
|
fs.Resp = resp
|
|
return nil
|
|
}
|