public void noCleanupAndCleanup() throws IOException {
TestShuffleDataContext dataContext = createSomeData();
ExternalShuffleBlockResolver resolver =
new ExternalShuffleBlockResolver(conf, null, sameThreadExecutor);
resolver.registerExecutor("app", "exec0", dataContext.createExecutorInfo("shuffleMgr"));
resolver.applicationRemoved("app", false /* cleanup */);
assertStillThere(dataContext);
resolver.registerExecutor("app", "exec1", dataContext.createExecutorInfo("shuffleMgr"));
resolver.applicationRemoved("app", true /* cleanup */);
assertCleanedUp(dataContext);
}
|