private void VerifyMaximumResourceCapability(Org.Apache.Hadoop.Yarn.Api.Records.Resource
expectedMaximumResource, AbstractYarnScheduler scheduler)
{
Org.Apache.Hadoop.Yarn.Api.Records.Resource schedulerMaximumResourceCapability =
scheduler.GetMaximumResourceCapability();
NUnit.Framework.Assert.AreEqual(expectedMaximumResource.GetMemory(), schedulerMaximumResourceCapability
.GetMemory());
NUnit.Framework.Assert.AreEqual(expectedMaximumResource.GetVirtualCores(), schedulerMaximumResourceCapability
.GetVirtualCores());
}
/// <exception cref="System.Exception"/>
private void TestMaximumAllocationMemoryHelper(AbstractYarnScheduler scheduler, int
node1MaxMemory, int node2MaxMemory, int node3MaxMemory, params int[] expectedMaxMemory
)
{
NUnit.Framework.Assert.AreEqual(6, expectedMaxMemory.Length);
NUnit.Framework.Assert.AreEqual(0, scheduler.GetNumClusterNodes());
int maxMemory = scheduler.GetMaximumResourceCapability().GetMemory();
NUnit.Framework.Assert.AreEqual(expectedMaxMemory[0], maxMemory);
RMNode node1 = MockNodes.NewNodeInfo(0, Resources.CreateResource(node1MaxMemory),
1, "127.0.0.2");
scheduler.Handle(new NodeAddedSchedulerEvent(node1));
NUnit.Framework.Assert.AreEqual(1, scheduler.GetNumClusterNodes());
maxMemory = scheduler.GetMaximumResourceCapability().GetMemory();
NUnit.Framework.Assert.AreEqual(expectedMaxMemory[1], maxMemory);
scheduler.Handle(new NodeRemovedSchedulerEvent(node1));
NUnit.Framework.Assert.AreEqual(0, scheduler.GetNumClusterNodes());
maxMemory = scheduler.GetMaximumResourceCapability().GetMemory();
NUnit.Framework.Assert.AreEqual(expectedMaxMemory[2], maxMemory);
RMNode node2 = MockNodes.NewNodeInfo(0, Resources.CreateResource(node2MaxMemory),
2, "127.0.0.3");
scheduler.Handle(new NodeAddedSchedulerEvent(node2));
NUnit.Framework.Assert.AreEqual(1, scheduler.GetNumClusterNodes());
maxMemory = scheduler.GetMaximumResourceCapability().GetMemory();
NUnit.Framework.Assert.AreEqual(expectedMaxMemory[3], maxMemory);
RMNode node3 = MockNodes.NewNodeInfo(0, Resources.CreateResource(node3MaxMemory),
3, "127.0.0.4");
scheduler.Handle(new NodeAddedSchedulerEvent(node3));
NUnit.Framework.Assert.AreEqual(2, scheduler.GetNumClusterNodes());
maxMemory = scheduler.GetMaximumResourceCapability().GetMemory();
NUnit.Framework.Assert.AreEqual(expectedMaxMemory[4], maxMemory);
scheduler.Handle(new NodeRemovedSchedulerEvent(node3));
NUnit.Framework.Assert.AreEqual(1, scheduler.GetNumClusterNodes());
maxMemory = scheduler.GetMaximumResourceCapability().GetMemory();
NUnit.Framework.Assert.AreEqual(expectedMaxMemory[5], maxMemory);
scheduler.Handle(new NodeRemovedSchedulerEvent(node2));
NUnit.Framework.Assert.AreEqual(0, scheduler.GetNumClusterNodes());
}
