Esempio n. 1
0
        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
                                                ());
        }
Esempio n. 2
0
        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
                                                ());
        }