study/Spring-Cloud-Alibaba
1
0
Fork 0
mirror of https://gitee.com/mirrors/Spring-Cloud-Alibaba.git synced 2021-06-26 13:25:11 +08:00
Code Issues Releases Wiki Activity

add NacosRegistrationCustomizer.java

Browse Source
This commit is contained in:
如梦技术 2020-07-26 22:49:10 +08:00
parent 588f72a933
commit 20f0e57714
1 changed files with 49 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

97
spring-cloud-alibaba-starters/spring-cloud-starter-alibaba-nacos-discovery/src/test/java/com/alibaba/cloud/nacos/registry/NacosRegistrationCustomizerTest.java
View File

@ -16,6 +16,11 @@
package com.alibaba.cloud.nacos.registry; package com.alibaba.cloud.nacos.registry;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.util.Map;
import java.util.Properties;
import com.alibaba.cloud.nacos.discovery.NacosDiscoveryClientConfiguration; import com.alibaba.cloud.nacos.discovery.NacosDiscoveryClientConfiguration;
import com.alibaba.nacos.api.NacosFactory; import com.alibaba.nacos.api.NacosFactory;
import org.junit.Assert; import org.junit.Assert;
@ -27,6 +32,7 @@ import org.powermock.core.classloader.annotations.PowerMockIgnore;
import org.powermock.core.classloader.annotations.PrepareForTest; import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner; import org.powermock.modules.junit4.PowerMockRunner;
import org.powermock.modules.junit4.PowerMockRunnerDelegate; import org.powermock.modules.junit4.PowerMockRunnerDelegate;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.EnableAutoConfiguration; import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
import org.springframework.boot.autoconfigure.ImportAutoConfiguration; import org.springframework.boot.autoconfigure.ImportAutoConfiguration;
@ -36,11 +42,6 @@ import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
import org.springframework.test.context.junit4.SpringRunner; import org.springframework.test.context.junit4.SpringRunner;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.util.Map;
import java.util.Properties;
import static org.springframework.boot.test.context.SpringBootTest.WebEnvironment.RANDOM_PORT; import static org.springframework.boot.test.context.SpringBootTest.WebEnvironment.RANDOM_PORT;
/** /**
@ -50,54 +51,54 @@ import static org.springframework.boot.test.context.SpringBootTest.WebEnvironmen
@PowerMockIgnore("javax.management.*") @PowerMockIgnore("javax.management.*")
@PowerMockRunnerDelegate(SpringRunner.class) @PowerMockRunnerDelegate(SpringRunner.class)
@PrepareForTest({ NacosFactory.class }) @PrepareForTest({ NacosFactory.class })
@SpringBootTest( @SpringBootTest(classes = NacosRegistrationCustomizerTest.TestConfig.class,
classes = NacosRegistrationCustomizerTest.TestConfig.class, properties = { "spring.application.name=myTestService1",
properties = { "spring.application.name=myTestService1", "spring.cloud.nacos.discovery.server-addr=127.0.0.1:8848" },
"spring.cloud.nacos.discovery.server-addr=127.0.0.1:8848" }, webEnvironment = RANDOM_PORT)
webEnvironment = RANDOM_PORT)
public class NacosRegistrationCustomizerTest { public class NacosRegistrationCustomizerTest {
@Autowired @Autowired
private NacosAutoServiceRegistration nacosAutoServiceRegistration; private NacosAutoServiceRegistration nacosAutoServiceRegistration;
static { static {
try { try {
Method method = PowerMockito.method(NacosFactory.class, "createNamingService", Method method = PowerMockito.method(NacosFactory.class, "createNamingService",
Properties.class); Properties.class);
MethodProxy.proxy(method, new InvocationHandler() { MethodProxy.proxy(method, new InvocationHandler() {
@Override @Override
public Object invoke(Object proxy, Method method, Object[] args) public Object invoke(Object proxy, Method method, Object[] args)
throws Throwable { throws Throwable {
return new MockNamingService(); return new MockNamingService();
} }
}); });
} }
catch (Exception e) { catch (Exception e) {
e.printStackTrace(); e.printStackTrace();
} }
} }
@Test @Test
public void contextLoads() throws Exception { public void contextLoads() throws Exception {
NacosRegistration registration = nacosAutoServiceRegistration.getRegistration(); NacosRegistration registration = nacosAutoServiceRegistration.getRegistration();
Map<String, String> metadata = registration.getMetadata(); Map<String, String> metadata = registration.getMetadata();
Assert.assertEquals("test1", metadata.get("test1")); Assert.assertEquals("test1", metadata.get("test1"));
} }
@Configuration @Configuration
@EnableAutoConfiguration @EnableAutoConfiguration
@ImportAutoConfiguration({ AutoServiceRegistrationConfiguration.class, @ImportAutoConfiguration({ AutoServiceRegistrationConfiguration.class,
NacosDiscoveryClientConfiguration.class, NacosDiscoveryClientConfiguration.class,
NacosServiceRegistryAutoConfiguration.class }) NacosServiceRegistryAutoConfiguration.class })
public static class TestConfig { public static class TestConfig {
@Bean @Bean
public NacosRegistrationCustomizer nacosRegistrationCustomizer() { public NacosRegistrationCustomizer nacosRegistrationCustomizer() {
return registration -> { return registration -> {
Map<String, String> metadata = registration.getMetadata(); Map<String, String> metadata = registration.getMetadata();
metadata.put("test1", "test1"); metadata.put("test1", "test1");
}; };
} }
}
}
} }