|
@@ -82,6 +82,7 @@ import org.apache.hadoop.yarn.api.records.Container;
|
|
|
import org.apache.hadoop.yarn.api.records.ContainerId;
|
|
|
import org.apache.hadoop.yarn.api.records.NodeId;
|
|
|
import org.apache.hadoop.yarn.api.records.Resource;
|
|
|
+import org.apache.hadoop.yarn.conf.YarnConfiguration;
|
|
|
import org.apache.hadoop.yarn.event.Event;
|
|
|
import org.apache.hadoop.yarn.event.EventHandler;
|
|
|
import org.junit.Test;
|
|
@@ -197,8 +198,9 @@ public class TestTaskAttempt{
|
|
|
Configuration conf = new Configuration();
|
|
|
conf.setInt(MRJobConfig.MAP_MEMORY_MB, mapMemMb);
|
|
|
conf.setInt(MRJobConfig.REDUCE_MEMORY_MB, reduceMemMb);
|
|
|
- app.setClusterInfo(new ClusterInfo(Resource
|
|
|
- .newInstance(minContainerSize, 1), Resource.newInstance(10240, 1)));
|
|
|
+ conf.setInt(YarnConfiguration.RM_SCHEDULER_MINIMUM_ALLOCATION_MB,
|
|
|
+ minContainerSize);
|
|
|
+ app.setClusterInfo(new ClusterInfo(Resource.newInstance(10240, 1)));
|
|
|
|
|
|
Job job = app.submit(conf);
|
|
|
app.waitForState(job, JobState.RUNNING);
|
|
@@ -392,7 +394,6 @@ public class TestTaskAttempt{
|
|
|
ClusterInfo clusterInfo = mock(ClusterInfo.class);
|
|
|
Resource resource = mock(Resource.class);
|
|
|
when(appCtx.getClusterInfo()).thenReturn(clusterInfo);
|
|
|
- when(clusterInfo.getMinContainerCapability()).thenReturn(resource);
|
|
|
when(resource.getMemory()).thenReturn(1024);
|
|
|
|
|
|
TaskAttemptImpl taImpl =
|
|
@@ -450,7 +451,6 @@ public class TestTaskAttempt{
|
|
|
ClusterInfo clusterInfo = mock(ClusterInfo.class);
|
|
|
Resource resource = mock(Resource.class);
|
|
|
when(appCtx.getClusterInfo()).thenReturn(clusterInfo);
|
|
|
- when(clusterInfo.getMinContainerCapability()).thenReturn(resource);
|
|
|
when(resource.getMemory()).thenReturn(1024);
|
|
|
|
|
|
TaskAttemptImpl taImpl =
|
|
@@ -511,7 +511,6 @@ public class TestTaskAttempt{
|
|
|
ClusterInfo clusterInfo = mock(ClusterInfo.class);
|
|
|
Resource resource = mock(Resource.class);
|
|
|
when(appCtx.getClusterInfo()).thenReturn(clusterInfo);
|
|
|
- when(clusterInfo.getMinContainerCapability()).thenReturn(resource);
|
|
|
when(resource.getMemory()).thenReturn(1024);
|
|
|
|
|
|
TaskAttemptImpl taImpl =
|
|
@@ -579,7 +578,6 @@ public class TestTaskAttempt{
|
|
|
ClusterInfo clusterInfo = mock(ClusterInfo.class);
|
|
|
Resource resource = mock(Resource.class);
|
|
|
when(appCtx.getClusterInfo()).thenReturn(clusterInfo);
|
|
|
- when(clusterInfo.getMinContainerCapability()).thenReturn(resource);
|
|
|
when(resource.getMemory()).thenReturn(1024);
|
|
|
|
|
|
TaskAttemptImpl taImpl = new MapTaskAttemptImpl(taskId, 1, eventHandler,
|
|
@@ -629,7 +627,6 @@ public class TestTaskAttempt{
|
|
|
ClusterInfo clusterInfo = mock(ClusterInfo.class);
|
|
|
Resource resource = mock(Resource.class);
|
|
|
when(appCtx.getClusterInfo()).thenReturn(clusterInfo);
|
|
|
- when(clusterInfo.getMinContainerCapability()).thenReturn(resource);
|
|
|
when(resource.getMemory()).thenReturn(1024);
|
|
|
|
|
|
TaskAttemptImpl taImpl = new MapTaskAttemptImpl(taskId, 1, eventHandler,
|