Skip to content

Commit 880c96f

Browse files
committed
Fix test
Signed-off-by: Craig Perkins <[email protected]>
1 parent ffe3da2 commit 880c96f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

plugin/src/test/java/org/opensearch/ml/action/models/SearchModelTransportActionTests.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -370,7 +370,7 @@ public void test_RSC_featureEnabled_typeEnabled_callsGetAccessibleIds() throws E
370370
// Simulate async success with some accessible IDs
371371
l.onResponse(Set.of("model_group_IT"));
372372
return null;
373-
}).when(rsc).getAccessibleResourceIds(eq(CommonValue.ML_MODEL_GROUP_INDEX), rscListenerCaptor.capture());
373+
}).when(rsc).getAccessibleResourceIds(eq(CommonValue.ML_MODEL_GROUP_RESOURCE_TYPE), rscListenerCaptor.capture());
374374

375375
// The final remote search goes through client.search(...); return a normal response
376376
doAnswer(inv -> {
@@ -385,7 +385,7 @@ public void test_RSC_featureEnabled_typeEnabled_callsGetAccessibleIds() throws E
385385
searchModelTransportAction.doExecute(null, mlSearchActionRequest, actionListener);
386386

387387
// Verify RSC path was taken
388-
verify(rsc, times(1)).getAccessibleResourceIds(eq(CommonValue.ML_MODEL_GROUP_INDEX), any());
388+
verify(rsc, times(1)).getAccessibleResourceIds(eq(CommonValue.ML_MODEL_GROUP_RESOURCE_TYPE), any());
389389

390390
// Verify we executed the final search and returned
391391
verify(client, times(1)).search(any(), any());

0 commit comments

Comments
 (0)