mirror of
https://github.com/grafana/grafana.git
synced 2025-01-08 15:13:30 -06:00
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
59 lines
3.2 KiB
Go
59 lines
3.2 KiB
Go
package querydata
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/grafana/grafana/pkg/promlib/models"
|
|
"github.com/grafana/grafana/pkg/promlib/querydata/exemplar"
|
|
)
|
|
|
|
func TestQueryData_parseResponse(t *testing.T) {
|
|
qd := QueryData{exemplarSampler: exemplar.NewStandardDeviationSampler}
|
|
|
|
t.Run("resultType is before result the field must parsed normally", func(t *testing.T) {
|
|
resBody := `{"data":{"resultType":"vector", "result":[{"metric":{"__name__":"some_name","environment":"some_env","id":"some_id","instance":"some_instance:1234","job":"some_job","name":"another_name","region":"some_region"},"value":[1.1,"2"]}]},"status":"success"}`
|
|
res := &http.Response{Body: io.NopCloser(bytes.NewBufferString(resBody))}
|
|
result := qd.parseResponse(context.Background(), &models.Query{}, res, false)
|
|
assert.Nil(t, result.Error)
|
|
assert.Len(t, result.Frames, 1)
|
|
})
|
|
|
|
t.Run("resultType is after the result field must parsed normally", func(t *testing.T) {
|
|
resBody := `{"data":{"result":[{"metric":{"__name__":"some_name","environment":"some_env","id":"some_id","instance":"some_instance:1234","job":"some_job","name":"another_name","region":"some_region"},"value":[1.1,"2"]}],"resultType":"vector"},"status":"success"}`
|
|
res := &http.Response{Body: io.NopCloser(bytes.NewBufferString(resBody))}
|
|
result := qd.parseResponse(context.Background(), &models.Query{}, res, false)
|
|
assert.Nil(t, result.Error)
|
|
assert.Len(t, result.Frames, 1)
|
|
})
|
|
|
|
t.Run("no resultType is existed in the data", func(t *testing.T) {
|
|
resBody := `{"data":{"result":[{"metric":{"__name__":"some_name","environment":"some_env","id":"some_id","instance":"some_instance:1234","job":"some_job","name":"another_name","region":"some_region"},"value":[1.1,"2"]}]},"status":"success"}`
|
|
res := &http.Response{Body: io.NopCloser(bytes.NewBufferString(resBody))}
|
|
result := qd.parseResponse(context.Background(), &models.Query{}, res, false)
|
|
assert.Error(t, result.Error)
|
|
assert.Equal(t, result.Error.Error(), "no resultType found")
|
|
})
|
|
|
|
t.Run("resultType is set as empty string before result", func(t *testing.T) {
|
|
resBody := `{"data":{"resultType":"", "result":[{"metric":{"__name__":"some_name","environment":"some_env","id":"some_id","instance":"some_instance:1234","job":"some_job","name":"another_name","region":"some_region"},"value":[1.1,"2"]}]},"status":"success"}`
|
|
res := &http.Response{Body: io.NopCloser(bytes.NewBufferString(resBody))}
|
|
result := qd.parseResponse(context.Background(), &models.Query{}, res, false)
|
|
assert.Error(t, result.Error)
|
|
assert.Equal(t, result.Error.Error(), "unknown result type: ")
|
|
})
|
|
|
|
t.Run("resultType is set as empty string after result", func(t *testing.T) {
|
|
resBody := `{"data":{"result":[{"metric":{"__name__":"some_name","environment":"some_env","id":"some_id","instance":"some_instance:1234","job":"some_job","name":"another_name","region":"some_region"},"value":[1.1,"2"]}],"resultType":""},"status":"success"}`
|
|
res := &http.Response{Body: io.NopCloser(bytes.NewBufferString(resBody))}
|
|
result := qd.parseResponse(context.Background(), &models.Query{}, res, false)
|
|
assert.Error(t, result.Error)
|
|
assert.Equal(t, result.Error.Error(), "unknown result type: ")
|
|
})
|
|
}
|