fix UT failures

Signed-off-by: Kaushal Kumar <[email protected]>
kaushalmahi12 · Jan 23, 2025 · 6e5f44d · 6e5f44d
1 parent 745bd3a
commit 6e5f44d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.../src/test/java/org/opensearch/wlm/tracker/ResourceUsageCalculatorTrackerServiceTests.java b/.../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 -> {