Skip to content

Commit c06b372

Browse files
committed
add test case
1 parent 6b47ef7 commit c06b372

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

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

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
import org.tron.common.utils.ReflectUtils;
2626
import org.tron.common.utils.Sha256Hash;
2727
import org.tron.core.ChainBaseManager;
28+
import org.tron.core.ChainBaseManager.NodeType;
2829
import org.tron.core.Constant;
2930
import org.tron.core.capsule.BlockCapsule;
3031
import org.tron.core.config.DefaultConfig;
@@ -256,6 +257,15 @@ public void testLowAndGenesisBlockNum() throws NoSuchMethodException {
256257
.setNumber(sid.getNum())
257258
.build();
258259
builder.setSolidBlockId(sBlockId);
260+
ChainBaseManager.getChainBaseManager().setNodeType(NodeType.FULL);
261+
try {
262+
HelloMessage helloMessage = new HelloMessage(builder.build().toByteArray());
263+
method.invoke(p2pEventHandler, peer, helloMessage.getSendBytes());
264+
} catch (Exception e) {
265+
Assert.fail();
266+
}
267+
268+
ChainBaseManager.getChainBaseManager().setNodeType(NodeType.LITE);
259269
try {
260270
HelloMessage helloMessage = new HelloMessage(builder.build().toByteArray());
261271
method.invoke(p2pEventHandler, peer, helloMessage.getSendBytes());

0 commit comments

Comments
 (0)