1212import org .opensearch .action .admin .cluster .node .tasks .cancel .CancelTasksRequest ;
1313import org .opensearch .action .search .SearchRequest ;
1414import org .opensearch .action .support .ActionFilterChain ;
15+ import org .opensearch .action .support .ActionRequestMetadata ;
1516import org .opensearch .common .util .concurrent .ThreadContext ;
1617import org .opensearch .core .action .ActionListener ;
1718import org .opensearch .core .action .ActionResponse ;
@@ -70,7 +71,7 @@ public void testApplyForValidRequest() {
7071 when (request .indices ()).thenReturn (new String [] { "foo" });
7172 try (ThreadContext .StoredContext context = threadPool .getThreadContext ().stashContext ()) {
7273 when (ruleProcessingService .evaluateLabel (anyList ())).thenReturn (Optional .of ("TestQG_ID" ));
73- autoTaggingActionFilter .apply (mock (Task .class ), "Test" , request , null , mockFilterChain );
74+ autoTaggingActionFilter .apply (mock (Task .class ), "Test" , request , ActionRequestMetadata . empty (), null , mockFilterChain );
7475
7576 assertEquals ("TestQG_ID" , threadPool .getThreadContext ().getHeader (WorkloadGroupTask .WORKLOAD_GROUP_ID_HEADER ));
7677 verify (ruleProcessingService , times (1 )).evaluateLabel (anyList ());
@@ -80,7 +81,7 @@ public void testApplyForValidRequest() {
8081 public void testApplyForInValidRequest () {
8182 ActionFilterChain <ActionRequest , ActionResponse > mockFilterChain = mock (TestActionFilterChain .class );
8283 CancelTasksRequest request = new CancelTasksRequest ();
83- autoTaggingActionFilter .apply (mock (Task .class ), "Test" , request , null , mockFilterChain );
84+ autoTaggingActionFilter .apply (mock (Task .class ), "Test" , request , ActionRequestMetadata . empty (), null , mockFilterChain );
8485
8586 verify (ruleProcessingService , times (0 )).evaluateLabel (anyList ());
8687 }
0 commit comments