@@ -42,6 +42,18 @@ public class VertexAiSessionServiceTest {
4242 private static final MediaType JSON_MEDIA_TYPE =
4343 MediaType .parse ("application/json; charset=utf-8" );
4444
45+ private static ApiResponse apiResponseJson (String json ) {
46+ return new ApiResponse () {
47+ @ Override
48+ public ResponseBody getResponseBody () {
49+ return ResponseBody .create (JSON_MEDIA_TYPE , json );
50+ }
51+
52+ @ Override
53+ public void close () {}
54+ };
55+ }
56+
4557 private static final String MOCK_SESSION_STRING_1 =
4658 """
4759 {
@@ -329,16 +341,7 @@ public void listSessions_empty() {
329341 @ Test
330342 public void listSessions_missingSessionsField_returnsEmpty () {
331343 when (mockApiClient .request ("GET" , "reasoningEngines/123/sessions?filter=user_id=userX" , "" ))
332- .thenReturn (
333- new ApiResponse () {
334- @ Override
335- public ResponseBody getResponseBody () {
336- return ResponseBody .create (JSON_MEDIA_TYPE , "{}" );
337- }
338-
339- @ Override
340- public void close () {}
341- });
344+ .thenReturn (apiResponseJson ("{}" ));
342345
343346 assertThat (vertexAiSessionService .listSessions ("123" , "userX" ).blockingGet ().sessions ())
344347 .isEmpty ();
@@ -347,16 +350,7 @@ public void close() {}
347350 @ Test
348351 public void listSessions_nullSessionsField_returnsEmpty () {
349352 when (mockApiClient .request ("GET" , "reasoningEngines/123/sessions?filter=user_id=userY" , "" ))
350- .thenReturn (
351- new ApiResponse () {
352- @ Override
353- public ResponseBody getResponseBody () {
354- return ResponseBody .create (JSON_MEDIA_TYPE , "{\" sessions\" : null}" );
355- }
356-
357- @ Override
358- public void close () {}
359- });
353+ .thenReturn (apiResponseJson ("{\" sessions\" : null}" ));
360354
361355 assertThat (vertexAiSessionService .listSessions ("123" , "userY" ).blockingGet ().sessions ())
362356 .isEmpty ();
0 commit comments