Merge pull request tronprotocol#2709 from chengtx01/new_test

unit test about network discover table

Author: xxo1shine

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 java.util.List;
+import org.junit.Assert;
+import org.junit.Test;
+import org.tron.common.overlay.discover.node.Node;
+
+public class NodeTableTest {
+  @Test
+  public void test() {
+    Node node1 = Node.instanceOf("127.0.0.1:10002");
+
+    NodeTable table = new NodeTable(node1);
+    Node nodeTemp = table.getNode();
+    Assert.assertEquals(10002, nodeTemp.getPort());
+    Assert.assertEquals(0, table.getNodesCount());
+    Assert.assertEquals(0, table.getBucketsCount());
+
+    Node node2 = Node.instanceOf("127.0.0.1:10003");
+    Node node3 = Node.instanceOf("127.0.0.2:10004");
+    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);
+
+  }
+}