From 5b3b9c11132685b07a628ce2a10f95992336bb50 Mon Sep 17 00:00:00 2001
From: Ryan McKinley <ryantxu@gmail.com>
Date: Mon, 6 Jan 2025 19:28:08 +0300
Subject: [PATCH] Search: Add label query test (#98348)

---
 pkg/storage/unified/search/bleve_test.go | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/pkg/storage/unified/search/bleve_test.go b/pkg/storage/unified/search/bleve_test.go
index 60f795a80f9..b3b9873646f 100644
--- a/pkg/storage/unified/search/bleve_test.go
+++ b/pkg/storage/unified/search/bleve_test.go
@@ -77,6 +77,9 @@ func TestBleveBackend(t *testing.T) {
 					DASHBOARD_PANEL_TYPES:  []string{"timeseries", "table"},
 					DASHBOARD_ERRORS_TODAY: 25,
 				},
+				Labels: map[string]string{
+					utils.LabelKeyDeprecatedInternalID: "10", // nolint:staticcheck
+				},
 				Tags: []string{"aa", "bb"},
 			})
 			_ = index.Write(&resource.IndexableDocument{
@@ -97,7 +100,8 @@ func TestBleveBackend(t *testing.T) {
 				},
 				Tags: []string{"aa"},
 				Labels: map[string]string{
-					"region": "east",
+					"region":                           "east",
+					utils.LabelKeyDeprecatedInternalID: "11", // nolint:staticcheck
 				},
 			})
 			_ = index.Write(&resource.IndexableDocument{
@@ -176,6 +180,24 @@ func TestBleveBackend(t *testing.T) {
 
 		count, _ = index.DocCount(ctx, "zzz")
 		assert.Equal(t, int64(1), count)
+
+		rsp, err = index.Search(ctx, nil, &resource.ResourceSearchRequest{
+			Options: &resource.ListOptions{
+				Key: key,
+				Labels: []*resource.Requirement{{
+					Key:      utils.LabelKeyDeprecatedInternalID, // nolint:staticcheck
+					Operator: "in",
+					Values:   []string{"10", "11"},
+				}},
+			},
+			Limit: 100000,
+		}, nil)
+		require.NoError(t, err)
+		require.Equal(t, int64(2), rsp.TotalHits)
+		require.Equal(t, []string{"aaa", "bbb"}, []string{
+			rsp.Results.Rows[0].Key.Name,
+			rsp.Results.Rows[1].Key.Name,
+		})
 	})
 
 	t.Run("build folders", func(t *testing.T) {