feat(querier): propagate all known alerting headers (#92873)

pkg/registry/apis/query/query_test.go

@@ -14,6 +14,7 @@ import (
 	"github.com/grafana/grafana/pkg/infra/log"
 	"github.com/grafana/grafana/pkg/infra/tracing"
 	"github.com/grafana/grafana/pkg/services/featuremgmt"
+	"github.com/grafana/grafana/pkg/services/ngalert/models"
 	"github.com/stretchr/testify/require"
 	"k8s.io/apimachinery/pkg/runtime"
 )
@@ -37,7 +38,7 @@ func TestQueryRestConnectHandler(t *testing.T) {
 
 	rr := httptest.NewRecorder()
 	body := runtime.RawExtension{
-		Raw: []byte(`{ 
+		Raw: []byte(`{
 			"queries": [
 				{
 					"datasource": {
@@ -53,14 +54,23 @@ func TestQueryRestConnectHandler(t *testing.T) {
 			"to": "now"}`),
 	}
 	req := httptest.NewRequest(http.MethodGet, "/some-path", bytes.NewReader(body.Raw))
-	req.Header.Set("fromAlert", "true")
+	req.Header.Set(models.FromAlertHeaderName, "true")
+	req.Header.Set(models.CacheSkipHeaderName, "true")
+	req.Header.Set("X-Rule-Uid", "abc")
+	req.Header.Set("X-Rule-Folder", "folder-1")
+	req.Header.Set("X-Rule-Source", "grafana-ruler")
+	req.Header.Set("X-Grafana-Org-Id", "1")
 	req.Header.Set("Content-Type", "application/json")
 	req.Header.Set("some-unexpected-header", "some-value")
 	handler.ServeHTTP(rr, req)
 
 	require.Equal(t, map[string]string{
-		"FromAlert":    "true",
-		"Content-Type": "application/json",
+		models.FromAlertHeaderName: "true",
+		models.CacheSkipHeaderName: "true",
+		"X-Rule-Uid":               "abc",
+		"X-Rule-Folder":            "folder-1",
+		"X-Rule-Source":            "grafana-ruler",
+		"X-Grafana-Org-Id":         "1",
 	}, *b.client.(mockClient).lastCalledWithHeaders)
 }