Fix unit test

hms-dbmi · Jun 27, 2023 · 5a68f36 · 5a68f36
1 parent af5dbd4
commit 5a68f36
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pic-sure-api-war/src/test/java/edu/harvard/dbmi/avillach/PicsureQueryServiceTest.java b/pic-sure-api-war/src/test/java/edu/harvard/dbmi/avillach/PicsureQueryServiceTest.java
@@ -385,7 +385,7 @@ public void testQuerySyncValidNoResponseId() {
 
 		// Return mocks when needed
 		Response resp = mock(Response.class);
-		when(webClient.querySync(any(), any())).thenReturn(resp);
+		when(webClient.querySync(any(), any(), any())).thenReturn(resp);
 
 		// Mock persisting the queryentity, so that it has an ID and we can test that
 		// the correct information is stored in it
@@ -439,7 +439,7 @@ public void testQuerySyncValidWithResponseId() {
 		MultivaluedMap<String, Object> headerMap= new MultivaluedHashMap<String, Object>();
 		headerMap.add(ResourceWebClient.QUERY_METADATA_FIELD, resultId);
 		when(resp.getHeaders()).thenReturn(headerMap);
-		when(webClient.querySync(any(), any())).thenReturn(resp);
+		when(webClient.querySync(any(), any(), any())).thenReturn(resp);
 
 		// Mock persisting the queryentity, so that it has an ID and we can test that
 		// the correct information is stored in it