doReturn(1).when(outputContext).getTaskIndex();
doReturn(1).when(outputContext).getTaskVertexIndex();
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);