protected void uploadIfNeeded() { processUploadDecision(strategy.isUploadNeeded(storage.getStatus())); }
private void processUploadDecision(LogUploadStrategyDecision decision) { switch (decision) { case UPLOAD: if (isUploadAllowed()) { LOG.debug("Going to upload logs"); transport.sync(); } break; case NOOP: if (strategy.getUploadCheckPeriod() > 0 && storage.getStatus().getRecordCount() > 0) { scheduleUploadCheck(); } break; default: break; } }
processUploadDecision(strategy.isUploadNeeded(storage.getStatus()));
Assert.assertEquals(insertionCount - receivedCount, storage.getStatus().getRecordCount()); Assert.assertEquals((insertionCount - receivedCount) * 3, storage.getStatus().getConsumedVolume()); Assert.assertEquals(insertionCount - receivedCount, storage.getStatus().getRecordCount()); Assert.assertEquals((insertionCount - receivedCount) * 3, storage.getStatus().getConsumedVolume()); Assert.assertEquals(insertionCount - receivedCount, storage.getStatus().getRecordCount()); Assert.assertEquals((insertionCount - receivedCount) * 3, storage.getStatus().getConsumedVolume()); Assert.assertEquals(insertionCount - receivedCount, storage.getStatus().getRecordCount()); Assert.assertEquals((insertionCount - receivedCount) * 3, storage.getStatus().getConsumedVolume()); storage.close();
@Test public void testRecordCountAndConsumedBytes() { long bucketSize = 3; int recordCount = 3; LogStorage storage = (LogStorage) getStorage(bucketSize, recordCount); LogRecord record = new LogRecord(); int insertionCount = 3; /* * Size of each record is 3B */ int iter = insertionCount; while (iter-- > 0) { storage.addLogRecord(record); } Assert.assertTrue(storage.getStatus().getRecordCount() == insertionCount); Assert.assertTrue(storage.getStatus().getConsumedVolume() == (insertionCount * record.getSize())); storage.close(); }
logCollector.setStorage(storage); Mockito.when(storage.getStatus()).thenReturn(new LogStorageStatus() {
Mockito.when(storage.getStatus()).thenReturn(new LogStorageStatus() { @Override public long getRecordCount() { logCollector.addLogRecord(record); Mockito.when(storage.addLogRecord(Mockito.any(LogRecord.class))).thenReturn(new BucketInfo(1, 1)); Mockito.when(storage.getStatus()).thenReturn(new LogStorageStatus() {