public virtual void TestNodeHeartBeatResponse() { NodeHeartbeatResponse record = Org.Apache.Hadoop.Yarn.Util.Records.NewRecord <NodeHeartbeatResponse >(); IDictionary <ApplicationId, ByteBuffer> appCredentials = new Dictionary <ApplicationId , ByteBuffer>(); Credentials app1Cred = new Credentials(); Org.Apache.Hadoop.Security.Token.Token <DelegationTokenIdentifier> token1 = new Org.Apache.Hadoop.Security.Token.Token <DelegationTokenIdentifier>(); token1.SetKind(new Text("kind1")); app1Cred.AddToken(new Text("token1"), token1); Org.Apache.Hadoop.Security.Token.Token <DelegationTokenIdentifier> token2 = new Org.Apache.Hadoop.Security.Token.Token <DelegationTokenIdentifier>(); token2.SetKind(new Text("kind2")); app1Cred.AddToken(new Text("token2"), token2); DataOutputBuffer dob = new DataOutputBuffer(); app1Cred.WriteTokenStorageToStream(dob); ByteBuffer byteBuffer1 = ByteBuffer.Wrap(dob.GetData(), 0, dob.GetLength()); appCredentials[ApplicationId.NewInstance(1234, 1)] = byteBuffer1; record.SetSystemCredentialsForApps(appCredentials); NodeHeartbeatResponse proto = new NodeHeartbeatResponsePBImpl(((NodeHeartbeatResponsePBImpl )record).GetProto()); NUnit.Framework.Assert.AreEqual(appCredentials, proto.GetSystemCredentialsForApps ()); }
public virtual void TestNodeHeartbeatResponsePBImpl() { NodeHeartbeatResponsePBImpl original = new NodeHeartbeatResponsePBImpl(); original.SetDiagnosticsMessage("testDiagnosticMessage"); original.SetContainerTokenMasterKey(GetMasterKey()); original.SetNMTokenMasterKey(GetMasterKey()); original.SetNextHeartBeatInterval(1000); original.SetNodeAction(NodeAction.Normal); original.SetResponseId(100); NodeHeartbeatResponsePBImpl copy = new NodeHeartbeatResponsePBImpl(original.GetProto ()); NUnit.Framework.Assert.AreEqual(100, copy.GetResponseId()); NUnit.Framework.Assert.AreEqual(NodeAction.Normal, copy.GetNodeAction()); NUnit.Framework.Assert.AreEqual(1000, copy.GetNextHeartBeatInterval()); NUnit.Framework.Assert.AreEqual(1, copy.GetContainerTokenMasterKey().GetKeyId()); NUnit.Framework.Assert.AreEqual(1, copy.GetNMTokenMasterKey().GetKeyId()); NUnit.Framework.Assert.AreEqual("testDiagnosticMessage", copy.GetDiagnosticsMessage ()); }