private RequestLogs getRequestLogs(String request1Id) { LogQuery logQuery = new LogQuery().requestIds(Collections.singletonList(request1Id)); Iterator<RequestLogs> iterator = findLogLine(logQuery, 2); if (iterator == null || !iterator.hasNext()) { return null; } return iterator.next(); }
@Test @InSequence(20) public void testRequestIds() throws Exception { LogService service = LogServiceFactory.getLogService(); LogQuery logQuery = new LogQuery().requestIds(Arrays.asList(request1Id, request2Id)); Iterator<RequestLogs> iterator = service.fetch(logQuery).iterator(); assertEquals(request1Id, iterator.next().getRequestId()); assertEquals(request2Id, iterator.next().getRequestId()); assertFalse(iterator.hasNext()); logQuery = new LogQuery().requestIds(Arrays.asList(request2Id)); iterator = service.fetch(logQuery).iterator(); assertEquals(request2Id, iterator.next().getRequestId()); assertFalse(iterator.hasNext()); }
@Test @InSequence(20) public void testGetBatchSize() throws Exception { long size = 1; LogService service = LogServiceFactory.getLogService(); LogQuery logQuery = new LogQuery().requestIds(Arrays.asList(request1Id, request2Id)).batchSize((int)size); Iterator<RequestLogs> iterator = service.fetch(logQuery).iterator(); assertNotNull(iterator.next()); // TODO: renable when expected behavior is confirmed. // assertFalse("Batch size 1 so there should not be another log", iterator.hasNext()); long batchSize = logQuery.getBatchSize(); assertEquals(size, batchSize); }
assertLogQueryExecutes(new LogQuery().endTimeMillis(System.currentTimeMillis()), "testEndTimeMillis", exceptions); assertLogQueryExecutes(new LogQuery().endTimeUsec(1000L * System.currentTimeMillis()), "testEndTimeUsec", exceptions); assertLogQueryExecutes(new LogQuery().requestIds(Arrays.asList(getCurrentRequestId())), "testRequestIds", exceptions); assertLogQueryExecutes( new LogQuery()