|
@@ -38,6 +38,7 @@ import org.apache.hadoop.mapreduce.v2.api.records.TaskType;
|
|
|
import org.apache.hadoop.yarn.api.records.ApplicationId;
|
|
|
import org.apache.hadoop.yarn.api.records.LocalResource;
|
|
|
import org.apache.hadoop.yarn.api.records.LocalResourceType;
|
|
|
+import org.apache.hadoop.yarn.conf.YarnConfiguration;
|
|
|
import org.apache.hadoop.yarn.factory.providers.RecordFactoryProvider;
|
|
|
import org.junit.Test;
|
|
|
|
|
@@ -136,16 +137,12 @@ public class TestMRApps {
|
|
|
Job job = Job.getInstance();
|
|
|
Map<String, String> environment = new HashMap<String, String>();
|
|
|
MRApps.setClasspath(environment, job.getConfiguration());
|
|
|
- assertEquals("$PWD:$HADOOP_CONF_DIR:" +
|
|
|
- "$HADOOP_COMMON_HOME/share/hadoop/common/*:" +
|
|
|
- "$HADOOP_COMMON_HOME/share/hadoop/common/lib/*:" +
|
|
|
- "$HADOOP_HDFS_HOME/share/hadoop/hdfs/*:" +
|
|
|
- "$HADOOP_HDFS_HOME/share/hadoop/hdfs/lib/*:" +
|
|
|
- "$YARN_HOME/share/hadoop/yarn/*:" +
|
|
|
- "$YARN_HOME/share/hadoop/mapreduce/*:" +
|
|
|
- "$YARN_HOME/share/hadoop/mapreduce/lib/*:" +
|
|
|
- "job.jar:$PWD/*",
|
|
|
- environment.get("CLASSPATH"));
|
|
|
+ assertTrue(environment.get("CLASSPATH").startsWith("$PWD:"));
|
|
|
+ String confClasspath = job.getConfiguration().get(YarnConfiguration.YARN_APPLICATION_CLASSPATH);
|
|
|
+ if (confClasspath != null) {
|
|
|
+ confClasspath = confClasspath.replaceAll(",\\s*", ":").trim();
|
|
|
+ }
|
|
|
+ assertTrue(environment.get("CLASSPATH").contains(confClasspath));
|
|
|
}
|
|
|
|
|
|
@Test public void testSetClasspathWithUserPrecendence() {
|