fix UT to remove the verify for final method

Signed-off-by: Kaushal Kumar <[email protected]>

server/src/test/java/org/opensearch/wlm/WorkloadManagementTransportRequestHandlerTests.java

@@ -49,8 +49,7 @@ public void testMessageReceivedForSearchWorkload() throws Exception {
         QueryGroupTask spyTask = getSpyTask();
 
         sut.messageReceived(request, mock(TransportChannel.class), spyTask);
-
-        verify(spyTask, times(1)).setQueryGroupId(threadPool.getThreadContext());
+        assertTrue(sut.isSearchWorkloadRequest(spyTask));
     }
 
     public void testMessageReceivedForNonSearchWorkload() throws Exception {