|
@@ -26,7 +26,6 @@ import java.io.File;
|
|
|
import java.io.FileReader;
|
|
|
import java.io.IOException;
|
|
|
import java.io.PrintWriter;
|
|
|
-import java.util.ArrayList;
|
|
|
import java.util.Arrays;
|
|
|
import java.util.HashMap;
|
|
|
import java.util.List;
|
|
@@ -58,12 +57,8 @@ import org.apache.hadoop.yarn.exceptions.YarnRemoteException;
|
|
|
import org.apache.hadoop.yarn.server.api.ResourceManagerConstants;
|
|
|
import org.apache.hadoop.yarn.server.nodemanager.CMgrCompletedAppsEvent;
|
|
|
import org.apache.hadoop.yarn.server.nodemanager.ContainerExecutor.ExitCode;
|
|
|
-import org.apache.hadoop.yarn.server.nodemanager.ContainerExecutor.Signal;
|
|
|
import org.apache.hadoop.yarn.server.nodemanager.DefaultContainerExecutor;
|
|
|
import org.apache.hadoop.yarn.server.nodemanager.DeletionService;
|
|
|
-import org.apache.hadoop.yarn.server.nodemanager.LocalRMInterface;
|
|
|
-import org.apache.hadoop.yarn.server.nodemanager.NodeStatusUpdater;
|
|
|
-import org.apache.hadoop.yarn.server.nodemanager.NodeStatusUpdaterImpl;
|
|
|
import org.apache.hadoop.yarn.server.nodemanager.containermanager.application.ApplicationState;
|
|
|
import org.apache.hadoop.yarn.server.nodemanager.containermanager.localizer.ContainerLocalizer;
|
|
|
import org.apache.hadoop.yarn.server.nodemanager.containermanager.localizer.ResourceLocalizationService;
|
|
@@ -161,6 +156,7 @@ public class TestContainerManager extends BaseContainerManagerTest {
|
|
|
when(mockContainer.getNodeId()).thenReturn(context.getNodeId());
|
|
|
when(mockContainer.getNodeHttpAddress()).thenReturn(
|
|
|
context.getNodeId().getHost() + ":12345");
|
|
|
+ when(mockContainer.getRMIdentifer()).thenReturn(super.DUMMY_RM_IDENTIFIER);
|
|
|
|
|
|
StartContainerRequest startRequest =
|
|
|
recordFactory.newRecordInstance(StartContainerRequest.class);
|
|
@@ -259,6 +255,7 @@ public class TestContainerManager extends BaseContainerManagerTest {
|
|
|
when(mockContainer.getNodeId()).thenReturn(context.getNodeId());
|
|
|
when(mockContainer.getNodeHttpAddress()).thenReturn(
|
|
|
context.getNodeId().getHost() + ":12345");
|
|
|
+ when(mockContainer.getRMIdentifer()).thenReturn(super.DUMMY_RM_IDENTIFIER);
|
|
|
|
|
|
StartContainerRequest startRequest = recordFactory.newRecordInstance(StartContainerRequest.class);
|
|
|
startRequest.setContainerLaunchContext(containerLaunchContext);
|
|
@@ -370,6 +367,8 @@ public class TestContainerManager extends BaseContainerManagerTest {
|
|
|
when(mockContainer.getNodeId()).thenReturn(context.getNodeId());
|
|
|
when(mockContainer.getNodeHttpAddress()).thenReturn(
|
|
|
context.getNodeId().getHost() + ":12345");
|
|
|
+ when(mockContainer.getRMIdentifer()).thenReturn(super.DUMMY_RM_IDENTIFIER);
|
|
|
+
|
|
|
StartContainerRequest startRequest = recordFactory.newRecordInstance(StartContainerRequest.class);
|
|
|
startRequest.setContainerLaunchContext(containerLaunchContext);
|
|
|
startRequest.setContainer(mockContainer);
|
|
@@ -470,6 +469,7 @@ public class TestContainerManager extends BaseContainerManagerTest {
|
|
|
when(mockContainer.getNodeId()).thenReturn(context.getNodeId());
|
|
|
when(mockContainer.getNodeHttpAddress()).thenReturn(
|
|
|
context.getNodeId().getHost() + ":12345");
|
|
|
+ when(mockContainer.getRMIdentifer()).thenReturn(super.DUMMY_RM_IDENTIFIER);
|
|
|
|
|
|
// containerLaunchContext.command = new ArrayList<CharSequence>();
|
|
|
|
|
@@ -541,19 +541,6 @@ public class TestContainerManager extends BaseContainerManagerTest {
|
|
|
@Test
|
|
|
public void testContainerLaunchFromPreviousRM() throws IOException,
|
|
|
InterruptedException {
|
|
|
- // There is no real RM registration, simulate and set RMIdentifier
|
|
|
- NodeStatusUpdater nodeStatusUpdater = mock(NodeStatusUpdater.class);
|
|
|
- when(nodeStatusUpdater.getRMIdentifier()).thenReturn((long) 1234);
|
|
|
- containerManager =
|
|
|
- new ContainerManagerImpl(context, exec, delSrvc, nodeStatusUpdater,
|
|
|
- metrics, new ApplicationACLsManager(conf), dirsHandler) {
|
|
|
- @Override
|
|
|
- public void setBlockNewContainerRequests(
|
|
|
- boolean blockNewContainerRequests) {
|
|
|
- // do nothing
|
|
|
- }
|
|
|
- };
|
|
|
- containerManager.init(conf);
|
|
|
containerManager.start();
|
|
|
|
|
|
ContainerLaunchContext containerLaunchContext =
|
|
@@ -595,7 +582,8 @@ public class TestContainerManager extends BaseContainerManagerTest {
|
|
|
// Construct the Container with a RMIdentifier within current RM
|
|
|
Container mockContainer2 = mock(Container.class);
|
|
|
when(mockContainer2.getId()).thenReturn(cId2);
|
|
|
- when(mockContainer2.getRMIdentifer()).thenReturn((long) 1234);
|
|
|
+ when(mockContainer2.getRMIdentifer()).thenReturn(super.DUMMY_RM_IDENTIFIER);
|
|
|
+
|
|
|
when(mockContainer2.getResource()).thenReturn(mockResource);
|
|
|
StartContainerRequest startRequest2 =
|
|
|
recordFactory.newRecordInstance(StartContainerRequest.class);
|