Example usage for org.apache.hadoop.yarn.api.records NodeId NodeId

Introduction

In this page you can find the example usage for org.apache.hadoop.yarn.api.records NodeId NodeId.

Prototype

NodeId

Source Link

Usage

From source file:org.apache.tez.dag.app.launcher.TestDeletionTracker.java

License:Apache License

@Test
public void testNodeIdShufflePortMap() throws Exception {
    DeletionTrackerImpl deletionTracker = new DeletionTrackerImpl(new Configuration());
    // test NodeId
    NodeId nodeId = new NodeId() {
        @Override/* w  w  w. j av a2 s .  com*/
        public String getHost() {
            return "testHost";
        }

        @Override
        protected void setHost(String s) {

        }

        @Override
        public int getPort() {
            return 1234;
        }

        @Override
        protected void setPort(int i) {

        }

        @Override
        protected void build() {

        }
    };
    // test shuffle port for the nodeId
    int shufflePort = 9999;
    deletionTracker.addNodeShufflePort(nodeId, shufflePort);
    Assert.assertEquals("Unexpected number of entries in NodeIdShufflePortMap!", 1,
            deletionTracker.getNodeIdShufflePortMap().size());
    deletionTracker.addNodeShufflePort(nodeId, shufflePort);
    Assert.assertEquals("Unexpected number of entries in NodeIdShufflePortMap!", 1,
            deletionTracker.getNodeIdShufflePortMap().size());
    deletionTracker.dagComplete(new TezDAGID(), new JobTokenSecretManager());
    Assert.assertEquals("Unexpected number of entries in NodeIdShufflePortMap after dagComplete!", 1,
            deletionTracker.getNodeIdShufflePortMap().size());
}

---