fix TestUberAM Junit Error.

Author: slfan1989

hadoop-mapreduce-project/hadoop-mapreduce-client/hadoop-mapreduce-client-jobclient/src/test/java/org/apache/hadoop/mapreduce/v2/TestMRJobs.java

@@ -183,6 +183,41 @@ public static void setup() throws IOException {
     localFs.setPermission(APP_JAR, new FsPermission("700"));
   }
 
+  @BeforeAll
+  public static void setupUberAM() throws IOException {
+    try {
+        dfsCluster = new MiniDFSCluster.Builder(conf).numDataNodes(2)
+                .format(true).racks(null).build();
+        remoteFs = dfsCluster.getFileSystem();
+    } catch (IOException io) {
+        throw new RuntimeException("problem starting mini dfs cluster", io);
+    }
+
+    if (!(new File(MiniMRYarnCluster.APPJAR)).exists()) {
+        LOG.info("MRAppJar {} not found. Not running test.",
+            MiniMRYarnCluster.APPJAR);
+        return;
+    }
+
+    if (mrCluster == null) {
+        mrCluster = new MiniMRYarnCluster(TestMRJobs.class.getName(),
+            NUM_NODE_MGRS);
+        Configuration conf = new Configuration();
+        conf.set("fs.defaultFS", remoteFs.getUri().toString());   // use HDFS
+        conf.set(MRJobConfig.MR_AM_STAGING_DIR, "/apps_staging_dir");
+        conf.setInt(YarnConfiguration.MAX_CLUSTER_LEVEL_APPLICATION_PRIORITY, 10);
+        conf.setBoolean(MRJobConfig.JOB_UBERTASK_ENABLE, true);
+        conf.setInt(MRJobConfig.JOB_UBERTASK_MAXREDUCES, 3);
+        mrCluster.init(conf);
+        mrCluster.start();
+    }
+
+    // Copy MRAppJar and make it private. TODO: FIXME. This is a hack to
+    // workaround the absent public discache.
+    localFs.copyFromLocalFile(new Path(MiniMRYarnCluster.APPJAR), APP_JAR);
+    localFs.setPermission(APP_JAR, new FsPermission("700"));
+  }
+
   @AfterAll
   public static void tearDown() throws IOException {
     if (mrCluster != null) {

hadoop-mapreduce-project/hadoop-mapreduce-client/hadoop-mapreduce-client-jobclient/src/test/java/org/apache/hadoop/mapreduce/v2/TestUberAM.java

@@ -45,11 +45,7 @@ public class TestUberAM extends TestMRJobs {
 
   @BeforeAll
   public static void setup() throws IOException {
-    TestMRJobs.setup();
-    if (mrCluster != null) {
-    	mrCluster.getConfig().setBoolean(MRJobConfig.JOB_UBERTASK_ENABLE, true);
-    	mrCluster.getConfig().setInt(MRJobConfig.JOB_UBERTASK_MAXREDUCES, 3);
-    }
+    TestMRJobs.setupUberAM();
   }
 
   @Override