LogBucket bucket = storage.getNextBucket();
private void testAddHelper(int addedN, int bucketSize, int recordCount, int expectedN) { LogStorage storage = (LogStorage) getStorage(bucketSize, recordCount); List<LogRecord> expectedList = new LinkedList<>(); LogRecord record = new LogRecord(); while (addedN-- > 0) { storage.addLogRecord(record); } while (expectedN-- > 0) { expectedList.add(record); } LogBucket group = storage.getNextBucket(); List<LogRecord> actualList = group.getRecords(); Assert.assertTrue("Expected: " + expectedList.size() + ", actual: " + actualList.size() , expectedList.size() == actualList.size()); Iterator<LogRecord> expectedIt = expectedList.iterator(); Iterator<LogRecord> actualIt = actualList.iterator(); while (expectedIt.hasNext()) { LogRecord expected = expectedIt.next(); LogRecord actual = actualIt.next(); Assert.assertTrue(expected.getSize() == actual.getSize()); Assert.assertArrayEquals(expected.getData(), actual.getData()); } storage.close(); }
LogBucket removingBlock1 = storage.getNextBucket(); insertionCount -= removingBlock1.getRecords().size(); LogBucket removingBlock2 = storage.getNextBucket(); insertionCount -= removingBlock2.getRecords().size(); LogBucket removingBlock3 = storage.getNextBucket(); insertionCount -= removingBlock3.getRecords().size(); insertionCount += removingBlock1.getRecords().size(); LogBucket leftBlock1 = storage.getNextBucket(); LogBucket leftBlock2 = storage.getNextBucket(); int leftSize = leftBlock1.getRecords().size(); if (leftBlock2 != null) {
@Test public void testEmptyLogRecord() { long bucketSize = 3; int recordCount = 3; LogStorage storage = (LogStorage) getStorage(bucketSize, recordCount); LogBucket group = storage.getNextBucket(); Assert.assertTrue(group == null); storage.close(); }
@Test public void testLogRecordRemoval() { long bucketSize = 9; int recordCount = 3; LogStorage storage = (LogStorage) getStorage(bucketSize, recordCount); LogRecord record = new LogRecord(); int insertionCount = 7; /* * Size of each record is 3B */ int iter = insertionCount; while (iter-- > 0) { storage.addLogRecord(record); } LogBucket removingBlock = storage.getNextBucket(); insertionCount -= removingBlock.getRecords().size(); storage.removeBucket(removingBlock.getBucketId()); removingBlock = storage.getNextBucket(); insertionCount -= removingBlock.getRecords().size(); storage.removeBucket(removingBlock.getBucketId()); LogBucket leftBlock = storage.getNextBucket(); Assert.assertTrue(leftBlock.getRecords().size() == insertionCount); storage.close(); }
@Test public void testUniqueIdGeneration() { 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); } LogBucket group1 = storage.getNextBucket(); LogBucket group2 = storage.getNextBucket(); Assert.assertNotEquals(group1.getBucketId(), group2.getBucketId()); storage.close(); }
@Test public void testGetSameLogBlock() { 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); } LogBucket group1 = storage.getNextBucket(); storage.rollbackBucket(group1.getBucketId()); LogBucket group2 = storage.getNextBucket(); Assert.assertTrue("Expected: " + group1.getRecords().size() + ", actual: " + group2.getRecords().size() , group1.getRecords().size() == group2.getRecords().size()); Iterator<LogRecord> expectedIt = group1.getRecords().iterator(); Iterator<LogRecord> actualIt = group2.getRecords().iterator(); while (expectedIt.hasNext()) { LogRecord expected = expectedIt.next(); LogRecord actual = actualIt.next(); Assert.assertTrue(expected.getSize() == actual.getSize()); Assert.assertArrayEquals(expected.getData(), actual.getData()); } storage.close(); }
logRecords.add(new LogRecord()); LogBucket logBlock = new LogBucket(defaultId, logRecords); Mockito.when(storage.getNextBucket()).thenReturn(logBlock);
@Test public void testGetBigRecordBlock() { long bucketSize = 8192; int recordCount = 1000; LogStorage storage = (LogStorage) getStorage(bucketSize, recordCount); LogRecord record = new LogRecord(); int insertionCount = 7; /* * Size of each record is 3B */ int iter = insertionCount; while (iter-- > 0) { storage.addLogRecord(record); } LogBucket logBlock = storage.getNextBucket(); Assert.assertEquals(insertionCount, logBlock.getRecords().size()); storage.close(); }
@Test public void testPersistDBState() { LogStorage storage = (LogStorage) getStorage(BUCKET_SIZE, RECORD_COUNT); LogRecord record = new LogRecord(); int insertionCount = 7; /* * Size of each record is 3B */ int iter = insertionCount; while (iter-- > 0) { storage.addLogRecord(record); } LogBucket beforePersist = storage.getNextBucket(); storage.close(); storage = (LogStorage) getStorage(BUCKET_SIZE, RECORD_COUNT); LogStorageStatus storageStatus = (LogStorageStatus) storage; Assert.assertEquals(insertionCount, storageStatus.getRecordCount()); Assert.assertEquals(insertionCount * 3, storageStatus.getConsumedVolume()); LogBucket afterPersist = storage.getNextBucket(); Assert.assertEquals(beforePersist.getRecords().size(), afterPersist.getRecords().size()); storage.close(); }
logCollector.addLogRecord(record); Mockito.when(storage.getNextBucket()).thenReturn( new LogBucket(1, Arrays.asList(new LogRecord(record), new LogRecord(record), new LogRecord(record))));
logCollector.addLogRecord(record); Mockito.when(storage.getNextBucket()).thenReturn( new LogBucket(1, Arrays.asList(new LogRecord(record), new LogRecord(record), new LogRecord(record))));
LogBucket logBlock = storage.getNextBucket(); receivedCount = addIfNotEmpty(receivedCount, logBlock); Assert.assertEquals(insertionCount - receivedCount, storage.getStatus().getRecordCount()); Assert.assertEquals((insertionCount - receivedCount) * 3, storage.getStatus().getConsumedVolume()); logBlock = storage.getNextBucket(); receivedCount = addIfNotEmpty(receivedCount, logBlock); Assert.assertEquals(insertionCount - receivedCount, storage.getStatus().getRecordCount()); Assert.assertEquals((insertionCount - receivedCount) * 3, storage.getStatus().getConsumedVolume()); logBlock = storage.getNextBucket(); receivedCount = addIfNotEmpty(receivedCount, logBlock); Assert.assertEquals(insertionCount - receivedCount, storage.getStatus().getRecordCount());