|
@@ -229,19 +229,16 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
public void testStartMultipleContainers() throws Exception {
|
|
|
containerManager.start();
|
|
|
|
|
|
- ContainerLaunchContext containerLaunchContext =
|
|
|
- recordFactory.newRecordInstance(ContainerLaunchContext.class);
|
|
|
-
|
|
|
List<StartContainerRequest> list = new ArrayList<>();
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(0), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(1024, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.GUARANTEED)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(1), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(1024, 1),
|
|
@@ -285,19 +282,16 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
public void testQueueMultipleContainers() throws Exception {
|
|
|
containerManager.start();
|
|
|
|
|
|
- ContainerLaunchContext containerLaunchContext =
|
|
|
- recordFactory.newRecordInstance(ContainerLaunchContext.class);
|
|
|
-
|
|
|
List<StartContainerRequest> list = new ArrayList<>();
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(0), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(3072, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.GUARANTEED)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(1), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(3072, 1),
|
|
@@ -343,26 +337,23 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
public void testStartAndQueueMultipleContainers() throws Exception {
|
|
|
containerManager.start();
|
|
|
|
|
|
- ContainerLaunchContext containerLaunchContext =
|
|
|
- recordFactory.newRecordInstance(ContainerLaunchContext.class);
|
|
|
-
|
|
|
List<StartContainerRequest> list = new ArrayList<>();
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(0), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(2048, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.OPPORTUNISTIC)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(1), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(1024, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.OPPORTUNISTIC)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(2), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(1024, 1),
|
|
@@ -415,12 +406,9 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
public void testStartOpportunistcsWhenOppQueueIsFull() throws Exception {
|
|
|
containerManager.start();
|
|
|
|
|
|
- ContainerLaunchContext containerLaunchContext =
|
|
|
- recordFactory.newRecordInstance(ContainerLaunchContext.class);
|
|
|
-
|
|
|
List<StartContainerRequest> list = new ArrayList<>();
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(0), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(2048, 1),
|
|
@@ -432,7 +420,7 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
YarnConfiguration.DEFAULT_NM_OPPORTUNISTIC_CONTAINERS_MAX_QUEUE_LENGTH);
|
|
|
for (int i = 1; i < maxOppQueueLength + 2; i++) {
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(i), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(2048, 1),
|
|
@@ -500,26 +488,23 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
public void testKillOpportunisticForGuaranteedContainer() throws Exception {
|
|
|
containerManager.start();
|
|
|
|
|
|
- ContainerLaunchContext containerLaunchContext =
|
|
|
- recordFactory.newRecordInstance(ContainerLaunchContext.class);
|
|
|
-
|
|
|
List<StartContainerRequest> list = new ArrayList<>();
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(0), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(2048, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.OPPORTUNISTIC)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(1), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(2048, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.OPPORTUNISTIC)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(2), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(2048, 1),
|
|
@@ -589,12 +574,10 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
Listener listener = new Listener();
|
|
|
((NodeManager.DefaultContainerStateListener)containerManager.getContext().
|
|
|
getContainerStateTransitionListener()).addListener(listener);
|
|
|
- ContainerLaunchContext containerLaunchContext =
|
|
|
- recordFactory.newRecordInstance(ContainerLaunchContext.class);
|
|
|
|
|
|
List<StartContainerRequest> list = new ArrayList<>();
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(0), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(2048, 1),
|
|
@@ -610,7 +593,7 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
|
|
|
list = new ArrayList<>();
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(1), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(2048, 1),
|
|
@@ -718,42 +701,42 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
|
|
|
list = new ArrayList<>();
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(1), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(512, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.OPPORTUNISTIC)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(2), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(512, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.OPPORTUNISTIC)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(3), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(512, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.OPPORTUNISTIC)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(4), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(512, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.OPPORTUNISTIC)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(5), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(512, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.OPPORTUNISTIC)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(6), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(512, 1),
|
|
@@ -840,14 +823,14 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
|
|
|
list = new ArrayList<>();
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(1), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(512, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.OPPORTUNISTIC)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(2), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(512, 1),
|
|
@@ -887,26 +870,23 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
public void testKillMultipleOpportunisticContainers() throws Exception {
|
|
|
containerManager.start();
|
|
|
|
|
|
- ContainerLaunchContext containerLaunchContext =
|
|
|
- recordFactory.newRecordInstance(ContainerLaunchContext.class);
|
|
|
-
|
|
|
List<StartContainerRequest> list = new ArrayList<>();
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(0), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(512, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.OPPORTUNISTIC)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(1), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(512, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.OPPORTUNISTIC)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(2), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(512, 1),
|
|
@@ -919,7 +899,7 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
|
|
|
list = new ArrayList<>();
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(3), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(1500, 1),
|
|
@@ -967,14 +947,11 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
public void testKillOnlyRequiredOpportunisticContainers() throws Exception {
|
|
|
containerManager.start();
|
|
|
|
|
|
- ContainerLaunchContext containerLaunchContext =
|
|
|
- recordFactory.newRecordInstance(ContainerLaunchContext.class);
|
|
|
-
|
|
|
List<StartContainerRequest> list = new ArrayList<>();
|
|
|
// Fill NM with Opportunistic containers
|
|
|
for (int i = 0; i < 4; i++) {
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(i), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(512, 1),
|
|
@@ -990,7 +967,7 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
// Now ask for two Guaranteed containers
|
|
|
for (int i = 4; i < 6; i++) {
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(i), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(512, 1),
|
|
@@ -1036,26 +1013,23 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
public void testStopQueuedContainer() throws Exception {
|
|
|
containerManager.start();
|
|
|
|
|
|
- ContainerLaunchContext containerLaunchContext =
|
|
|
- recordFactory.newRecordInstance(ContainerLaunchContext.class);
|
|
|
-
|
|
|
List<StartContainerRequest> list = new ArrayList<>();
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(0), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(2048, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.GUARANTEED)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(1), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(512, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.OPPORTUNISTIC)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(2), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(512, 1),
|
|
@@ -1142,19 +1116,16 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
((NodeManager.DefaultContainerStateListener)containerManager.getContext().
|
|
|
getContainerStateTransitionListener()).addListener(listener);
|
|
|
|
|
|
- ContainerLaunchContext containerLaunchContext =
|
|
|
- recordFactory.newRecordInstance(ContainerLaunchContext.class);
|
|
|
-
|
|
|
List<StartContainerRequest> list = new ArrayList<>();
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(0), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(2048, 1),
|
|
|
context.getContainerTokenSecretManager(), null,
|
|
|
ExecutionType.OPPORTUNISTIC)));
|
|
|
list.add(StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(createContainerId(1), DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(),
|
|
|
user, BuilderUtils.newResource(1024, 1),
|
|
@@ -1265,12 +1236,9 @@ public class TestContainerSchedulerQueuing extends BaseContainerManagerTest {
|
|
|
containerManager.start();
|
|
|
// Construct the Container-id
|
|
|
ContainerId cId = createContainerId(0);
|
|
|
- ContainerLaunchContext containerLaunchContext =
|
|
|
- recordFactory.newRecordInstance(ContainerLaunchContext.class);
|
|
|
-
|
|
|
StartContainerRequest scRequest =
|
|
|
StartContainerRequest.newInstance(
|
|
|
- containerLaunchContext,
|
|
|
+ recordFactory.newRecordInstance(ContainerLaunchContext.class),
|
|
|
createContainerToken(cId, DUMMY_RM_IDENTIFIER,
|
|
|
context.getNodeId(), user, BuilderUtils.newResource(512, 1),
|
|
|
context.getContainerTokenSecretManager(), null));
|