Skip to content

Commit 468db6a

Browse files
add services and task types to fix RequestManagerTests
1 parent a3231a3 commit 468db6a

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/external/http/sender/RequestManagerTests.java

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
import org.apache.logging.log4j.Logger;
1111
import org.elasticsearch.action.ActionListener;
1212
import org.elasticsearch.inference.InferenceServiceResults;
13+
import org.elasticsearch.inference.TaskType;
1314
import org.elasticsearch.xpack.inference.external.http.retry.RequestSender;
1415
import org.elasticsearch.xpack.inference.external.http.retry.ResponseHandler;
1516
import org.elasticsearch.xpack.inference.external.request.RequestTests;
@@ -34,7 +35,10 @@ public static RequestManager createMock(RequestSender requestSender) {
3435
}
3536

3637
public static RequestManager createMock(RequestSender requestSender, String inferenceEntityId) {
37-
return createMock(requestSender, inferenceEntityId, new RateLimitSettings(1));
38+
var mockRequestSender = createMock(requestSender, inferenceEntityId, new RateLimitSettings(1));
39+
when(mockRequestSender.service()).thenReturn("test-service");
40+
when(mockRequestSender.taskType()).thenReturn(TaskType.ANY);
41+
return mockRequestSender;
3842
}
3943

4044
public static RequestManager createMock(RequestSender requestSender, String inferenceEntityId, RateLimitSettings settings) {

0 commit comments

Comments
 (0)