|
@@ -29,7 +29,6 @@ import org.apache.hadoop.conf.Configuration;
|
|
|
import org.apache.hadoop.fs.FileSystem;
|
|
|
import org.apache.hadoop.fs.FilterFileSystem;
|
|
|
import org.apache.hadoop.fs.Path;
|
|
|
-import org.apache.hadoop.mapred.InvalidJobConfException;
|
|
|
import org.apache.hadoop.mapreduce.Job;
|
|
|
import org.apache.hadoop.mapreduce.JobID;
|
|
|
import org.apache.hadoop.mapreduce.MRJobConfig;
|
|
@@ -38,7 +37,6 @@ import org.apache.hadoop.mapreduce.v2.api.records.JobId;
|
|
|
import org.apache.hadoop.mapreduce.v2.api.records.TaskAttemptId;
|
|
|
import org.apache.hadoop.mapreduce.v2.api.records.TaskId;
|
|
|
import org.apache.hadoop.mapreduce.v2.api.records.TaskType;
|
|
|
-import org.apache.hadoop.mapreduce.v2.util.MRApps;
|
|
|
import org.apache.hadoop.yarn.api.records.ApplicationId;
|
|
|
import org.apache.hadoop.yarn.api.records.LocalResource;
|
|
|
import org.apache.hadoop.yarn.api.records.LocalResourceType;
|
|
@@ -47,6 +45,7 @@ import org.apache.hadoop.yarn.factory.providers.RecordFactoryProvider;
|
|
|
import org.junit.AfterClass;
|
|
|
import org.junit.BeforeClass;
|
|
|
import org.junit.Test;
|
|
|
+
|
|
|
import static org.junit.Assert.*;
|
|
|
import static org.mockito.Mockito.*;
|
|
|
|
|
@@ -245,7 +244,7 @@ public class TestMRApps {
|
|
|
}
|
|
|
|
|
|
@SuppressWarnings("deprecation")
|
|
|
- @Test(expected = InvalidJobConfException.class)
|
|
|
+ @Test
|
|
|
public void testSetupDistributedCacheConflicts() throws Exception {
|
|
|
Configuration conf = new Configuration();
|
|
|
conf.setClass("fs.mockfs.impl", MockFileSystem.class, FileSystem.class);
|
|
@@ -273,10 +272,18 @@ public class TestMRApps {
|
|
|
Map<String, LocalResource> localResources =
|
|
|
new HashMap<String, LocalResource>();
|
|
|
MRApps.setupDistributedCache(conf, localResources);
|
|
|
+
|
|
|
+ assertEquals(1, localResources.size());
|
|
|
+ LocalResource lr = localResources.get("something");
|
|
|
+ //Archive wins
|
|
|
+ assertNotNull(lr);
|
|
|
+ assertEquals(10l, lr.getSize());
|
|
|
+ assertEquals(10l, lr.getTimestamp());
|
|
|
+ assertEquals(LocalResourceType.ARCHIVE, lr.getType());
|
|
|
}
|
|
|
|
|
|
@SuppressWarnings("deprecation")
|
|
|
- @Test(expected = InvalidJobConfException.class)
|
|
|
+ @Test
|
|
|
public void testSetupDistributedCacheConflictsFiles() throws Exception {
|
|
|
Configuration conf = new Configuration();
|
|
|
conf.setClass("fs.mockfs.impl", MockFileSystem.class, FileSystem.class);
|
|
@@ -301,6 +308,14 @@ public class TestMRApps {
|
|
|
Map<String, LocalResource> localResources =
|
|
|
new HashMap<String, LocalResource>();
|
|
|
MRApps.setupDistributedCache(conf, localResources);
|
|
|
+
|
|
|
+ assertEquals(1, localResources.size());
|
|
|
+ LocalResource lr = localResources.get("something");
|
|
|
+ //First one wins
|
|
|
+ assertNotNull(lr);
|
|
|
+ assertEquals(10l, lr.getSize());
|
|
|
+ assertEquals(10l, lr.getTimestamp());
|
|
|
+ assertEquals(LocalResourceType.FILE, lr.getType());
|
|
|
}
|
|
|
|
|
|
@SuppressWarnings("deprecation")
|