private AllocateResponse CreateAllocateResponse(IList <ContainerStatus> completed, IList <Container> allocated, IList <NMToken> nmTokens) { AllocateResponse response = AllocateResponse.NewInstance(0, completed, allocated, new AList <NodeReport>(), null, null, 1, null, nmTokens); return(response); }
public virtual void TestAllocateResponseWithoutIncDecContainers() { AllocateResponse r = AllocateResponse.NewInstance(3, new AList <ContainerStatus>() , new AList <Container>(), new AList <NodeReport>(), null, AMCommand.AmResync, 3, null, new AList <NMToken>(), null, null); // serde YarnServiceProtos.AllocateResponseProto p = ((AllocateResponsePBImpl)r).GetProto( ); r = new AllocateResponsePBImpl(p); // check value NUnit.Framework.Assert.AreEqual(0, r.GetIncreasedContainers().Count); NUnit.Framework.Assert.AreEqual(0, r.GetDecreasedContainers().Count); }
/// <exception cref="Org.Apache.Hadoop.Yarn.Exceptions.YarnException"/> /// <exception cref="System.IO.IOException"/> public virtual AllocateResponse Allocate(AllocateRequest request) { NUnit.Framework.Assert.AreEqual("response ID mismatch", responseId, request.GetResponseId ()); ++responseId; Org.Apache.Hadoop.Yarn.Api.Records.Token yarnToken = null; if (amToken != null) { yarnToken = Org.Apache.Hadoop.Yarn.Api.Records.Token.NewInstance(amToken.GetIdentifier (), amToken.GetKind().ToString(), amToken.GetPassword(), amToken.GetService().ToString ()); } return(AllocateResponse.NewInstance(responseId, Collections.EmptyList <ContainerStatus >(), Collections.EmptyList <Container>(), Collections.EmptyList <NodeReport>(), Resources .None(), null, 1, null, Collections.EmptyList <NMToken>(), yarnToken, Collections .EmptyList <ContainerResourceIncrease>(), Collections.EmptyList <ContainerResourceDecrease >())); }
public virtual void TestAllocateResponseWithIncDecContainers() { IList <ContainerResourceIncrease> incContainers = new AList <ContainerResourceIncrease >(); IList <ContainerResourceDecrease> decContainers = new AList <ContainerResourceDecrease >(); for (int i = 0; i < 3; i++) { incContainers.AddItem(ContainerResourceIncrease.NewInstance(null, Resource.NewInstance (1024, i), null)); } for (int i_1 = 0; i_1 < 5; i_1++) { decContainers.AddItem(ContainerResourceDecrease.NewInstance(null, Resource.NewInstance (1024, i_1))); } AllocateResponse r = AllocateResponse.NewInstance(3, new AList <ContainerStatus>() , new AList <Container>(), new AList <NodeReport>(), null, AMCommand.AmResync, 3, null, new AList <NMToken>(), incContainers, decContainers); // serde YarnServiceProtos.AllocateResponseProto p = ((AllocateResponsePBImpl)r).GetProto( ); r = new AllocateResponsePBImpl(p); // check value NUnit.Framework.Assert.AreEqual(incContainers.Count, r.GetIncreasedContainers().Count ); NUnit.Framework.Assert.AreEqual(decContainers.Count, r.GetDecreasedContainers().Count ); for (int i_2 = 0; i_2 < incContainers.Count; i_2++) { NUnit.Framework.Assert.AreEqual(i_2, r.GetIncreasedContainers()[i_2].GetCapability ().GetVirtualCores()); } for (int i_3 = 0; i_3 < decContainers.Count; i_3++) { NUnit.Framework.Assert.AreEqual(i_3, r.GetDecreasedContainers()[i_3].GetCapability ().GetVirtualCores()); } }
public virtual AllocateResponse CreateFakeAllocateResponse() { return(AllocateResponse.NewInstance(-1, new AList <ContainerStatus>(), new AList <Container >(), new AList <NodeReport>(), Resource.NewInstance(1024, 2), null, 1, null, new AList <NMToken>())); }