public/grafana
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Loki: fix status in body not matching http status code (#114201)

Browse Source 
* fix: status in body not matching http status code
This commit is contained in:
Galen Kistler
2025-11-21 04:40:29 -06:00
committed by GitHub
parent 42babc7ce7
commit 3d2f702ae5
2 changed files with 46 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/tsdb/loki/api_test.go
+41 -8
View File
@@ -250,6 +250,7 @@ func TestApiReturnValues(t *testing.T) {
require.True(t, called)
require.Equal(t, "gzip", encodedBytes.Encoding)
require.Equal(t, []byte("{\"message\":\"foo\"}"), encodedBytes.Body)
require.Equal(t, 400, encodedBytes.Status)
})
t.Run("Loki should return the error as is", func(t *testing.T) {
@@ -280,6 +281,18 @@ func TestApiReturnValues(t *testing.T) {
require.Error(t, err)
require.ErrorContains(t, err, "foo")
})
t.Run("should set status for successful requests", func(t *testing.T) {
called := false
api := makeMockedAPI(200, "application/json", []byte("{\"message\":\"foo\"}"), func(req *http.Request) {
called = true
})
res, err := api.DataQuery(context.Background(), lokiQuery{Expr: "", SupportingQueryType: SupportingQueryLogsVolume, QueryType: QueryTypeRange}, ResponseOpts{})
require.NoError(t, err)
require.True(t, called)
require.Equal(t, backend.Status(http.StatusOK), res.Status)
})
}
func TestErrorSources(t *testing.T) {
@@ -287,7 +300,7 @@ func TestErrorSources(t *testing.T) {
t.Run("should set correct error source for downstream errors", func(t *testing.T) {
called := false
api := makeMockedAPI(400, "application/json", errorResponse, func(req *http.Request) {
api := makeMockedAPI(http.StatusBadRequest, "application/json", errorResponse, func(req *http.Request) {
called = true
})
@@ -296,11 +309,12 @@ func TestErrorSources(t *testing.T) {
require.True(t, called)
require.NotNil(t, res.Error)
require.Equal(t, backend.ErrorSourceDownstream, res.ErrorSource)
require.Equal(t, backend.Status(http.StatusBadRequest), res.Status)
})
t.Run("should set correct error source for plugin errors", func(t *testing.T) {
called := false
api := makeMockedAPI(406, "application/json", errorResponse, func(req *http.Request) {
api := makeMockedAPI(http.StatusNotAcceptable, "application/json", errorResponse, func(req *http.Request) {
called = true
})
@@ -309,11 +323,12 @@ func TestErrorSources(t *testing.T) {
require.True(t, called)
require.NotNil(t, res.Error)
require.Equal(t, backend.ErrorSourcePlugin, res.ErrorSource)
require.Equal(t, backend.Status(http.StatusNotAcceptable), res.Status)
})
t.Run("should set correct error source for server errors", func(t *testing.T) {
called := false
api := makeMockedAPI(500, "application/json", errorResponse, func(req *http.Request) {
api := makeMockedAPI(http.StatusInternalServerError, "application/json", errorResponse, func(req *http.Request) {
called = true
})
@@ -322,11 +337,26 @@ func TestErrorSources(t *testing.T) {
require.True(t, called)
require.NotNil(t, res.Error)
require.Equal(t, backend.ErrorSourceDownstream, res.ErrorSource)
require.Equal(t, backend.Status(http.StatusInternalServerError), res.Status)
})
t.Run("should set correct error source for server timeout error", func(t *testing.T) {
called := false
api := makeMockedAPI(http.StatusGatewayTimeout, "application/json", errorResponse, func(req *http.Request) {
called = true
})
res, err := api.DataQuery(context.Background(), lokiQuery{QueryType: QueryTypeRange}, ResponseOpts{})
require.NoError(t, err)
require.True(t, called)
require.NotNil(t, res.Error)
require.Equal(t, backend.ErrorSourceDownstream, res.ErrorSource)
require.Equal(t, backend.Status(http.StatusGatewayTimeout), res.Status)
})
t.Run("should handle downstream HTTP errors", func(t *testing.T) {
called := false
api := makeMockedAPI(400, "application/json", errorResponse, func(req *http.Request) {
api := makeMockedAPI(http.StatusBadRequest, "application/json", errorResponse, func(req *http.Request) {
called = true
})
@@ -336,30 +366,33 @@ func TestErrorSources(t *testing.T) {
require.NotNil(t, res.Error)
require.Equal(t, backend.ErrorSourceDownstream, res.ErrorSource)
require.Contains(t, res.Error.Error(), "test error")
require.Equal(t, backend.Status(http.StatusBadRequest), res.Status)
})
t.Run("should handle client errors in RawQuery", func(t *testing.T) {
called := false
api := makeMockedAPI(400, "application/json", errorResponse, func(req *http.Request) {
api := makeMockedAPI(http.StatusBadRequest, "application/json", errorResponse, func(req *http.Request) {
called = true
})
res, err := api.RawQuery(context.Background(), "/loki/api/v1/labels")
require.NoError(t, err)
require.True(t, called)
require.Equal(t, 400, res.Status)
require.Equal(t, http.StatusBadRequest, res.Status)
require.Contains(t, string(res.Body), "test error")
})
t.Run("should handle server errors in RawQuery", func(t *testing.T) {
called := false
api := makeMockedAPI(500, "application/json", errorResponse, func(req *http.Request) {
api := makeMockedAPI(http.StatusInternalServerError, "application/json", errorResponse, func(req *http.Request) {
called = true
})
_, err := api.RawQuery(context.Background(), "/loki/api/v1/labels")
res, err := api.RawQuery(context.Background(), "/loki/api/v1/labels")
require.Error(t, err)
require.True(t, called)
require.Contains(t, err.Error(), "test error")
// Status code of 0 gets mapped to InternalServerError (500)
require.Equal(t, 0, res.Status)
})
}