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()); }