test:add junit test to polaris-discovery. (#205)
Haotian Zhang
3 years ago
committed by
GitHub
parent
b97f77a418
commit
6a90de6d1e
@ -0,0 +1,90 @@
|
|||||||
|
/*
|
||||||
|
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||||
|
*
|
||||||
|
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||||
|
*
|
||||||
|
* Licensed under the BSD 3-Clause License (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* https://opensource.org/licenses/BSD-3-Clause
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software distributed
|
||||||
|
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
||||||
|
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
||||||
|
* specific language governing permissions and limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package com.tencent.cloud.polaris;
|
||||||
|
|
||||||
|
import com.tencent.cloud.polaris.context.PolarisContextAutoConfiguration;
|
||||||
|
import com.tencent.cloud.polaris.discovery.PolarisDiscoveryHandler;
|
||||||
|
import com.tencent.cloud.polaris.extend.consul.ConsulContextProperties;
|
||||||
|
import com.tencent.polaris.api.core.ConsumerAPI;
|
||||||
|
import com.tencent.polaris.api.core.ProviderAPI;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import org.springframework.boot.autoconfigure.AutoConfigurations;
|
||||||
|
import org.springframework.boot.test.context.runner.ApplicationContextRunner;
|
||||||
|
import org.springframework.context.annotation.Bean;
|
||||||
|
import org.springframework.context.annotation.Configuration;
|
||||||
|
|
||||||
|
import static org.assertj.core.api.Assertions.assertThat;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test for {@link DiscoveryPropertiesAutoConfiguration}.
|
||||||
|
*
|
||||||
|
* @author Haotian Zhang
|
||||||
|
*/
|
||||||
|
public class DiscoveryPropertiesAutoConfigurationTest {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testDefaultInitialization() {
|
||||||
|
ApplicationContextRunner applicationContextRunner = new ApplicationContextRunner().withConfiguration(
|
||||||
|
AutoConfigurations.of(PolarisContextAutoConfiguration.class,
|
||||||
|
DiscoveryPropertiesAutoConfiguration.class));
|
||||||
|
applicationContextRunner.run(context -> {
|
||||||
|
assertThat(context).hasSingleBean(DiscoveryPropertiesAutoConfiguration.class);
|
||||||
|
assertThat(context).hasSingleBean(PolarisDiscoveryProperties.class);
|
||||||
|
assertThat(context).hasSingleBean(ConsulContextProperties.class);
|
||||||
|
assertThat(context).hasSingleBean(ProviderAPI.class);
|
||||||
|
assertThat(context).hasSingleBean(ConsumerAPI.class);
|
||||||
|
assertThat(context).hasSingleBean(PolarisDiscoveryHandler.class);
|
||||||
|
assertThat(context).hasSingleBean(DiscoveryConfigModifier.class);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testInit() {
|
||||||
|
ApplicationContextRunner applicationContextRunner = new ApplicationContextRunner().withConfiguration(
|
||||||
|
AutoConfigurations.of(PolarisContextAutoConfiguration.class,
|
||||||
|
TestConfiguration.class,
|
||||||
|
DiscoveryPropertiesAutoConfiguration.class))
|
||||||
|
.withPropertyValues("spring.cloud.polaris.discovery.register=false")
|
||||||
|
.withPropertyValues("spring.cloud.consul.discovery.register=false")
|
||||||
|
.withPropertyValues("spring.cloud.consul.discovery.enabled=false");
|
||||||
|
applicationContextRunner.run(context -> {
|
||||||
|
assertThat(context).hasSingleBean(DiscoveryPropertiesAutoConfiguration.class);
|
||||||
|
DiscoveryPropertiesAutoConfiguration discoveryPropertiesAutoConfiguration = context.getBean(DiscoveryPropertiesAutoConfiguration.class);
|
||||||
|
assertThat(discoveryPropertiesAutoConfiguration.isRegisterEnabled()).isFalse();
|
||||||
|
assertThat(discoveryPropertiesAutoConfiguration.isDiscoveryEnabled()).isFalse();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
@Configuration
|
||||||
|
static class TestConfiguration {
|
||||||
|
@Bean
|
||||||
|
public PolarisDiscoveryProperties polarisDiscoveryProperties() {
|
||||||
|
PolarisDiscoveryProperties polarisDiscoveryProperties = new PolarisDiscoveryProperties();
|
||||||
|
polarisDiscoveryProperties.setEnabled(false);
|
||||||
|
return polarisDiscoveryProperties;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Bean
|
||||||
|
public ConsulContextProperties consulContextProperties() {
|
||||||
|
ConsulContextProperties consulContextProperties = new ConsulContextProperties();
|
||||||
|
consulContextProperties.setEnabled(true);
|
||||||
|
return consulContextProperties;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,46 @@
|
|||||||
|
/*
|
||||||
|
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||||
|
*
|
||||||
|
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||||
|
*
|
||||||
|
* Licensed under the BSD 3-Clause License (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* https://opensource.org/licenses/BSD-3-Clause
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software distributed
|
||||||
|
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
||||||
|
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
||||||
|
* specific language governing permissions and limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package com.tencent.cloud.polaris;
|
||||||
|
|
||||||
|
import com.tencent.cloud.polaris.context.PolarisContextAutoConfiguration;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import org.springframework.boot.autoconfigure.AutoConfigurations;
|
||||||
|
import org.springframework.boot.test.context.runner.ApplicationContextRunner;
|
||||||
|
|
||||||
|
import static org.assertj.core.api.Assertions.assertThat;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test for {@link DiscoveryPropertiesBootstrapAutoConfiguration}.
|
||||||
|
*
|
||||||
|
* @author Haotian Zhang
|
||||||
|
*/
|
||||||
|
public class DiscoveryPropertiesBootstrapAutoConfigurationTest {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testDefaultInitialization() {
|
||||||
|
ApplicationContextRunner applicationContextRunner = new ApplicationContextRunner().withConfiguration(
|
||||||
|
AutoConfigurations.of(PolarisContextAutoConfiguration.class,
|
||||||
|
DiscoveryPropertiesBootstrapAutoConfiguration.class))
|
||||||
|
.withPropertyValues("spring.cloud.polaris.enabled=true");
|
||||||
|
applicationContextRunner.run(context -> {
|
||||||
|
assertThat(context).hasSingleBean(DiscoveryPropertiesBootstrapAutoConfiguration.class);
|
||||||
|
assertThat(context).hasSingleBean(DiscoveryPropertiesAutoConfiguration.class);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,102 @@
|
|||||||
|
/*
|
||||||
|
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||||
|
*
|
||||||
|
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||||
|
*
|
||||||
|
* Licensed under the BSD 3-Clause License (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* https://opensource.org/licenses/BSD-3-Clause
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software distributed
|
||||||
|
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
||||||
|
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
||||||
|
* specific language governing permissions and limitations under the License.
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
package com.tencent.cloud.polaris;
|
||||||
|
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import static com.tencent.polaris.test.common.Consts.NAMESPACE_TEST;
|
||||||
|
import static com.tencent.polaris.test.common.Consts.PORT;
|
||||||
|
import static com.tencent.polaris.test.common.Consts.PROVIDER_TOKEN;
|
||||||
|
import static com.tencent.polaris.test.common.Consts.SERVICE_PROVIDER;
|
||||||
|
import static org.assertj.core.api.Assertions.assertThat;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test for {@link PolarisDiscoveryProperties}.
|
||||||
|
*
|
||||||
|
* @author Haotian Zhang
|
||||||
|
*/
|
||||||
|
public class PolarisDiscoveryPropertiesTest {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetAndSet() {
|
||||||
|
PolarisDiscoveryProperties polarisDiscoveryProperties = new PolarisDiscoveryProperties();
|
||||||
|
|
||||||
|
// HeartbeatEnabled
|
||||||
|
polarisDiscoveryProperties.setHeartbeatEnabled(true);
|
||||||
|
assertThat(polarisDiscoveryProperties.isHeartbeatEnabled()).isTrue();
|
||||||
|
|
||||||
|
// Namespace
|
||||||
|
polarisDiscoveryProperties.setNamespace(NAMESPACE_TEST);
|
||||||
|
assertThat(polarisDiscoveryProperties.getNamespace()).isEqualTo(NAMESPACE_TEST);
|
||||||
|
|
||||||
|
// Weight
|
||||||
|
polarisDiscoveryProperties.setWeight(10);
|
||||||
|
assertThat(polarisDiscoveryProperties.getWeight()).isEqualTo(10);
|
||||||
|
|
||||||
|
// Service
|
||||||
|
polarisDiscoveryProperties.setService(SERVICE_PROVIDER);
|
||||||
|
assertThat(polarisDiscoveryProperties.getService()).isEqualTo(SERVICE_PROVIDER);
|
||||||
|
|
||||||
|
// Enabled
|
||||||
|
polarisDiscoveryProperties.setEnabled(true);
|
||||||
|
assertThat(polarisDiscoveryProperties.isEnabled()).isTrue();
|
||||||
|
|
||||||
|
// RegisterEnabled
|
||||||
|
polarisDiscoveryProperties.setRegisterEnabled(true);
|
||||||
|
assertThat(polarisDiscoveryProperties.isRegisterEnabled()).isTrue();
|
||||||
|
|
||||||
|
// Token
|
||||||
|
polarisDiscoveryProperties.setToken(PROVIDER_TOKEN);
|
||||||
|
assertThat(polarisDiscoveryProperties.getToken()).isEqualTo(PROVIDER_TOKEN);
|
||||||
|
|
||||||
|
// Version
|
||||||
|
polarisDiscoveryProperties.setVersion("1.0.0");
|
||||||
|
assertThat(polarisDiscoveryProperties.getVersion()).isEqualTo("1.0.0");
|
||||||
|
|
||||||
|
// HTTP
|
||||||
|
polarisDiscoveryProperties.setProtocol("HTTP");
|
||||||
|
assertThat(polarisDiscoveryProperties.getProtocol()).isEqualTo("HTTP");
|
||||||
|
|
||||||
|
// Port
|
||||||
|
polarisDiscoveryProperties.setPort(PORT);
|
||||||
|
assertThat(polarisDiscoveryProperties.getPort()).isEqualTo(PORT);
|
||||||
|
|
||||||
|
// HealthCheckUrl
|
||||||
|
polarisDiscoveryProperties.setHealthCheckUrl("/health");
|
||||||
|
assertThat(polarisDiscoveryProperties.getHealthCheckUrl()).isEqualTo("/health");
|
||||||
|
|
||||||
|
// ServiceListRefreshInterval
|
||||||
|
polarisDiscoveryProperties.setServiceListRefreshInterval(1000L);
|
||||||
|
assertThat(polarisDiscoveryProperties.getServiceListRefreshInterval()).isEqualTo(1000L);
|
||||||
|
|
||||||
|
assertThat(polarisDiscoveryProperties.toString())
|
||||||
|
.isEqualTo("PolarisDiscoveryProperties{"
|
||||||
|
+ "namespace='Test'"
|
||||||
|
+ ", service='java_provider_test'"
|
||||||
|
+ ", token='19485a7674294e3c88dba293373c1534'"
|
||||||
|
+ ", weight=10, version='1.0.0'"
|
||||||
|
+ ", protocol='HTTP'"
|
||||||
|
+ ", port=9091"
|
||||||
|
+ ", enabled=true"
|
||||||
|
+ ", registerEnabled=true"
|
||||||
|
+ ", heartbeatEnabled=true"
|
||||||
|
+ ", healthCheckUrl='/health'"
|
||||||
|
+ ", serviceListRefreshInterval=1000}");
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,114 @@
|
|||||||
|
/*
|
||||||
|
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||||
|
*
|
||||||
|
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||||
|
*
|
||||||
|
* Licensed under the BSD 3-Clause License (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* https://opensource.org/licenses/BSD-3-Clause
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software distributed
|
||||||
|
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
||||||
|
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
||||||
|
* specific language governing permissions and limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package com.tencent.cloud.polaris.discovery.refresh;
|
||||||
|
|
||||||
|
import java.lang.reflect.Field;
|
||||||
|
import java.util.Collections;
|
||||||
|
|
||||||
|
import com.tencent.polaris.api.pojo.DefaultInstance;
|
||||||
|
import com.tencent.polaris.api.pojo.ServiceEventKey;
|
||||||
|
import com.tencent.polaris.api.pojo.ServiceInfo;
|
||||||
|
import com.tencent.polaris.api.pojo.ServiceKey;
|
||||||
|
import com.tencent.polaris.client.pojo.ServiceInstancesByProto;
|
||||||
|
import com.tencent.polaris.client.pojo.ServicesByProto;
|
||||||
|
import org.assertj.core.api.Assertions;
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import org.springframework.context.ApplicationEvent;
|
||||||
|
import org.springframework.context.ApplicationEventPublisher;
|
||||||
|
|
||||||
|
import static com.tencent.polaris.test.common.Consts.HOST;
|
||||||
|
import static com.tencent.polaris.test.common.Consts.NAMESPACE_TEST;
|
||||||
|
import static com.tencent.polaris.test.common.Consts.PORT;
|
||||||
|
import static com.tencent.polaris.test.common.Consts.SERVICE_PROVIDER;
|
||||||
|
import static org.mockito.ArgumentMatchers.any;
|
||||||
|
import static org.mockito.Mockito.doNothing;
|
||||||
|
import static org.mockito.Mockito.mock;
|
||||||
|
import static org.mockito.Mockito.times;
|
||||||
|
import static org.mockito.Mockito.verify;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test for {@link PolarisServiceStatusChangeListener}.
|
||||||
|
*
|
||||||
|
* @author Haotian Zhang
|
||||||
|
*/
|
||||||
|
public class PolarisServiceStatusChangeListenerTest {
|
||||||
|
|
||||||
|
private ApplicationEventPublisher publisher;
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void setUp() {
|
||||||
|
publisher = mock(ApplicationEventPublisher.class);
|
||||||
|
doNothing().when(publisher).publishEvent(any(ApplicationEvent.class));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testOnResourceUpdated() {
|
||||||
|
PolarisServiceStatusChangeListener polarisServiceStatusChangeListener = new PolarisServiceStatusChangeListener();
|
||||||
|
polarisServiceStatusChangeListener.setApplicationEventPublisher(publisher);
|
||||||
|
|
||||||
|
// Service update event
|
||||||
|
ServiceEventKey serviceUpdateEventKey = new ServiceEventKey(new ServiceKey(NAMESPACE_TEST, SERVICE_PROVIDER), ServiceEventKey.EventType.SERVICE);
|
||||||
|
ServiceInfo serviceInfo = new ServiceInfo();
|
||||||
|
serviceInfo.setNamespace(NAMESPACE_TEST);
|
||||||
|
serviceInfo.setService(SERVICE_PROVIDER);
|
||||||
|
// Need update
|
||||||
|
ServicesByProto oldServices = new ServicesByProto(Collections.emptyList());
|
||||||
|
ServicesByProto newServices = new ServicesByProto(Collections.singletonList(serviceInfo));
|
||||||
|
polarisServiceStatusChangeListener.onResourceUpdated(serviceUpdateEventKey, oldServices, newServices);
|
||||||
|
verify(publisher, times(1)).publishEvent(any(ApplicationEvent.class));
|
||||||
|
// No need update
|
||||||
|
oldServices = new ServicesByProto(Collections.singletonList(serviceInfo));
|
||||||
|
newServices = new ServicesByProto(Collections.singletonList(serviceInfo));
|
||||||
|
polarisServiceStatusChangeListener.onResourceUpdated(serviceUpdateEventKey, oldServices, newServices);
|
||||||
|
verify(publisher, times(1)).publishEvent(any(ApplicationEvent.class));
|
||||||
|
|
||||||
|
|
||||||
|
// Instance update event
|
||||||
|
ServiceEventKey instanceUpdateEventKey = new ServiceEventKey(new ServiceKey(NAMESPACE_TEST, SERVICE_PROVIDER), ServiceEventKey.EventType.INSTANCE);
|
||||||
|
DefaultInstance instance = new DefaultInstance();
|
||||||
|
instance.setNamespace(NAMESPACE_TEST);
|
||||||
|
instance.setService(SERVICE_PROVIDER);
|
||||||
|
instance.setHost(HOST);
|
||||||
|
instance.setPort(PORT);
|
||||||
|
try {
|
||||||
|
Field instances = ServiceInstancesByProto.class.getDeclaredField("instances");
|
||||||
|
instances.setAccessible(true);
|
||||||
|
|
||||||
|
// Need update
|
||||||
|
ServiceInstancesByProto oldInstances = new ServiceInstancesByProto();
|
||||||
|
instances.set(oldInstances, Collections.emptyList());
|
||||||
|
ServiceInstancesByProto newInstances = new ServiceInstancesByProto();
|
||||||
|
instances.set(newInstances, Collections.singletonList(instance));
|
||||||
|
polarisServiceStatusChangeListener.onResourceUpdated(serviceUpdateEventKey, oldInstances, newInstances);
|
||||||
|
verify(publisher, times(2)).publishEvent(any(ApplicationEvent.class));
|
||||||
|
|
||||||
|
// No need update
|
||||||
|
oldInstances = new ServiceInstancesByProto();
|
||||||
|
instances.set(oldInstances, Collections.singletonList(instance));
|
||||||
|
newInstances = new ServiceInstancesByProto();
|
||||||
|
instances.set(newInstances, Collections.singletonList(instance));
|
||||||
|
polarisServiceStatusChangeListener.onResourceUpdated(serviceUpdateEventKey, oldInstances, newInstances);
|
||||||
|
verify(publisher, times(2)).publishEvent(any(ApplicationEvent.class));
|
||||||
|
}
|
||||||
|
catch (NoSuchFieldException | IllegalAccessException e) {
|
||||||
|
Assertions.fail("Exception encountered.", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,90 @@
|
|||||||
|
/*
|
||||||
|
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||||
|
*
|
||||||
|
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||||
|
*
|
||||||
|
* Licensed under the BSD 3-Clause License (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* https://opensource.org/licenses/BSD-3-Clause
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software distributed
|
||||||
|
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
||||||
|
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
||||||
|
* specific language governing permissions and limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package com.tencent.cloud.polaris.extend.consul;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
import com.tencent.cloud.polaris.DiscoveryPropertiesAutoConfiguration;
|
||||||
|
import com.tencent.cloud.polaris.context.PolarisContextAutoConfiguration;
|
||||||
|
import com.tencent.polaris.client.api.SDKContext;
|
||||||
|
import com.tencent.polaris.factory.config.global.ServerConnectorConfigImpl;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import org.springframework.boot.autoconfigure.AutoConfigurations;
|
||||||
|
import org.springframework.boot.test.context.runner.ApplicationContextRunner;
|
||||||
|
import org.springframework.context.annotation.Bean;
|
||||||
|
import org.springframework.context.annotation.Configuration;
|
||||||
|
|
||||||
|
import static com.tencent.polaris.plugins.connector.common.constant.ConsulConstant.MetadataMapKey.INSTANCE_ID_KEY;
|
||||||
|
import static com.tencent.polaris.plugins.connector.common.constant.ConsulConstant.MetadataMapKey.IP_ADDRESS_KEY;
|
||||||
|
import static com.tencent.polaris.plugins.connector.common.constant.ConsulConstant.MetadataMapKey.PREFER_IP_ADDRESS_KEY;
|
||||||
|
import static com.tencent.polaris.plugins.connector.common.constant.ConsulConstant.MetadataMapKey.SERVICE_NAME_KEY;
|
||||||
|
import static com.tencent.polaris.test.common.Consts.HOST;
|
||||||
|
import static com.tencent.polaris.test.common.Consts.SERVICE_PROVIDER;
|
||||||
|
import static org.assertj.core.api.Assertions.assertThat;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test for {@link ConsulContextProperties}.
|
||||||
|
*
|
||||||
|
* @author Haotian Zhang
|
||||||
|
*/
|
||||||
|
public class ConsulContextPropertiesTest {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testModify() {
|
||||||
|
ApplicationContextRunner applicationContextRunner = new ApplicationContextRunner().withConfiguration(
|
||||||
|
AutoConfigurations.of(PolarisContextAutoConfiguration.class,
|
||||||
|
ConsulContextPropertiesTest.TestConfiguration.class,
|
||||||
|
DiscoveryPropertiesAutoConfiguration.class))
|
||||||
|
.withPropertyValues("spring.cloud.consul.discovery.register=true")
|
||||||
|
.withPropertyValues("spring.cloud.consul.discovery.enabled=true")
|
||||||
|
.withPropertyValues("spring.application.name=" + SERVICE_PROVIDER)
|
||||||
|
.withPropertyValues("spring.cloud.consul.discovery.instance-id=ins-test")
|
||||||
|
.withPropertyValues("spring.cloud.consul.discovery.prefer-ip-address=true")
|
||||||
|
.withPropertyValues("spring.cloud.consul.discovery.ip-address=" + HOST);
|
||||||
|
applicationContextRunner.run(context -> {
|
||||||
|
assertThat(context).hasSingleBean(SDKContext.class);
|
||||||
|
SDKContext sdkContext = context.getBean(SDKContext.class);
|
||||||
|
com.tencent.polaris.api.config.Configuration configuration = sdkContext.getConfig();
|
||||||
|
List<ServerConnectorConfigImpl> serverConnectorConfigs = configuration.getGlobal().getServerConnectors();
|
||||||
|
Map<String, String> metadata = null;
|
||||||
|
for (ServerConnectorConfigImpl serverConnectorConfig : serverConnectorConfigs) {
|
||||||
|
if (serverConnectorConfig.getId().equals("consul")) {
|
||||||
|
metadata = serverConnectorConfig.getMetadata();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
assertThat(metadata).isNotNull();
|
||||||
|
assertThat(metadata.get(SERVICE_NAME_KEY)).isEqualTo(SERVICE_PROVIDER);
|
||||||
|
assertThat(metadata.get(INSTANCE_ID_KEY)).isEqualTo("ins-test");
|
||||||
|
assertThat(metadata.get(PREFER_IP_ADDRESS_KEY)).isEqualTo("true");
|
||||||
|
assertThat(metadata.get(IP_ADDRESS_KEY)).isEqualTo(HOST);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
@Configuration
|
||||||
|
static class TestConfiguration {
|
||||||
|
|
||||||
|
@Bean
|
||||||
|
public ConsulContextProperties consulContextProperties() {
|
||||||
|
ConsulContextProperties consulContextProperties = new ConsulContextProperties();
|
||||||
|
consulContextProperties.setEnabled(true);
|
||||||
|
return consulContextProperties;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,60 @@
|
|||||||
|
/*
|
||||||
|
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||||
|
*
|
||||||
|
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||||
|
*
|
||||||
|
* Licensed under the BSD 3-Clause License (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* https://opensource.org/licenses/BSD-3-Clause
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software distributed
|
||||||
|
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
||||||
|
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
||||||
|
* specific language governing permissions and limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package com.tencent.cloud.polaris.util;
|
||||||
|
|
||||||
|
import org.assertj.core.util.Maps;
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.junit.runner.RunWith;
|
||||||
|
|
||||||
|
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
||||||
|
import org.springframework.boot.test.context.SpringBootTest;
|
||||||
|
import org.springframework.boot.web.server.LocalServerPort;
|
||||||
|
import org.springframework.test.context.junit4.SpringRunner;
|
||||||
|
import org.springframework.web.bind.annotation.GetMapping;
|
||||||
|
import org.springframework.web.bind.annotation.RestController;
|
||||||
|
|
||||||
|
import static org.assertj.core.api.Assertions.assertThat;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test for {@link OkHttpUtil}.
|
||||||
|
*
|
||||||
|
* @author Haotian Zhang
|
||||||
|
*/
|
||||||
|
@RunWith(SpringRunner.class)
|
||||||
|
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT, classes = OkHttpUtilTest.TestApplication.class)
|
||||||
|
public class OkHttpUtilTest {
|
||||||
|
|
||||||
|
@LocalServerPort
|
||||||
|
private int port;
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGet() {
|
||||||
|
assertThat(OkHttpUtil.get("http://localhost:" + port + "/test", Maps.newHashMap("key", "value"))).isTrue();
|
||||||
|
assertThat(OkHttpUtil.get("http://localhost:" + port + "/error", Maps.newHashMap("key", "value"))).isFalse();
|
||||||
|
assertThat(OkHttpUtil.get("http://localhost:55555/error", Maps.newHashMap("key", "value"))).isFalse();
|
||||||
|
}
|
||||||
|
|
||||||
|
@SpringBootApplication
|
||||||
|
@RestController
|
||||||
|
static class TestApplication {
|
||||||
|
@GetMapping("/test")
|
||||||
|
public String test() {
|
||||||
|
return "test";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue