Skip to content

Commit e122bcf

Browse files
committed
Deduplicate ApiResponse test setup
1 parent 64343fe commit e122bcf

File tree

1 file changed

+14
-20
lines changed

1 file changed

+14
-20
lines changed

core/src/test/java/com/google/adk/sessions/VertexAiSessionServiceTest.java

Lines changed: 14 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)