Skip to content

Commit 75fba7a

Browse files
committed
fix(net): stable test case; add parameters in config
1 parent 74b34bc commit 75fba7a

8 files changed

Lines changed: 53 additions & 5 deletions

File tree

framework/src/main/java/org/tron/core/net/TronNetService.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,8 @@ private static void setP2pConfig(P2pConfig config) {
8282
public void start() {
8383
try {
8484
init = true;
85-
setP2pConfig(getConfig());
85+
P2pConfig config = new P2pConfig();
86+
setP2pConfig(getConfig(config));
8687
p2pService.start(p2pConfig);
8788
p2pService.register(p2pEventHandler);
8889
advService.init();
@@ -147,11 +148,10 @@ public static boolean hasIpv4Stack(Set<String> ipSet) {
147148
return false;
148149
}
149150

150-
private P2pConfig getConfig() {
151+
private P2pConfig getConfig(P2pConfig config) {
151152
List<InetSocketAddress> seeds = parameter.getSeedNode().getAddressList();
152153
seeds.addAll(nodePersistService.dbRead());
153154
logger.debug("Seed InetSocketAddress: {}", seeds);
154-
P2pConfig config = new P2pConfig();
155155
config.getSeedNodes().addAll(seeds);
156156
config.getActiveNodes().addAll(parameter.getActiveNodes());
157157
config.getTrustNodes().addAll(parameter.getPassiveNodes());

framework/src/main/resources/config.conf

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -202,6 +202,19 @@ node {
202202
solidityPort = 8091
203203
}
204204

205+
# use your ipv6 address for node discovery and tcp connection, default false
206+
enableIpv6 = false
207+
208+
# if your node's highest block num is below than all your pees', try to acquire new connection. default false
209+
effectiveCheckEnable = false
210+
211+
dns {
212+
# dns urls to get nodes, url format tree://{pubkey}@{domain}, default empty
213+
treeUrls = [
214+
#"tree://APFGGTFOBVE2ZNAB3CSMNNX6RRK3ODIRLP2AA5U4YFAA6MSYZUYTQ@nodes1.example.org",
215+
]
216+
}
217+
205218
rpc {
206219
port = 50051
207220
#solidityPort = 50061

framework/src/test/java/org/tron/common/config/args/ArgsTest.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,16 +11,18 @@
1111

1212
public class ArgsTest {
1313

14+
private static final String dbPath = "output_arg_test";
15+
1416
@Before
1517
public void init() {
16-
Args.setParam(new String[]{"--output-directory", "output-directory", "--p2p-disable", "true",
18+
Args.setParam(new String[]{"--output-directory", dbPath, "--p2p-disable", "true",
1719
"--debug"}, Constant.TEST_CONF);
1820
}
1921

2022
@After
2123
public void destroy() {
2224
Args.clearParam();
23-
FileUtil.deleteDir(new File("output-directory"));
25+
FileUtil.deleteDir(new File(dbPath));
2426
}
2527

2628
@Test

framework/src/test/java/org/tron/core/actuator/utils/ProposalUtilTest.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -357,6 +357,10 @@ public void validateCheck() {
357357
Assert.assertEquals("Bad chain parameter value, valid range is [0, 1_000_000_000_000L]",
358358
e.getMessage());
359359
}
360+
361+
forkUtils.getManager().getDynamicPropertiesStore()
362+
.statsByVersion(ForkBlockVersionEnum.ENERGY_LIMIT.getValue(), stats);
363+
forkUtils.reset();
360364
}
361365

362366
@Test

framework/src/test/java/org/tron/core/db/ManagerTest.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -226,6 +226,13 @@ public void pushBlock() {
226226
}
227227
}
228228

229+
try {
230+
chainManager.getBlockIdByNum(-1);
231+
Assert.fail();
232+
} catch (ItemNotFoundException e) {
233+
Assert.assertTrue(true);
234+
}
235+
229236
Assert.assertTrue("hasBlocks is error", chainManager.hasBlocks());
230237
}
231238

framework/src/test/java/org/tron/core/net/MessageTest.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,11 @@ public Class<?> getAnswerMessage() {
107107
messageStatistics.addTcpOutMessage(message4);
108108
messageStatistics.addTcpInMessage(message5);
109109
messageStatistics.addTcpOutMessage(message5);
110+
try {
111+
Thread.sleep(2000);// so that gap > 1 in MessageCount.update method
112+
} catch (InterruptedException e) {
113+
//ignore
114+
}
110115
messageStatistics.addTcpInMessage(message6);
111116
messageStatistics.addTcpOutMessage(message6);
112117
messageStatistics.addTcpInMessage(message7);

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

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
package org.tron.core.net.services;
22

33
import java.io.File;
4+
import java.lang.reflect.Method;
45
import java.net.InetSocketAddress;
56
import org.junit.After;
67
import org.junit.Assert;
@@ -37,6 +38,18 @@ public void destroy() {
3738
FileUtil.deleteDir(new File(dbPath));
3839
}
3940

41+
@Test
42+
public void testNoIpv4() throws Exception {
43+
TronNetService tronNetService = context.getBean(TronNetService.class);
44+
Method privateMethod = tronNetService.getClass()
45+
.getDeclaredMethod("getConfig", P2pConfig.class);
46+
privateMethod.setAccessible(true);
47+
P2pConfig config = new P2pConfig();
48+
config.setIp(null);
49+
P2pConfig newConfig = (P2pConfig) privateMethod.invoke(tronNetService, config);
50+
Assert.assertNotNull(newConfig.getIp());
51+
}
52+
4053
@Test
4154
public void testFind() {
4255
TronNetService tronNetService = context.getBean(TronNetService.class);

framework/src/test/resources/config-test.conf

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,8 +105,12 @@ node {
105105
solidityPort = 8091
106106
}
107107

108+
# use your ipv6 address for node discovery and tcp connection, default false
108109
enableIpv6 = false
109110

111+
# if your node's highest block num is below than all your pees', try to acquire new connection
112+
effectiveCheckEnable = false
113+
110114
dns {
111115
# dns urls to get nodes, url format tree://{pubkey}@{domain}, default empty
112116
treeUrls = [

0 commit comments

Comments
 (0)