Skip to content

Commit a606b83

Browse files
committed
feat: add address param
1 parent 08ec4bd commit a606b83

2 files changed

Lines changed: 54 additions & 0 deletions

File tree

trpc-core/src/test/java/com/tencent/trpc/core/common/config/ProtocolConfigTest.java

Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,11 @@
1414
import static org.junit.Assert.assertEquals;
1515
import static org.junit.Assert.assertFalse;
1616
import static org.junit.Assert.assertNotNull;
17+
import static org.junit.Assert.assertNull;
1718
import static org.junit.Assert.assertSame;
1819
import static org.junit.Assert.assertTrue;
1920

21+
import java.net.InetSocketAddress;
2022
import com.google.common.collect.ImmutableMap;
2123
import com.tencent.trpc.core.common.Constants;
2224
import org.junit.Test;
@@ -129,4 +131,43 @@ public void testConfig() {
129131
assertTrue(config.isSetDefault());
130132
assertSame(serviceConfig, config.getServiceConfig());
131133
}
134+
135+
@Test
136+
public void testSocketAddress() {
137+
ProtocolConfig config = new ProtocolConfig();
138+
config.setIp("127.0.0.1");
139+
config.setPort(8080);
140+
config.setDefault();
141+
InetSocketAddress socketAddress = config.toInetSocketAddress();
142+
assertNotNull(socketAddress);
143+
assertEquals("127.0.0.1", socketAddress.getHostString());
144+
assertEquals(8080, socketAddress.getPort());
145+
}
146+
147+
@Test
148+
public void testSocketAddressNull() {
149+
ProtocolConfig config = new ProtocolConfig();
150+
config.setIp("192.168.1.1");
151+
config.setPort(9090);
152+
InetSocketAddress addr = config.toInetSocketAddress();
153+
assertNotNull(addr);
154+
assertEquals("192.168.1.1", addr.getHostString());
155+
assertEquals(9090, addr.getPort());
156+
}
157+
158+
@Test
159+
public void testIpBlank() {
160+
ProtocolConfig config = new ProtocolConfig();
161+
assertNull(config.toInetSocketAddress());
162+
}
163+
164+
@Test
165+
public void testToString() {
166+
ProtocolConfig config = new ProtocolConfig();
167+
config.setIp("127.0.0.1");
168+
config.setPort(8080);
169+
config.setDefault();
170+
String str = config.toString();
171+
assertTrue(str.contains("socketAddress="));
172+
}
132173
}

trpc-core/src/test/java/com/tencent/trpc/core/common/config/ServiceConfigTest.java

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -236,4 +236,17 @@ public byte[] invoke(RpcClientContext context, byte[] body) {
236236
config.register();
237237
config.unRegister();
238238
}
239+
240+
@Test
241+
public void testToStringContainsAddress() {
242+
ServiceConfig config = new ServiceConfig();
243+
config.setName("name");
244+
config.setIp("127.0.0.1");
245+
config.setPort(8080);
246+
config.setAddress("127.0.0.1:9092?topics=quickstart-events&group=quickstart-group");
247+
config.setDefault();
248+
String str = config.toString();
249+
assertTrue(str.contains("address="));
250+
assertTrue(str.contains("127.0.0.1:9092?topics=quickstart-events&group=quickstart-group"));
251+
}
239252
}

0 commit comments

Comments
 (0)