doReturn("vertexName").when(outputContext).getTaskVertexName();
doReturn(uniqueId).when(outputContext).getUniqueIdentifier();
ByteBuffer portBuffer = ByteBuffer.allocate(4);
portBuffer.mark();
portBuffer.putInt(SHUFFLE_PORT);
portBuffer.reset();
doReturn(portBuffer).when(outputContext).getServiceProviderMetaData(
eq(ShuffleUtils.SHUFFLE_HANDLER_SERVICE_ID));
Path outDirBase = new Path(TEST_ROOT_DIR, "outDir_" + uniqueId);
String[] outDirs = new String[] { outDirBase.toString() };
doReturn(outDirs).when(outputContext).getWorkDirs();