unit test

Author: chengtx01

framework/src/test/java/org/tron/common/overlay/discover/table/NodeEntryTest.java

@@ -0,0 +1,29 @@
+package org.tron.common.overlay.discover.table;
+
+import org.junit.Assert;
+import org.junit.Test;
+import org.tron.common.overlay.discover.node.Node;
+
+public class NodeEntryTest {
+
+    @Test
+    public void test() throws InterruptedException {
+        Node node1 = Node.instanceOf("127.0.0.1:10001");
+        NodeEntry nodeEntry = new NodeEntry(node1);
+        int distance = nodeEntry.getDistance();
+        Assert.assertEquals(-256, distance);
+
+        long lastModified = nodeEntry.getModified();
+        //System.out.println(lastModified);
+        Thread.sleep(1);
+        nodeEntry.touch();
+        long nowModified = nodeEntry.getModified();
+        //System.out.println(nowModified);
+        Assert.assertNotEquals(lastModified, nowModified);
+
+        Node node2 = Node.instanceOf("127.0.0.1:10002");
+        NodeEntry nodeEntry2 = new NodeEntry(node2);
+        boolean isDif = nodeEntry.equals(nodeEntry2);
+        Assert.assertTrue(isDif);
+    }
+}

framework/src/test/java/org/tron/common/overlay/discover/table/NodeTableTest.java

@@ -0,0 +1,37 @@
+package org.tron.common.overlay.discover.table;
+
+import org.junit.Assert;
+import org.junit.Test;
+import org.tron.common.overlay.discover.node.Node;
+
+import java.util.List;
+
+public class NodeTableTest {
+    @Test
+    public void test() {
+        Node node1 = Node.instanceOf("127.0.0.1:10002");
+        Node node2 = Node.instanceOf("127.0.0.1:10003");
+        Node node3 = Node.instanceOf("127.0.0.2:10004");
+        NodeTable table = new NodeTable(node1);
+        Node nodeTemp = table.getNode();
+        Assert.assertEquals(10002, nodeTemp.getPort());
+        Assert.assertEquals(0, table.getNodesCount());
+        Assert.assertEquals(0, table.getBucketsCount());
+
+        table.addNode(node2);
+        table.addNode(node3);
+        int bucketsCount = table.getBucketsCount();
+        int nodeCount = table.getNodesCount();
+        Assert.assertEquals(2, nodeCount);
+        Assert.assertEquals(2, bucketsCount);
+
+        boolean isExist = table.contains(node2);
+        table.touchNode(node2);
+        Assert.assertTrue(isExist);
+
+        byte[] targetId = Node.getNodeId();
+        List<Node> nodeList = table.getClosestNodes(targetId);
+        Assert.assertTrue(nodeList.isEmpty());
+        //Assert.assertTrue(true);
+    }
+}

framework/src/test/java/org/tron/common/overlay/discover/table/TimeComparatorTest.java

@@ -0,0 +1,21 @@
+package org.tron.common.overlay.discover.table;
+
+import org.junit.Assert;
+import org.junit.Test;
+import org.tron.common.overlay.discover.node.Node;
+
+public class TimeComparatorTest {
+
+    @Test
+    public void test() throws InterruptedException {
+        Node node1 = Node.instanceOf("127.0.0.1:10001");
+        NodeEntry ne1 = new NodeEntry(node1);
+        Thread.sleep(1);
+        Node node2 = Node.instanceOf("127.0.0.1:10002");
+        NodeEntry ne2 = new NodeEntry(node2);
+        TimeComparator tc = new TimeComparator();
+        int result = tc.compare(ne1, ne2);
+        Assert.assertEquals(1, result);
+
+    }
+}