Alerting: Include annotations in prometheus Alert response. (#45970)

* Alerting: Include annotations in prometheus Alert response.

* add tests

* re-order depedencies

pkg/services/ngalert/api/testing.go

@@ -2,10 +2,18 @@ package api
 
 import (
 	"context"
+	"fmt"
+	"sync"
 	"testing"
+	"time"
 
+	"github.com/grafana/grafana/pkg/services/ngalert/eval"
 	"github.com/grafana/grafana/pkg/services/ngalert/models"
+	"github.com/grafana/grafana/pkg/services/ngalert/state"
 	"github.com/grafana/grafana/pkg/services/ngalert/store"
+	"github.com/grafana/grafana/pkg/util"
+
+	"github.com/grafana/grafana-plugin-sdk-go/data"
 )
 
 type FakeAlertingStore struct {
@@ -30,3 +38,85 @@ func (f FakeAlertingStore) GetLatestAlertmanagerConfiguration(_ context.Context,
 	}
 	return store.ErrNoAlertmanagerConfiguration
 }
+
+type fakeAlertInstanceManager struct {
+	mtx sync.Mutex
+	// orgID -> RuleID -> States
+	states map[int64]map[string][]*state.State
+}
+
+func NewFakeAlertInstanceManager(t *testing.T) *fakeAlertInstanceManager {
+	t.Helper()
+
+	return &fakeAlertInstanceManager{
+		states: map[int64]map[string][]*state.State{},
+	}
+}
+
+func (f *fakeAlertInstanceManager) GetAll(orgID int64) []*state.State {
+	f.mtx.Lock()
+	defer f.mtx.Unlock()
+	var s []*state.State
+
+	for orgID := range f.states {
+		for _, states := range f.states[orgID] {
+			s = append(s, states...)
+		}
+	}
+
+	return s
+}
+
+func (f *fakeAlertInstanceManager) GetStatesForRuleUID(orgID int64, alertRuleUID string) []*state.State {
+	f.mtx.Lock()
+	defer f.mtx.Unlock()
+	return f.states[orgID][alertRuleUID]
+}
+
+func (f *fakeAlertInstanceManager) GenerateAlertInstances(orgID int64, count int) {
+	f.mtx.Lock()
+	defer f.mtx.Unlock()
+
+	evaluationTime := time.Now()
+	evaluationDuration := 1 * time.Minute
+	alertRuleUID := util.GenerateShortUID()
+
+	for i := 0; i < count; i++ {
+		_, ok := f.states[orgID]
+		if !ok {
+			f.states[orgID] = map[string][]*state.State{}
+		}
+		_, ok = f.states[orgID][alertRuleUID]
+		if !ok {
+			f.states[orgID][alertRuleUID] = []*state.State{}
+		}
+
+		f.states[orgID][alertRuleUID] = append(f.states[orgID][alertRuleUID], &state.State{
+			AlertRuleUID: fmt.Sprintf("alert_rule_%v", i),
+			OrgID:        1,
+			Labels: data.Labels{
+				"__alert_rule_namespace_uid__": "test_namespace_uid",
+				"__alert_rule_uid__":           fmt.Sprintf("test_alert_rule_uid_%v", i),
+				"alertname":                    fmt.Sprintf("test_title_%v", i),
+				"label":                        "test",
+				"instance_label":               "test",
+			},
+			State: eval.Normal,
+			Results: []state.Evaluation{
+				{
+					EvaluationTime:  evaluationTime,
+					EvaluationState: eval.Normal,
+					Values:          make(map[string]*float64),
+				},
+				{
+					EvaluationTime:  evaluationTime.Add(1 * time.Minute),
+					EvaluationState: eval.Normal,
+					Values:          make(map[string]*float64),
+				},
+			},
+			LastEvaluationTime: evaluationTime.Add(1 * time.Minute),
+			EvaluationDuration: evaluationDuration,
+			Annotations:        map[string]string{"annotation": "test"},
+		})
+	}
+}