@Test
public void testTransferWindowFiles() throws IOException, InterruptedException {
testMeta.checkpointManager.setup(testMeta.managedStateContext);
Map<Long, Map<Slice, Bucket.BucketedValue>> buckets5 =
ManagedStateTestUtils.getTestData(0, 5, 0);
testMeta.checkpointManager.save(buckets5, 10, false);
// Need to synchronously call transfer window files so shutting down the other thread.
testMeta.checkpointManager.teardown();
Thread.sleep(500);
testMeta.checkpointManager.committed(10);
testMeta.checkpointManager.transferWindowFiles();
for (int i = 0; i < 5; i++) {
ManagedStateTestUtils.transferBucketHelper(
testMeta.managedStateContext.getFileAccess(), i, buckets5.get((long) i), 1);
}
}
@Test
public void testCommitted() throws IOException, InterruptedException {
CountDownLatch latch = new CountDownLatch(5);
MockBucketsFileSystem mockBucketsFileSystem = new MockBucketsFileSystem(latch);
testMeta.managedStateContext.setBucketsFileSystem(mockBucketsFileSystem);
mockBucketsFileSystem.setup(testMeta.managedStateContext);
testMeta.checkpointManager.setup(testMeta.managedStateContext);
Map<Long, Map<Slice, Bucket.BucketedValue>> data = ManagedStateTestUtils.getTestData(0, 5, 0);
testMeta.checkpointManager.save(data, 10, false);
testMeta.checkpointManager.committed(10);
latch.await();
testMeta.checkpointManager.teardown();
Thread.sleep(500);
for (int i = 0; i < 5; i++) {
ManagedStateTestUtils.transferBucketHelper(
testMeta.managedStateContext.getFileAccess(), i, data.get((long) i), 1);
}
}
