@AfterClass
public void afterClass() throws IOException {
buffer.close();
deleteUsedFiles(FILE_COUNT);
storageLocal.delete();
}
private void fillFilesWithContent() throws InterruptedException, ExecutionException, IOException {
for (int i = 0; i < THREAD_COUNT; i++) {
futures.add(executorService.submit(new Writer()));
}
for (Future<Void> future : futures) future.get();
futures.clear();
buffer.flushBuffer();
}
@BeforeMethod
public void beforeMethod() throws IOException {
if (buffer != null) {
buffer.close();
deleteUsedFiles(FILE_COUNT);
}
initBuffer();
Random random = new Random();
seed = (byte) (random.nextInt() & 0xFF);
}
public void testAdd() throws Exception {
getIdentitiesOfFiles();
fillFilesWithContent();
validateFilesContent(version.byteValue());
version.compareAndSet(1, 2);
continuousWrite.compareAndSet(true, false);
generateRemainingPagesQueueForAllFiles();
executeConcurrentRandomReadAndWriteOperations();
buffer.flushBuffer();
validateFilesContent(version.byteValue());
}
private void getIdentitiesOfFiles() throws IOException {
for (int i = 0; i < fileIds.length(); i++) {
fileIds.set(i, buffer.openFile(fileNames[i]));
}
}