ismail simsek
3fb6319d1b
* Move files to prometheus-library * refactor core prometheus to use prometheus-library * modify client transport options * mock * have a type * import aliases * rename * call the right method * remove unrelated test from the library * update codeowners * go work sync * update go.work.sum * make swagger-clean && make openapi3-gen * add promlib to makefile * remove clilogger * Export the function * update unit test * add prometheus_test.go * fix mock type * use mapUtil from grafana-plugin-sdk-go
89 lines
2.8 KiB
Go
89 lines
2.8 KiB
Go
package promlib
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend/datasource"
|
|
sdkhttpclient "github.com/grafana/grafana-plugin-sdk-go/backend/httpclient"
|
|
)
|
|
|
|
type heuristicsSuccessRoundTripper struct {
|
|
res io.ReadCloser
|
|
status int
|
|
}
|
|
|
|
func (rt *heuristicsSuccessRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
return &http.Response{
|
|
Status: strconv.Itoa(rt.status),
|
|
StatusCode: rt.status,
|
|
Header: nil,
|
|
Body: rt.res,
|
|
ContentLength: 0,
|
|
Request: req,
|
|
}, nil
|
|
}
|
|
|
|
func newHeuristicsSDKProvider(hrt heuristicsSuccessRoundTripper) *sdkhttpclient.Provider {
|
|
anotherFN := func(o sdkhttpclient.Options, next http.RoundTripper) http.RoundTripper {
|
|
return &hrt
|
|
}
|
|
fn := sdkhttpclient.MiddlewareFunc(anotherFN)
|
|
mid := sdkhttpclient.NamedMiddlewareFunc("mock", fn)
|
|
return sdkhttpclient.NewProvider(sdkhttpclient.ProviderOptions{Middlewares: []sdkhttpclient.Middleware{mid}})
|
|
}
|
|
|
|
func mockExtendClientOpts(ctx context.Context, settings backend.DataSourceInstanceSettings, clientOpts *sdkhttpclient.Options) error {
|
|
return nil
|
|
}
|
|
|
|
func Test_GetHeuristics(t *testing.T) {
|
|
t.Run("should return Prometheus", func(t *testing.T) {
|
|
rt := heuristicsSuccessRoundTripper{
|
|
res: io.NopCloser(strings.NewReader("{\"status\":\"success\",\"data\":{\"version\":\"1.0\"}}")),
|
|
status: http.StatusOK,
|
|
}
|
|
httpProvider := newHeuristicsSDKProvider(rt)
|
|
s := &Service{
|
|
im: datasource.NewInstanceManager(newInstanceSettings(httpProvider, backend.NewLoggerWith("logger", "test"), mockExtendClientOpts)),
|
|
}
|
|
|
|
req := HeuristicsRequest{
|
|
PluginContext: getPluginContext(),
|
|
}
|
|
res, err := s.GetHeuristics(context.Background(), req)
|
|
assert.NoError(t, err)
|
|
require.NotNil(t, res)
|
|
assert.Equal(t, KindPrometheus, res.Application)
|
|
assert.Equal(t, Features{RulerApiEnabled: false}, res.Features)
|
|
})
|
|
|
|
t.Run("should return Mimir", func(t *testing.T) {
|
|
rt := heuristicsSuccessRoundTripper{
|
|
res: io.NopCloser(strings.NewReader("{\"status\":\"success\",\"data\":{\"features\":{\"foo\":\"bar\"},\"version\":\"1.0\"}}")),
|
|
status: http.StatusOK,
|
|
}
|
|
httpProvider := newHeuristicsSDKProvider(rt)
|
|
s := &Service{
|
|
im: datasource.NewInstanceManager(newInstanceSettings(httpProvider, backend.NewLoggerWith("logger", "test"), mockExtendClientOpts)),
|
|
}
|
|
|
|
req := HeuristicsRequest{
|
|
PluginContext: getPluginContext(),
|
|
}
|
|
res, err := s.GetHeuristics(context.Background(), req)
|
|
assert.NoError(t, err)
|
|
require.NotNil(t, res)
|
|
assert.Equal(t, KindMimir, res.Application)
|
|
assert.Equal(t, Features{RulerApiEnabled: true}, res.Features)
|
|
})
|
|
}
|