ScheduledTask task = new ScheduledTask()
.setStatus(RUNNING)
.setAssignedTask(
new AssignedTask()
.setTask(TASK_CONFIG.newBuilder()));
Response response = new Response()
.setResponseCode(OK)
.setResult(Result.scheduleStatusResult(new ScheduleStatusResult()
.setTasks(ImmutableList.of(task))));
TaskQuery query = new TaskQuery()
.setStatuses(ImmutableSet.of(RUNNING))
.setTaskIds(ImmutableSet.of("a"));
expect(thrift.getTasksStatus(query)).andReturn(response);
replayAndStart();
Response actualResponse = getRequestBuilder("/apibeta/getTasksStatus")
.entity(ImmutableMap.of("query", query), MediaType.APPLICATION_JSON)
.post(Response.class);
assertEquals(response, actualResponse);
}