@@ -185,6 +185,7 @@ public void testGetAuthorization_ReturnsAValidResponse() throws IOException {
185185 verifyNoMoreInteractions (logger );
186186 }
187187 }
188+
188189 @ SuppressWarnings ("unchecked" )
189190 public void testGetAuthorization_OnResponseCalledOnce () throws IllegalStateException {
190191 var senderFactory = HttpRequestSenderTests .createSenderFactory (threadPool , clientManager );
@@ -195,15 +196,15 @@ public void testGetAuthorization_OnResponseCalledOnce() throws IllegalStateExcep
195196 ActionListener <ElasticInferenceServiceAuthorization > listener = mock (ActionListener .class );
196197
197198 String responseJson = """
198- {
199- "models": [
200- {
201- "model_name": "model-a",
202- "task_types": ["embedding/text/sparse", "chat/completion"]
203- }
204- ]
205- }
206- """ ;
199+ {
200+ "models": [
201+ {
202+ "model_name": "model-a",
203+ "task_types": ["embedding/text/sparse", "chat/completion"]
204+ }
205+ ]
206+ }
207+ """ ;
207208
208209 webServer .enqueue (new MockResponse ().setResponseCode (200 ).setBody (responseJson ));
209210
@@ -214,45 +215,46 @@ public void testGetAuthorization_OnResponseCalledOnce() throws IllegalStateExcep
214215 verifyNoMoreInteractions (logger );
215216 }
216217
217- public void testGetAuthorization_InvalidResponse () throws IOException {
218- var senderFactory = HttpRequestSenderTests .createSenderFactory (threadPool , clientManager );
219- var eisGatewayUrl = getUrl (webServer );
220- var logger = mock (Logger .class );
221- var authHandler = new ElasticInferenceServiceAuthorizationHandler (eisGatewayUrl , threadPool , logger );
218+ public void testGetAuthorization_InvalidResponse () throws IOException {
219+ var senderFactory = HttpRequestSenderTests .createSenderFactory (threadPool , clientManager );
220+ var eisGatewayUrl = getUrl (webServer );
221+ var logger = mock (Logger .class );
222+ var authHandler = new ElasticInferenceServiceAuthorizationHandler (eisGatewayUrl , threadPool , logger );
222223
223- try (var sender = senderFactory .createSender ()) {
224- String responseJson = """
225- {
226- "completion": [
227- {
228- "result": "some result 1"
229- },
230- {
231- "result": "some result 2"
232- }
233- ]
234- }
235- """ ;
224+ try (var sender = senderFactory .createSender ()) {
225+ String responseJson = """
226+ {
227+ "completion": [
228+ {
229+ "result": "some result 1"
230+ },
231+ {
232+ "result": "some result 2"
233+ }
234+ ]
235+ }
236+ """ ;
236237
237- webServer .enqueue (new MockResponse ().setResponseCode (200 ).setBody (responseJson ));
238+ webServer .enqueue (new MockResponse ().setResponseCode (200 ).setBody (responseJson ));
238239
239- PlainActionFuture <ElasticInferenceServiceAuthorization > listener = new PlainActionFuture <>();
240- authHandler .getAuthorization (listener , sender );
241-
242- var authResponse = listener .actionGet (TIMEOUT );
243- assertTrue (authResponse .enabledTaskTypes ().isEmpty ());
244- assertFalse (authResponse .isEnabled ());
245-
246- var loggerArgsCaptor = ArgumentCaptor .forClass (String .class );
247- verify (logger ).warn (loggerArgsCaptor .capture ());
248- var message = loggerArgsCaptor .getValue ();
249- assertThat (
250- message ,
251- is (
252- "Failed to retrieve the authorization information from the Elastic Inference Service."
253- + " Received an invalid response type: InferenceServiceResults"
254- ));
255- }
240+ PlainActionFuture <ElasticInferenceServiceAuthorization > listener = new PlainActionFuture <>();
241+ authHandler .getAuthorization (listener , sender );
242+
243+ var authResponse = listener .actionGet (TIMEOUT );
244+ assertTrue (authResponse .enabledTaskTypes ().isEmpty ());
245+ assertFalse (authResponse .isEnabled ());
246+
247+ var loggerArgsCaptor = ArgumentCaptor .forClass (String .class );
248+ verify (logger ).warn (loggerArgsCaptor .capture ());
249+ var message = loggerArgsCaptor .getValue ();
250+ assertThat (
251+ message ,
252+ is (
253+ "Failed to retrieve the authorization information from the Elastic Inference Service."
254+ + " Received an invalid response type: InferenceServiceResults"
255+ )
256+ );
257+ }
256258
257259 }
258260
0 commit comments