From bb0a6d7d184fecdf76de1353a2b97e97378e3d83 Mon Sep 17 00:00:00 2001
From: "grafana-delivery-bot[bot]"
 <132647405+grafana-delivery-bot[bot]@users.noreply.github.com>
Date: Wed, 19 Jul 2023 15:07:25 +0300
Subject: [PATCH] [v10.0.x] Alerting: Add tests for matching captures (#71939)

Alerting: Add tests for matching captures (#71928)

This commit adds tests for matching captures, which we do not have
at present.

(cherry picked from commit f1af0502db219871d9f771a6e27555baf057334c)

Co-authored-by: George Robinson <george.robinson@grafana.com>
---
 pkg/services/ngalert/eval/eval_test.go | 105 +++++++++++++++++++++++++
 1 file changed, 105 insertions(+)

diff --git a/pkg/services/ngalert/eval/eval_test.go b/pkg/services/ngalert/eval/eval_test.go
index 5708bfe927e..b93dc33f953 100644
--- a/pkg/services/ngalert/eval/eval_test.go
+++ b/pkg/services/ngalert/eval/eval_test.go
@@ -595,6 +595,111 @@ func TestEvaluate(t *testing.T) {
 				"ref_id":         "A",
 			},
 		}},
+	}, {
+		name: "results contains captured values for exact label matches",
+		cond: models.Condition{
+			Condition: "B",
+		},
+		resp: backend.QueryDataResponse{
+			Responses: backend.Responses{
+				"A": {
+					Frames: []*data.Frame{{
+						RefID: "A",
+						Fields: []*data.Field{
+							data.NewField(
+								"Value",
+								data.Labels{"foo": "bar"},
+								[]*float64{util.Pointer(10.0)},
+							),
+						},
+					}},
+				},
+				"B": {
+					Frames: []*data.Frame{{
+						RefID: "B",
+						Fields: []*data.Field{
+							data.NewField(
+								"Value",
+								data.Labels{"foo": "bar"},
+								[]*float64{util.Pointer(1.0)},
+							),
+						},
+					}},
+				},
+			},
+		},
+		expected: Results{{
+			State: Alerting,
+			Instance: data.Labels{
+				"foo": "bar",
+			},
+			Values: map[string]NumberValueCapture{
+				"A": {
+					Var:    "A",
+					Labels: data.Labels{"foo": "bar"},
+					Value:  util.Pointer(10.0),
+				},
+				"B": {
+					Var:    "B",
+					Labels: data.Labels{"foo": "bar"},
+					Value:  util.Pointer(1.0),
+				},
+			},
+			EvaluationString: "[ var='A' labels={foo=bar} value=10 ], [ var='B' labels={foo=bar} value=1 ]",
+		}},
+	}, {
+		name: "results contains captured values for subset of labels",
+		cond: models.Condition{
+			Condition: "B",
+		},
+		resp: backend.QueryDataResponse{
+			Responses: backend.Responses{
+				"A": {
+					Frames: []*data.Frame{{
+						RefID: "A",
+						Fields: []*data.Field{
+							data.NewField(
+								"Value",
+								data.Labels{"foo": "bar"},
+								[]*float64{util.Pointer(10.0)},
+							),
+						},
+					}},
+				},
+				"B": {
+					Frames: []*data.Frame{{
+						RefID: "B",
+						Fields: []*data.Field{
+							data.NewField(
+								"Value",
+								data.Labels{"foo": "bar", "bar": "baz"},
+								[]*float64{util.Pointer(1.0)},
+							),
+						},
+					}},
+				},
+			},
+		},
+		expected: Results{{
+			State: Alerting,
+			Instance: data.Labels{
+				"foo": "bar",
+				"bar": "baz",
+			},
+			Values: map[string]NumberValueCapture{
+				"A": {
+					Var:    "A",
+					Labels: data.Labels{"foo": "bar"},
+					Value:  util.Pointer(10.0),
+				},
+				"B": {
+					Var:    "B",
+					Labels: data.Labels{"foo": "bar", "bar": "baz"},
+					Value:  util.Pointer(1.0),
+				},
+			},
+			EvaluationString: "[ var='A' labels={foo=bar} value=10 ], [ var='B' labels={bar=baz, foo=bar} value=1 ]",
+		}},
 	}}
 
 	for _, tc := range cases {