From 6e5f44d856fc9a9a5cbdfbd1eda5dcc01497de34 Mon Sep 17 00:00:00 2001
From: Kaushal Kumar <ravi.kaushal97@gmail.com>
Date: Wed, 22 Jan 2025 20:49:15 -0800
Subject: [PATCH] fix UT failures

Signed-off-by: Kaushal Kumar <ravi.kaushal97@gmail.com>
---
 .../wlm/tracker/ResourceUsageCalculatorTrackerServiceTests.java  | 1 +
 1 file changed, 1 insertion(+)

diff --git a/server/src/test/java/org/opensearch/wlm/tracker/ResourceUsageCalculatorTrackerServiceTests.java b/server/src/test/java/org/opensearch/wlm/tracker/ResourceUsageCalculatorTrackerServiceTests.java
index fe72bd6e710c8..7de612969f8d3 100644
--- a/server/src/test/java/org/opensearch/wlm/tracker/ResourceUsageCalculatorTrackerServiceTests.java
+++ b/server/src/test/java/org/opensearch/wlm/tracker/ResourceUsageCalculatorTrackerServiceTests.java
@@ -146,6 +146,7 @@ private <T extends QueryGroupTask> T createMockTask(Class<T> type, long cpuUsage
         when(task.getTotalResourceUtilization(ResourceStats.MEMORY)).thenReturn(heapUsage);
         when(task.getStartTimeNanos()).thenReturn((long) 0);
         when(task.getElapsedTime()).thenReturn(clock.getTime());
+        when(task.isEligibleForTracking()).thenReturn(true);
 
         AtomicBoolean isCancelled = new AtomicBoolean(false);
         doAnswer(invocation -> {