|
@@ -837,6 +837,9 @@ public class TestShuffleHandler {
|
|
|
Configuration conf = new Configuration();
|
|
|
conf.setInt(ShuffleHandler.SHUFFLE_PORT_CONFIG_KEY, 0);
|
|
|
conf.setInt(ShuffleHandler.MAX_SHUFFLE_CONNECTIONS, 3);
|
|
|
+ conf.set(CommonConfigurationKeysPublic.HADOOP_SECURITY_AUTHENTICATION,
|
|
|
+ "simple");
|
|
|
+ UserGroupInformation.setConfiguration(conf);
|
|
|
File absLogDir = new File("target", TestShuffleHandler.class.
|
|
|
getSimpleName() + "LocDir").getAbsoluteFile();
|
|
|
conf.set(YarnConfiguration.NM_LOCAL_DIRS, absLogDir.getAbsolutePath());
|
|
@@ -924,7 +927,8 @@ public class TestShuffleHandler {
|
|
|
} catch (EOFException e) {
|
|
|
// ignore
|
|
|
}
|
|
|
- Assert.assertEquals(failures.size(), 0);
|
|
|
+ Assert.assertEquals("sendError called due to shuffle error",
|
|
|
+ 0, failures.size());
|
|
|
} finally {
|
|
|
shuffleHandler.stop();
|
|
|
FileUtil.fullyDelete(absLogDir);
|