[ISSUE #5155] Add unit tests for package com.alibaba.nacos.client.naming.remote in nacos 2.0 (#5679)
* wip: add NacosNamingMaintainServiceTest * wip * add NacosNamingServiceTest
This commit is contained in:
shalk(xiao kun)
GitHub
parent
b6d96b69c7
commit
c8ad434ba9
@ -0,0 +1,282 @@
|
|||||||
|
/*
|
||||||
|
*
|
||||||
|
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* 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.alibaba.nacos.client.naming;
|
||||||
|
|
||||||
|
import com.alibaba.nacos.api.PropertyKeyConst;
|
||||||
|
import com.alibaba.nacos.api.common.Constants;
|
||||||
|
import com.alibaba.nacos.api.exception.NacosException;
|
||||||
|
import com.alibaba.nacos.api.naming.pojo.Instance;
|
||||||
|
import com.alibaba.nacos.api.naming.pojo.Service;
|
||||||
|
import com.alibaba.nacos.api.selector.AbstractSelector;
|
||||||
|
import com.alibaba.nacos.api.selector.ExpressionSelector;
|
||||||
|
import com.alibaba.nacos.api.selector.NoneSelector;
|
||||||
|
import com.alibaba.nacos.client.naming.remote.http.NamingHttpClientProxy;
|
||||||
|
import org.junit.After;
|
||||||
|
import org.junit.Assert;
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.mockito.ArgumentMatcher;
|
||||||
|
|
||||||
|
import java.lang.reflect.Field;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.Properties;
|
||||||
|
|
||||||
|
import static org.mockito.ArgumentMatchers.argThat;
|
||||||
|
import static org.mockito.Mockito.mock;
|
||||||
|
import static org.mockito.Mockito.times;
|
||||||
|
import static org.mockito.Mockito.verify;
|
||||||
|
|
||||||
|
public class NacosNamingMaintainServiceTest {
|
||||||
|
|
||||||
|
private NacosNamingMaintainService nacosNamingMaintainService;
|
||||||
|
|
||||||
|
private NamingHttpClientProxy serverProxy;
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void setUp() throws Exception {
|
||||||
|
Properties prop = new Properties();
|
||||||
|
prop.setProperty(PropertyKeyConst.NAMESPACE, "public");
|
||||||
|
nacosNamingMaintainService = new NacosNamingMaintainService(prop);
|
||||||
|
serverProxy = mock(NamingHttpClientProxy.class);
|
||||||
|
Field serverProxyField = NacosNamingMaintainService.class.getDeclaredField("serverProxy");
|
||||||
|
serverProxyField.setAccessible(true);
|
||||||
|
serverProxyField.set(nacosNamingMaintainService, serverProxy);
|
||||||
|
}
|
||||||
|
|
||||||
|
@After
|
||||||
|
public void tearDown() throws Exception {
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testConstructor() throws NacosException {
|
||||||
|
NacosNamingMaintainService client = new NacosNamingMaintainService("localhost");
|
||||||
|
Assert.assertNotNull(client);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testUpdateInstance1() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
Instance instance = new Instance();
|
||||||
|
//when
|
||||||
|
nacosNamingMaintainService.updateInstance(serviceName, groupName, instance);
|
||||||
|
//then
|
||||||
|
verify(serverProxy, times(1)).updateInstance(serviceName, groupName, instance);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testUpdateInstance2() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
Instance instance = new Instance();
|
||||||
|
//when
|
||||||
|
nacosNamingMaintainService.updateInstance(serviceName, instance);
|
||||||
|
//then
|
||||||
|
verify(serverProxy, times(1)).updateInstance(serviceName, Constants.DEFAULT_GROUP, instance);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testQueryService1() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
//when
|
||||||
|
nacosNamingMaintainService.queryService(serviceName, groupName);
|
||||||
|
//then
|
||||||
|
verify(serverProxy, times(1)).queryService(serviceName, groupName);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testQueryService2() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
Instance instance = new Instance();
|
||||||
|
//when
|
||||||
|
nacosNamingMaintainService.queryService(serviceName);
|
||||||
|
//then
|
||||||
|
verify(serverProxy, times(1)).queryService(serviceName, Constants.DEFAULT_GROUP);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testCreateService1() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
//when
|
||||||
|
nacosNamingMaintainService.createService(serviceName);
|
||||||
|
//then
|
||||||
|
verify(serverProxy, times(1)).createService(argThat(new ArgumentMatcher<Service>() {
|
||||||
|
@Override
|
||||||
|
public boolean matches(Service service) {
|
||||||
|
return service.getName().equals(serviceName) && service.getGroupName().equals(Constants.DEFAULT_GROUP)
|
||||||
|
&& Math.abs(service.getProtectThreshold() - Constants.DEFAULT_PROTECT_THRESHOLD) < 0.1f
|
||||||
|
&& service.getMetadata().size() == 0;
|
||||||
|
}
|
||||||
|
}), argThat(o -> o instanceof NoneSelector));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testCreateService2() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "groupName";
|
||||||
|
//when
|
||||||
|
nacosNamingMaintainService.createService(serviceName, groupName);
|
||||||
|
//then
|
||||||
|
verify(serverProxy, times(1)).createService(argThat(new ArgumentMatcher<Service>() {
|
||||||
|
@Override
|
||||||
|
public boolean matches(Service service) {
|
||||||
|
return service.getName().equals(serviceName) && service.getGroupName().equals(groupName)
|
||||||
|
&& Math.abs(service.getProtectThreshold() - Constants.DEFAULT_PROTECT_THRESHOLD) < 0.1f
|
||||||
|
&& service.getMetadata().size() == 0;
|
||||||
|
}
|
||||||
|
}), argThat(o -> o instanceof NoneSelector));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testCreateService3() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "groupName";
|
||||||
|
float protectThreshold = 0.1f;
|
||||||
|
//when
|
||||||
|
nacosNamingMaintainService.createService(serviceName, groupName, protectThreshold);
|
||||||
|
//then
|
||||||
|
verify(serverProxy, times(1)).createService(argThat(new ArgumentMatcher<Service>() {
|
||||||
|
@Override
|
||||||
|
public boolean matches(Service service) {
|
||||||
|
return service.getName().equals(serviceName) && service.getGroupName().equals(groupName)
|
||||||
|
&& Math.abs(service.getProtectThreshold() - protectThreshold) < 0.1f
|
||||||
|
&& service.getMetadata().size() == 0;
|
||||||
|
}
|
||||||
|
}), argThat(o -> o instanceof NoneSelector));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testCreateService5() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "groupName";
|
||||||
|
float protectThreshold = 0.1f;
|
||||||
|
String expression = "k=v";
|
||||||
|
//when
|
||||||
|
nacosNamingMaintainService.createService(serviceName, groupName, protectThreshold, expression);
|
||||||
|
//then
|
||||||
|
verify(serverProxy, times(1)).createService(argThat(new ArgumentMatcher<Service>() {
|
||||||
|
@Override
|
||||||
|
public boolean matches(Service service) {
|
||||||
|
return service.getName().equals(serviceName) && service.getGroupName().equals(groupName)
|
||||||
|
&& Math.abs(service.getProtectThreshold() - protectThreshold) < 0.1f
|
||||||
|
&& service.getMetadata().size() == 0;
|
||||||
|
}
|
||||||
|
}), argThat(o -> ((ExpressionSelector) o).getExpression().equals(expression)));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testCreateService4() throws NacosException {
|
||||||
|
//given
|
||||||
|
Service service = new Service();
|
||||||
|
AbstractSelector selector = new NoneSelector();
|
||||||
|
//when
|
||||||
|
nacosNamingMaintainService.createService(service, selector);
|
||||||
|
//then
|
||||||
|
verify(serverProxy, times(1)).createService(service, selector);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testDeleteService1() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
//when
|
||||||
|
nacosNamingMaintainService.deleteService(serviceName);
|
||||||
|
//then
|
||||||
|
verify(serverProxy, times(1)).deleteService(serviceName, Constants.DEFAULT_GROUP);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testDeleteService2() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "groupName";
|
||||||
|
//when
|
||||||
|
nacosNamingMaintainService.deleteService(serviceName, groupName);
|
||||||
|
//then
|
||||||
|
verify(serverProxy, times(1)).deleteService(serviceName, groupName);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testUpdateService1() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "groupName";
|
||||||
|
float protectThreshold = 0.1f;
|
||||||
|
|
||||||
|
//when
|
||||||
|
nacosNamingMaintainService.updateService(serviceName, groupName, protectThreshold);
|
||||||
|
//then
|
||||||
|
verify(serverProxy, times(1)).updateService(argThat(new ArgumentMatcher<Service>() {
|
||||||
|
@Override
|
||||||
|
public boolean matches(Service service) {
|
||||||
|
return service.getName().equals(serviceName) && service.getGroupName().equals(groupName)
|
||||||
|
&& Math.abs(service.getProtectThreshold() - protectThreshold) < 0.1f;
|
||||||
|
}
|
||||||
|
}), argThat(o -> o instanceof NoneSelector));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testUpdateService2() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "groupName";
|
||||||
|
float protectThreshold = 0.1f;
|
||||||
|
Map<String, String> meta = new HashMap<>();
|
||||||
|
meta.put("k", "v");
|
||||||
|
|
||||||
|
//when
|
||||||
|
nacosNamingMaintainService.updateService(serviceName, groupName, protectThreshold, meta);
|
||||||
|
//then
|
||||||
|
verify(serverProxy, times(1)).updateService(argThat(new ArgumentMatcher<Service>() {
|
||||||
|
@Override
|
||||||
|
public boolean matches(Service service) {
|
||||||
|
return service.getName().equals(serviceName) && service.getGroupName().equals(groupName)
|
||||||
|
&& Math.abs(service.getProtectThreshold() - protectThreshold) < 0.1f
|
||||||
|
&& service.getMetadata().size() == 1;
|
||||||
|
}
|
||||||
|
}), argThat(o -> o instanceof NoneSelector));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testUpdateService3() throws NacosException {
|
||||||
|
//given
|
||||||
|
Service service = new Service();
|
||||||
|
AbstractSelector selector = new NoneSelector();
|
||||||
|
//when
|
||||||
|
nacosNamingMaintainService.updateService(service, selector);
|
||||||
|
//then
|
||||||
|
verify(serverProxy, times(1)).updateService(service, selector);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testShutDown() throws NacosException {
|
||||||
|
nacosNamingMaintainService.shutDown();
|
||||||
|
//then
|
||||||
|
verify(serverProxy, times(1)).shutdown();
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,924 @@
|
|||||||
|
/*
|
||||||
|
*
|
||||||
|
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* 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.alibaba.nacos.client.naming;
|
||||||
|
|
||||||
|
import com.alibaba.nacos.api.PropertyKeyConst;
|
||||||
|
import com.alibaba.nacos.api.common.Constants;
|
||||||
|
import com.alibaba.nacos.api.exception.NacosException;
|
||||||
|
import com.alibaba.nacos.api.naming.listener.Event;
|
||||||
|
import com.alibaba.nacos.api.naming.listener.EventListener;
|
||||||
|
import com.alibaba.nacos.api.naming.pojo.Instance;
|
||||||
|
import com.alibaba.nacos.api.naming.pojo.ServiceInfo;
|
||||||
|
import com.alibaba.nacos.api.selector.AbstractSelector;
|
||||||
|
import com.alibaba.nacos.client.naming.event.InstancesChangeNotifier;
|
||||||
|
import com.alibaba.nacos.client.naming.remote.NamingClientProxy;
|
||||||
|
import com.alibaba.nacos.client.naming.remote.http.NamingHttpClientProxy;
|
||||||
|
import org.junit.Assert;
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.mockito.ArgumentMatcher;
|
||||||
|
|
||||||
|
import java.lang.reflect.Field;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.Arrays;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Properties;
|
||||||
|
|
||||||
|
import static org.mockito.ArgumentMatchers.anyBoolean;
|
||||||
|
import static org.mockito.ArgumentMatchers.anyInt;
|
||||||
|
import static org.mockito.ArgumentMatchers.anyString;
|
||||||
|
import static org.mockito.ArgumentMatchers.argThat;
|
||||||
|
import static org.mockito.ArgumentMatchers.eq;
|
||||||
|
import static org.mockito.Mockito.mock;
|
||||||
|
import static org.mockito.Mockito.times;
|
||||||
|
import static org.mockito.Mockito.verify;
|
||||||
|
import static org.mockito.Mockito.when;
|
||||||
|
|
||||||
|
public class NacosNamingServiceTest {
|
||||||
|
|
||||||
|
private NacosNamingService client;
|
||||||
|
|
||||||
|
private NamingClientProxy proxy;
|
||||||
|
|
||||||
|
private InstancesChangeNotifier changeNotifier;
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void before() throws NoSuchFieldException, NacosException, IllegalAccessException {
|
||||||
|
Properties prop = new Properties();
|
||||||
|
prop.put(PropertyKeyConst.NAMESPACE, "test");
|
||||||
|
client = new NacosNamingService(prop);
|
||||||
|
// inject proxy
|
||||||
|
proxy = mock(NamingHttpClientProxy.class);
|
||||||
|
Field serverProxyField = NacosNamingService.class.getDeclaredField("clientProxy");
|
||||||
|
serverProxyField.setAccessible(true);
|
||||||
|
serverProxyField.set(client, proxy);
|
||||||
|
// inject notifier
|
||||||
|
changeNotifier = mock(InstancesChangeNotifier.class);
|
||||||
|
Field changeNotifierField = NacosNamingService.class.getDeclaredField("changeNotifier");
|
||||||
|
changeNotifierField.setAccessible(true);
|
||||||
|
changeNotifierField.set(client, changeNotifier);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRegisterInstance1() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String ip = "1.1.1.1";
|
||||||
|
int port = 10000;
|
||||||
|
//when
|
||||||
|
client.registerInstance(serviceName, ip, port);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1))
|
||||||
|
.registerService(eq(serviceName), eq(Constants.DEFAULT_GROUP), argThat(new ArgumentMatcher<Instance>() {
|
||||||
|
@Override
|
||||||
|
public boolean matches(Instance instance) {
|
||||||
|
return instance.getIp().equals(ip) && instance.getPort() == port
|
||||||
|
&& Math.abs(instance.getWeight() - 1.0) < 0.01f && instance.getClusterName()
|
||||||
|
.equals(Constants.DEFAULT_CLUSTER_NAME);
|
||||||
|
}
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRegisterInstance2() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
String ip = "1.1.1.1";
|
||||||
|
int port = 10000;
|
||||||
|
//when
|
||||||
|
client.registerInstance(serviceName, groupName, ip, port);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1))
|
||||||
|
.registerService(eq(serviceName), eq(groupName), argThat(new ArgumentMatcher<Instance>() {
|
||||||
|
@Override
|
||||||
|
public boolean matches(Instance instance) {
|
||||||
|
return instance.getIp().equals(ip) && instance.getPort() == port
|
||||||
|
&& Math.abs(instance.getWeight() - 1.0) < 0.01f && instance.getClusterName()
|
||||||
|
.equals(Constants.DEFAULT_CLUSTER_NAME);
|
||||||
|
}
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRegisterInstance3() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String clusterName = "cluster1";
|
||||||
|
String ip = "1.1.1.1";
|
||||||
|
int port = 10000;
|
||||||
|
//when
|
||||||
|
client.registerInstance(serviceName, ip, port, clusterName);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1))
|
||||||
|
.registerService(eq(serviceName), eq(Constants.DEFAULT_GROUP), argThat(new ArgumentMatcher<Instance>() {
|
||||||
|
@Override
|
||||||
|
public boolean matches(Instance instance) {
|
||||||
|
return instance.getIp().equals(ip) && instance.getPort() == port
|
||||||
|
&& Math.abs(instance.getWeight() - 1.0) < 0.01f && instance.getClusterName()
|
||||||
|
.equals(clusterName);
|
||||||
|
}
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRegisterInstance4() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
String clusterName = "cluster1";
|
||||||
|
String ip = "1.1.1.1";
|
||||||
|
int port = 10000;
|
||||||
|
//when
|
||||||
|
client.registerInstance(serviceName, groupName, ip, port, clusterName);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1))
|
||||||
|
.registerService(eq(serviceName), eq(groupName), argThat(new ArgumentMatcher<Instance>() {
|
||||||
|
@Override
|
||||||
|
public boolean matches(Instance instance) {
|
||||||
|
return instance.getIp().equals(ip) && instance.getPort() == port
|
||||||
|
&& Math.abs(instance.getWeight() - 1.0) < 0.01f && instance.getClusterName()
|
||||||
|
.equals(clusterName);
|
||||||
|
}
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRegisterInstance5() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
Instance instance = new Instance();
|
||||||
|
//when
|
||||||
|
client.registerInstance(serviceName, instance);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).registerService(serviceName, Constants.DEFAULT_GROUP, instance);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRegisterInstance6() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
Instance instance = new Instance();
|
||||||
|
//when
|
||||||
|
client.registerInstance(serviceName, groupName, instance);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).registerService(serviceName, groupName, instance);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testDeregisterInstance1() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String ip = "1.1.1.1";
|
||||||
|
int port = 10000;
|
||||||
|
//when
|
||||||
|
client.deregisterInstance(serviceName, ip, port);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).deregisterService(eq(serviceName), eq(Constants.DEFAULT_GROUP),
|
||||||
|
argThat(new ArgumentMatcher<Instance>() {
|
||||||
|
@Override
|
||||||
|
public boolean matches(Instance instance) {
|
||||||
|
return instance.getIp().equals(ip) && instance.getPort() == port
|
||||||
|
&& Math.abs(instance.getWeight() - 1.0) < 0.01f && instance.getClusterName()
|
||||||
|
.equals(Constants.DEFAULT_CLUSTER_NAME);
|
||||||
|
}
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testDeregisterInstance2() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
String ip = "1.1.1.1";
|
||||||
|
int port = 10000;
|
||||||
|
//when
|
||||||
|
client.deregisterInstance(serviceName, groupName, ip, port);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1))
|
||||||
|
.deregisterService(eq(serviceName), eq(groupName), argThat(new ArgumentMatcher<Instance>() {
|
||||||
|
@Override
|
||||||
|
public boolean matches(Instance instance) {
|
||||||
|
return instance.getIp().equals(ip) && instance.getPort() == port
|
||||||
|
&& Math.abs(instance.getWeight() - 1.0) < 0.01f && instance.getClusterName()
|
||||||
|
.equals(Constants.DEFAULT_CLUSTER_NAME);
|
||||||
|
}
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testDeregisterInstance3() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String clusterName = "cluster1";
|
||||||
|
String ip = "1.1.1.1";
|
||||||
|
int port = 10000;
|
||||||
|
//when
|
||||||
|
client.deregisterInstance(serviceName, ip, port, clusterName);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).deregisterService(eq(serviceName), eq(Constants.DEFAULT_GROUP),
|
||||||
|
argThat(new ArgumentMatcher<Instance>() {
|
||||||
|
@Override
|
||||||
|
public boolean matches(Instance instance) {
|
||||||
|
return instance.getIp().equals(ip) && instance.getPort() == port
|
||||||
|
&& Math.abs(instance.getWeight() - 1.0) < 0.01f && instance.getClusterName()
|
||||||
|
.equals(clusterName);
|
||||||
|
}
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testDeregisterInstance4() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
String clusterName = "cluster1";
|
||||||
|
String ip = "1.1.1.1";
|
||||||
|
int port = 10000;
|
||||||
|
//when
|
||||||
|
client.deregisterInstance(serviceName, groupName, ip, port, clusterName);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1))
|
||||||
|
.deregisterService(eq(serviceName), eq(groupName), argThat(new ArgumentMatcher<Instance>() {
|
||||||
|
@Override
|
||||||
|
public boolean matches(Instance instance) {
|
||||||
|
return instance.getIp().equals(ip) && instance.getPort() == port
|
||||||
|
&& Math.abs(instance.getWeight() - 1.0) < 0.01f && instance.getClusterName()
|
||||||
|
.equals(clusterName);
|
||||||
|
}
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testDeregisterInstance5() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
Instance instance = new Instance();
|
||||||
|
//when
|
||||||
|
client.deregisterInstance(serviceName, instance);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).deregisterService(serviceName, Constants.DEFAULT_GROUP, instance);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testDeregisterInstance6() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
Instance instance = new Instance();
|
||||||
|
//when
|
||||||
|
client.deregisterInstance(serviceName, groupName, instance);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).deregisterService(serviceName, groupName, instance);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetAllInstances1() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
//when
|
||||||
|
client.getAllInstances(serviceName);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, Constants.DEFAULT_GROUP, "");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetAllInstances2() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
//when
|
||||||
|
client.getAllInstances(serviceName, groupName);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, groupName, "");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetAllInstances3() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
//when
|
||||||
|
client.getAllInstances(serviceName, false);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).queryInstancesOfService(serviceName, Constants.DEFAULT_GROUP, "", 0, false);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetAllInstances4() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
//when
|
||||||
|
client.getAllInstances(serviceName, groupName, false);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).queryInstancesOfService(serviceName, groupName, "", 0, false);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetAllInstances5() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
//when
|
||||||
|
client.getAllInstances(serviceName, clusterList);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, Constants.DEFAULT_GROUP, "cluster1,cluster2");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetAllInstances6() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
//when
|
||||||
|
client.getAllInstances(serviceName, groupName, clusterList);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, groupName, "cluster1,cluster2");
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetAllInstances7() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
//when
|
||||||
|
client.getAllInstances(serviceName, clusterList, false);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1))
|
||||||
|
.queryInstancesOfService(serviceName, Constants.DEFAULT_GROUP, "cluster1,cluster2", 0, false);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetAllInstances8() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
//when
|
||||||
|
client.getAllInstances(serviceName, groupName, clusterList, false);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).queryInstancesOfService(serviceName, groupName, "cluster1,cluster2", 0, false);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectInstances1() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
//when
|
||||||
|
client.selectInstances(serviceName, true);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, Constants.DEFAULT_GROUP, "");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectInstances2() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
//when
|
||||||
|
client.selectInstances(serviceName, groupName, true);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, groupName, "");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectInstances3() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
//when
|
||||||
|
client.selectInstances(serviceName, true, false);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).queryInstancesOfService(serviceName, Constants.DEFAULT_GROUP, "", 0, false);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectInstances4() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
//when
|
||||||
|
client.selectInstances(serviceName, groupName, true, false);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).queryInstancesOfService(serviceName, groupName, "", 0, false);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectInstances5() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
//when
|
||||||
|
client.selectInstances(serviceName, clusterList, true);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, Constants.DEFAULT_GROUP, "cluster1,cluster2");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectInstances6() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
//when
|
||||||
|
client.selectInstances(serviceName, groupName, clusterList, true);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, groupName, "cluster1,cluster2");
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectInstances7() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
//when
|
||||||
|
client.selectInstances(serviceName, clusterList, true, false);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1))
|
||||||
|
.queryInstancesOfService(serviceName, Constants.DEFAULT_GROUP, "cluster1,cluster2", 0, false);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectInstances8() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
//when
|
||||||
|
client.selectInstances(serviceName, groupName, clusterList, true, false);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).queryInstancesOfService(serviceName, groupName, "cluster1,cluster2", 0, false);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectInstancesWithHealthyFlag() throws NacosException {
|
||||||
|
//given
|
||||||
|
Instance healthyInstance = new Instance();
|
||||||
|
healthyInstance.setHealthy(true);
|
||||||
|
|
||||||
|
Instance instance1 = new Instance();
|
||||||
|
instance1.setHealthy(false);
|
||||||
|
|
||||||
|
Instance instance2 = new Instance();
|
||||||
|
instance2.setHealthy(true);
|
||||||
|
instance2.setEnabled(false);
|
||||||
|
Instance instance3 = new Instance();
|
||||||
|
instance3.setHealthy(true);
|
||||||
|
instance3.setWeight(0.0);
|
||||||
|
|
||||||
|
List<Instance> hosts = new ArrayList<>();
|
||||||
|
hosts.add(healthyInstance);
|
||||||
|
hosts.add(instance1);
|
||||||
|
hosts.add(instance2);
|
||||||
|
hosts.add(instance3);
|
||||||
|
|
||||||
|
ServiceInfo info = new ServiceInfo();
|
||||||
|
info.setHosts(hosts);
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
when(proxy.queryInstancesOfService(serviceName, groupName, "cluster1,cluster2", 0, false)).thenReturn(info);
|
||||||
|
|
||||||
|
//when
|
||||||
|
List<Instance> instances = client.selectInstances(serviceName, groupName, clusterList, true, false);
|
||||||
|
//then
|
||||||
|
Assert.assertEquals(1, instances.size());
|
||||||
|
Assert.assertSame(healthyInstance, instances.get(0));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectOneHealthyInstance1() throws NacosException {
|
||||||
|
//given
|
||||||
|
Instance healthyInstance = new Instance();
|
||||||
|
healthyInstance.setIp("1.1.1.1");
|
||||||
|
healthyInstance.setPort(1000);
|
||||||
|
List<Instance> hosts = new ArrayList<>();
|
||||||
|
hosts.add(healthyInstance);
|
||||||
|
ServiceInfo infoWithHealthyInstance = new ServiceInfo();
|
||||||
|
infoWithHealthyInstance.setHosts(hosts);
|
||||||
|
when(proxy.subscribe(anyString(), anyString(), anyString())).thenReturn(infoWithHealthyInstance);
|
||||||
|
|
||||||
|
String serviceName = "service1";
|
||||||
|
//when
|
||||||
|
client.selectOneHealthyInstance(serviceName);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, Constants.DEFAULT_GROUP, "");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectOneHealthyInstance2() throws NacosException {
|
||||||
|
//given
|
||||||
|
Instance healthyInstance = new Instance();
|
||||||
|
healthyInstance.setIp("1.1.1.1");
|
||||||
|
healthyInstance.setPort(1000);
|
||||||
|
List<Instance> hosts = new ArrayList<>();
|
||||||
|
hosts.add(healthyInstance);
|
||||||
|
ServiceInfo infoWithHealthyInstance = new ServiceInfo();
|
||||||
|
infoWithHealthyInstance.setHosts(hosts);
|
||||||
|
when(proxy.subscribe(anyString(), anyString(), anyString())).thenReturn(infoWithHealthyInstance);
|
||||||
|
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
//when
|
||||||
|
client.selectOneHealthyInstance(serviceName, groupName);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, groupName, "");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectOneHealthyInstance3() throws NacosException {
|
||||||
|
//given
|
||||||
|
Instance healthyInstance = new Instance();
|
||||||
|
healthyInstance.setIp("1.1.1.1");
|
||||||
|
healthyInstance.setPort(1000);
|
||||||
|
List<Instance> hosts = new ArrayList<>();
|
||||||
|
hosts.add(healthyInstance);
|
||||||
|
ServiceInfo infoWithHealthyInstance = new ServiceInfo();
|
||||||
|
infoWithHealthyInstance.setHosts(hosts);
|
||||||
|
when(proxy.queryInstancesOfService(anyString(), anyString(), anyString(), anyInt(), anyBoolean()))
|
||||||
|
.thenReturn(infoWithHealthyInstance);
|
||||||
|
|
||||||
|
String serviceName = "service1";
|
||||||
|
//when
|
||||||
|
client.selectOneHealthyInstance(serviceName, false);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).queryInstancesOfService(serviceName, Constants.DEFAULT_GROUP, "", 0, false);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectOneHealthyInstance4() throws NacosException {
|
||||||
|
//given
|
||||||
|
Instance healthyInstance = new Instance();
|
||||||
|
healthyInstance.setIp("1.1.1.1");
|
||||||
|
healthyInstance.setPort(1000);
|
||||||
|
List<Instance> hosts = new ArrayList<>();
|
||||||
|
hosts.add(healthyInstance);
|
||||||
|
ServiceInfo infoWithHealthyInstance = new ServiceInfo();
|
||||||
|
infoWithHealthyInstance.setHosts(hosts);
|
||||||
|
when(proxy.queryInstancesOfService(anyString(), anyString(), anyString(), anyInt(), anyBoolean()))
|
||||||
|
.thenReturn(infoWithHealthyInstance);
|
||||||
|
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
//when
|
||||||
|
client.selectOneHealthyInstance(serviceName, groupName, false);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).queryInstancesOfService(serviceName, groupName, "", 0, false);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectOneHealthyInstance5() throws NacosException {
|
||||||
|
//given
|
||||||
|
Instance healthyInstance = new Instance();
|
||||||
|
healthyInstance.setIp("1.1.1.1");
|
||||||
|
healthyInstance.setPort(1000);
|
||||||
|
List<Instance> hosts = new ArrayList<>();
|
||||||
|
hosts.add(healthyInstance);
|
||||||
|
ServiceInfo infoWithHealthyInstance = new ServiceInfo();
|
||||||
|
infoWithHealthyInstance.setHosts(hosts);
|
||||||
|
when(proxy.subscribe(anyString(), anyString(), anyString())).thenReturn(infoWithHealthyInstance);
|
||||||
|
|
||||||
|
String serviceName = "service1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
//when
|
||||||
|
client.selectOneHealthyInstance(serviceName, clusterList);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, Constants.DEFAULT_GROUP, "cluster1,cluster2");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectOneHealthyInstance6() throws NacosException {
|
||||||
|
//given
|
||||||
|
Instance healthyInstance = new Instance();
|
||||||
|
healthyInstance.setIp("1.1.1.1");
|
||||||
|
healthyInstance.setPort(1000);
|
||||||
|
List<Instance> hosts = new ArrayList<>();
|
||||||
|
hosts.add(healthyInstance);
|
||||||
|
ServiceInfo infoWithHealthyInstance = new ServiceInfo();
|
||||||
|
infoWithHealthyInstance.setHosts(hosts);
|
||||||
|
when(proxy.subscribe(anyString(), anyString(), anyString())).thenReturn(infoWithHealthyInstance);
|
||||||
|
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
//when
|
||||||
|
client.selectOneHealthyInstance(serviceName, groupName, clusterList);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, groupName, "cluster1,cluster2");
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectOneHealthyInstance7() throws NacosException {
|
||||||
|
//given
|
||||||
|
Instance healthyInstance = new Instance();
|
||||||
|
healthyInstance.setIp("1.1.1.1");
|
||||||
|
healthyInstance.setPort(1000);
|
||||||
|
List<Instance> hosts = new ArrayList<>();
|
||||||
|
hosts.add(healthyInstance);
|
||||||
|
ServiceInfo infoWithHealthyInstance = new ServiceInfo();
|
||||||
|
infoWithHealthyInstance.setHosts(hosts);
|
||||||
|
when(proxy.queryInstancesOfService(anyString(), anyString(), anyString(), anyInt(), anyBoolean()))
|
||||||
|
.thenReturn(infoWithHealthyInstance);
|
||||||
|
|
||||||
|
String serviceName = "service1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
//when
|
||||||
|
client.selectOneHealthyInstance(serviceName, clusterList, false);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1))
|
||||||
|
.queryInstancesOfService(serviceName, Constants.DEFAULT_GROUP, "cluster1,cluster2", 0, false);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectOneHealthyInstance8() throws NacosException {
|
||||||
|
//given
|
||||||
|
Instance healthyInstance = new Instance();
|
||||||
|
healthyInstance.setIp("1.1.1.1");
|
||||||
|
healthyInstance.setPort(1000);
|
||||||
|
List<Instance> hosts = new ArrayList<>();
|
||||||
|
hosts.add(healthyInstance);
|
||||||
|
ServiceInfo infoWithHealthyInstance = new ServiceInfo();
|
||||||
|
infoWithHealthyInstance.setHosts(hosts);
|
||||||
|
when(proxy.queryInstancesOfService(anyString(), anyString(), anyString(), anyInt(), anyBoolean()))
|
||||||
|
.thenReturn(infoWithHealthyInstance);
|
||||||
|
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
//when
|
||||||
|
client.selectOneHealthyInstance(serviceName, groupName, clusterList, false);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).queryInstancesOfService(serviceName, groupName, "cluster1,cluster2", 0, false);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSubscribe1() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
EventListener listener = new EventListener() {
|
||||||
|
@Override
|
||||||
|
public void onEvent(Event event) {
|
||||||
|
|
||||||
|
}
|
||||||
|
};
|
||||||
|
//when
|
||||||
|
client.subscribe(serviceName, listener);
|
||||||
|
//then
|
||||||
|
verify(changeNotifier, times(1)).registerListener(Constants.DEFAULT_GROUP, serviceName, "", listener);
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, Constants.DEFAULT_GROUP, "");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSubscribe2() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
EventListener listener = new EventListener() {
|
||||||
|
@Override
|
||||||
|
public void onEvent(Event event) {
|
||||||
|
|
||||||
|
}
|
||||||
|
};
|
||||||
|
//when
|
||||||
|
client.subscribe(serviceName, groupName, listener);
|
||||||
|
//then
|
||||||
|
verify(changeNotifier, times(1)).registerListener(groupName, serviceName, "", listener);
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, groupName, "");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSubscribe3() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
EventListener listener = new EventListener() {
|
||||||
|
@Override
|
||||||
|
public void onEvent(Event event) {
|
||||||
|
|
||||||
|
}
|
||||||
|
};
|
||||||
|
//when
|
||||||
|
client.subscribe(serviceName, clusterList, listener);
|
||||||
|
//then
|
||||||
|
verify(changeNotifier, times(1))
|
||||||
|
.registerListener(Constants.DEFAULT_GROUP, serviceName, "cluster1,cluster2", listener);
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, Constants.DEFAULT_GROUP, "cluster1,cluster2");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSubscribe4() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
EventListener listener = new EventListener() {
|
||||||
|
@Override
|
||||||
|
public void onEvent(Event event) {
|
||||||
|
|
||||||
|
}
|
||||||
|
};
|
||||||
|
//when
|
||||||
|
client.subscribe(serviceName, groupName, clusterList, listener);
|
||||||
|
//then
|
||||||
|
verify(changeNotifier, times(1)).registerListener(groupName, serviceName, "cluster1,cluster2", listener);
|
||||||
|
verify(proxy, times(1)).subscribe(serviceName, groupName, "cluster1,cluster2");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testUnSubscribe1() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
EventListener listener = new EventListener() {
|
||||||
|
@Override
|
||||||
|
public void onEvent(Event event) {
|
||||||
|
|
||||||
|
}
|
||||||
|
};
|
||||||
|
when(changeNotifier.isSubscribed(serviceName, Constants.DEFAULT_GROUP, "")).thenReturn(false);
|
||||||
|
//when
|
||||||
|
client.unsubscribe(serviceName, listener);
|
||||||
|
//then
|
||||||
|
verify(changeNotifier, times(1)).deregisterListener(Constants.DEFAULT_GROUP, serviceName, "", listener);
|
||||||
|
verify(proxy, times(1)).unsubscribe(serviceName, Constants.DEFAULT_GROUP, "");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testUnSubscribe2() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
EventListener listener = new EventListener() {
|
||||||
|
@Override
|
||||||
|
public void onEvent(Event event) {
|
||||||
|
|
||||||
|
}
|
||||||
|
};
|
||||||
|
when(changeNotifier.isSubscribed(serviceName, groupName, "")).thenReturn(false);
|
||||||
|
|
||||||
|
//when
|
||||||
|
client.unsubscribe(serviceName, groupName, listener);
|
||||||
|
//then
|
||||||
|
verify(changeNotifier, times(1)).deregisterListener(groupName, serviceName, "", listener);
|
||||||
|
verify(proxy, times(1)).unsubscribe(serviceName, groupName, "");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testUnSubscribe3() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
EventListener listener = new EventListener() {
|
||||||
|
@Override
|
||||||
|
public void onEvent(Event event) {
|
||||||
|
|
||||||
|
}
|
||||||
|
};
|
||||||
|
when(changeNotifier.isSubscribed(serviceName, Constants.DEFAULT_GROUP, "cluster1,cluster2")).thenReturn(false);
|
||||||
|
|
||||||
|
//when
|
||||||
|
client.unsubscribe(serviceName, clusterList, listener);
|
||||||
|
//then
|
||||||
|
verify(changeNotifier, times(1))
|
||||||
|
.deregisterListener(Constants.DEFAULT_GROUP, serviceName, "cluster1,cluster2", listener);
|
||||||
|
verify(proxy, times(1)).unsubscribe(serviceName, Constants.DEFAULT_GROUP, "cluster1,cluster2");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testUnSubscribe4() throws NacosException {
|
||||||
|
//given
|
||||||
|
String serviceName = "service1";
|
||||||
|
String groupName = "group1";
|
||||||
|
List<String> clusterList = Arrays.asList("cluster1", "cluster2");
|
||||||
|
EventListener listener = new EventListener() {
|
||||||
|
@Override
|
||||||
|
public void onEvent(Event event) {
|
||||||
|
|
||||||
|
}
|
||||||
|
};
|
||||||
|
when(changeNotifier.isSubscribed(serviceName, groupName, "cluster1,cluster2")).thenReturn(false);
|
||||||
|
|
||||||
|
//when
|
||||||
|
client.unsubscribe(serviceName, groupName, clusterList, listener);
|
||||||
|
//then
|
||||||
|
verify(changeNotifier, times(1)).deregisterListener(groupName, serviceName, "cluster1,cluster2", listener);
|
||||||
|
verify(proxy, times(1)).unsubscribe(serviceName, groupName, "cluster1,cluster2");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetServicesOfServer1() throws NacosException {
|
||||||
|
//given
|
||||||
|
int pageNo = 1;
|
||||||
|
int pageSize = 10;
|
||||||
|
//when
|
||||||
|
client.getServicesOfServer(pageNo, pageSize);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).getServiceList(pageNo, pageSize, Constants.DEFAULT_GROUP, null);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetServicesOfServer2() throws NacosException {
|
||||||
|
//given
|
||||||
|
int pageNo = 1;
|
||||||
|
int pageSize = 10;
|
||||||
|
String groupName = "group1";
|
||||||
|
//when
|
||||||
|
client.getServicesOfServer(pageNo, pageSize, groupName);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).getServiceList(pageNo, pageSize, groupName, null);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetServicesOfServer3() throws NacosException {
|
||||||
|
//given
|
||||||
|
int pageNo = 1;
|
||||||
|
int pageSize = 10;
|
||||||
|
AbstractSelector selector = new AbstractSelector("aaa") {
|
||||||
|
@Override
|
||||||
|
public String getType() {
|
||||||
|
return super.getType();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
//when
|
||||||
|
client.getServicesOfServer(pageNo, pageSize, selector);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).getServiceList(pageNo, pageSize, Constants.DEFAULT_GROUP, selector);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetServicesOfServer4() throws NacosException {
|
||||||
|
//given
|
||||||
|
int pageNo = 1;
|
||||||
|
int pageSize = 10;
|
||||||
|
String groupName = "group1";
|
||||||
|
|
||||||
|
AbstractSelector selector = new AbstractSelector("aaa") {
|
||||||
|
@Override
|
||||||
|
public String getType() {
|
||||||
|
return super.getType();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
//when
|
||||||
|
client.getServicesOfServer(pageNo, pageSize, groupName, selector);
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).getServiceList(pageNo, pageSize, groupName, selector);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetSubscribeServices() {
|
||||||
|
//when
|
||||||
|
client.getSubscribeServices();
|
||||||
|
//then
|
||||||
|
verify(changeNotifier, times(1)).getSubscribeServices();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetServerStatus() {
|
||||||
|
//given
|
||||||
|
when(proxy.serverHealthy()).thenReturn(true);
|
||||||
|
//when
|
||||||
|
String serverStatus = client.getServerStatus();
|
||||||
|
//then
|
||||||
|
Assert.assertEquals("UP", serverStatus);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetServerStatusFail() {
|
||||||
|
//given
|
||||||
|
when(proxy.serverHealthy()).thenReturn(false);
|
||||||
|
//when
|
||||||
|
String serverStatus = client.getServerStatus();
|
||||||
|
//then
|
||||||
|
Assert.assertEquals("DOWN", serverStatus);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testShutDown() throws NacosException {
|
||||||
|
//when
|
||||||
|
client.shutDown();
|
||||||
|
//then
|
||||||
|
verify(proxy, times(1)).shutdown();
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user