Apache
/
druid
mirror of https://github.com/apache/druid.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix JsonParserIteratorTest (#13083)

This commit is contained in:
Clint Wylie 2022-09-13 20:49:57 -07:00 committed by GitHub
parent a8fd3a9077
commit f4ec50bf7a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
server/src/test/java/org/apache/druid/client/JsonParserIteratorTest.java
View File

@ -30,6 +30,7 @@ import org.apache.druid.java.util.common.StringUtils;
import org.apache.druid.java.util.common.concurrent.Execs;
import org.apache.druid.query.Query;
import org.apache.druid.query.QueryCapacityExceededException;
import org.apache.druid.query.QueryContext;
import org.apache.druid.query.QueryException;
import org.apache.druid.query.QueryInterruptedException;
import org.apache.druid.query.QueryTimeoutException;
@ -306,10 +307,15 @@ public class JsonParserIteratorTest
private Query<?> mockQuery(String queryId, long timeoutAt)
{
Query<?> query = Mockito.mock(Query.class);
QueryContext context = Mockito.mock(QueryContext.class);
Mockito.when(query.getId()).thenReturn(queryId);
Mockito.when(query.getQueryContext().getAsLong(ArgumentMatchers.eq(DirectDruidClient.QUERY_FAIL_TIME),
ArgumentMatchers.eq(-1L)))
.thenReturn(timeoutAt);
Mockito.when(query.getQueryContext()).thenReturn(context);
Mockito.when(
context.getAsLong(
ArgumentMatchers.eq(DirectDruidClient.QUERY_FAIL_TIME),
ArgumentMatchers.eq(-1L)
)
).thenReturn(timeoutAt);
return query;
}
}