Skip to content

Commit 74c8179

Browse files
committed
fix(net):add unit test for peer connection
1 parent 659d80a commit 74c8179

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

framework/src/test/java/org/tron/core/net/services/AdvServiceTest.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,7 @@ private void testBroadcast() {
8585

8686
try {
8787
peer = context.getBean(PeerConnection.class);
88+
Assert.assertFalse(peer.isDisconnect());
8889
p2pEventHandler = context.getBean(P2pEventHandlerImpl.class);
8990

9091
List<PeerConnection> peers = Lists.newArrayList();
@@ -96,6 +97,9 @@ private void testBroadcast() {
9697
service.broadcast(msg);
9798
Item item = new Item(blockCapsule.getBlockId(), InventoryType.BLOCK);
9899
Assert.assertNotNull(service.getMessage(item));
100+
peer.checkAndPutAdvInvRequest(item, System.currentTimeMillis());
101+
boolean res = peer.checkAndPutAdvInvRequest(item, System.currentTimeMillis());
102+
Assert.assertFalse(res);
99103
} catch (NullPointerException e) {
100104
System.out.println(e);
101105
}

0 commit comments

Comments
 (0)