From 9e7137d28302c37acf4bce6bfb95c8e892e08d29 Mon Sep 17 00:00:00 2001 From: masterh Date: Tue, 7 Aug 2018 10:15:45 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0AbstractRegistry=E6=B5=8B?= =?UTF-8?q?=E8=AF=95=E7=94=A8=E4=BE=8B=20Unit=20Test=20for=20dubbo-registr?= =?UTF-8?q?y/dubbo-registry-api=20#2177?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../support/AbstractRegistryTest.java | 218 ++++++++++++++++++ 1 file changed, 218 insertions(+) create mode 100644 dubbo-registry/dubbo-registry-api/src/test/java/org/apache/dubbo/registry/support/AbstractRegistryTest.java diff --git a/dubbo-registry/dubbo-registry-api/src/test/java/org/apache/dubbo/registry/support/AbstractRegistryTest.java b/dubbo-registry/dubbo-registry-api/src/test/java/org/apache/dubbo/registry/support/AbstractRegistryTest.java new file mode 100644 index 00000000000..10880ccadf8 --- /dev/null +++ b/dubbo-registry/dubbo-registry-api/src/test/java/org/apache/dubbo/registry/support/AbstractRegistryTest.java @@ -0,0 +1,218 @@ +package org.apache.dubbo.registry.support; + +import org.apache.dubbo.common.URL; +import org.apache.dubbo.registry.NotifyListener; +import org.hamcrest.Matchers; +import org.junit.Assert; +import org.junit.Test; + +import java.util.*; +import java.util.concurrent.atomic.AtomicReference; + +public class AbstractRegistryTest { + private static final Map parametersProvider = new LinkedHashMap<>(); + private static final Map parametersConsumer = new LinkedHashMap<>(); + + static { + parametersProvider.put("anyhost", "true"); + parametersProvider.put("application", "demo-provider"); + parametersProvider.put("dubbo", "2.0.2"); + parametersProvider.put("generic", "false"); + parametersProvider.put("interface", "org.apache.dubbo.demo.DemoService"); + parametersProvider.put("methods", "sayHello"); + parametersProvider.put("pid", "1489"); + parametersProvider.put("side", "provider"); + parametersProvider.put("timestamp", String.valueOf(System.currentTimeMillis())); + + parametersConsumer.put("application", "demo-consumer"); + parametersConsumer.put("category", "consumer"); + parametersConsumer.put("check", "false"); + parametersConsumer.put("dubbo", "2.0.2"); + parametersConsumer.put("interface", "org.apache.dubbo.demo.DemoService"); + parametersConsumer.put("methods", "sayHello"); + parametersConsumer.put("pid", "1676"); + parametersConsumer.put("qos.port", "333333"); + parametersConsumer.put("side", "consumer"); + parametersConsumer.put("timestamp", String.valueOf(System.currentTimeMillis())); + } + + private URL mockUrl = new URL("dubbo", "127.0.0.0", 2200); + private AbstractRegistry abstractRegistry = new AbstractRegistry(mockUrl) { + @Override + public boolean isAvailable() { + return false; + } + }; + + private List getList() { + List list = new ArrayList<>(); + URL url_1 = new URL("dubbo", "127.0.0.0", 1000); + URL url_2 = new URL("dubbo", "127.0.0.1", 1001); + URL url_3 = new URL("dubbo", "127.0.0.2", 1002); + list.add(url_1); + list.add(url_2); + list.add(url_3); + return list; + } + + + /** + * 测试注册方法 + * 传入单个url进行注册，判断该url是否在已经注册成功的set里 + * 传入url list，判断已经注册成功的set大小是否和url list大小相同 + */ + @Test + public void testRegister() { + //测试单个注册 + abstractRegistry.register(mockUrl); + assert abstractRegistry.getRegistered().contains(mockUrl); + + //测试多个注册 + abstractRegistry.getRegistered().clear(); + List urlList = getList(); + for (URL url : urlList) { + abstractRegistry.register(url); + } + Assert.assertThat(abstractRegistry.getRegistered().size(), Matchers.equalTo(urlList.size())); + } + + /** + * 测试注销注册方法 + * 传入单个url，注销注册，判断已经注册的url列表里是否包含该url + * 传入url 列表，注销注册，判断已经注册的列表是否被清空 + */ + @Test + public void testUnregister() { + //测试单个注销注册 + URL url = new URL("dubbo", "127.0.0.0", 2200); + abstractRegistry.getRegistered().add(url); + abstractRegistry.unregister(url); + Assert.assertThat(false, Matchers.equalTo(abstractRegistry.getRegistered().contains(url))); + + //测试多个注销注册 + abstractRegistry.getRegistered().clear(); + + List urlList = getList(); + for (URL urlSub : urlList) { + abstractRegistry.getRegistered().add(urlSub); + } + for (URL urlSub : urlList) { + abstractRegistry.unregister(urlSub); + } + Assert.assertThat(0, Matchers.equalTo(abstractRegistry.getRegistered().size())); + + } + + /** + * 测试订阅和取消订阅方法 + * 传入url进行订阅，判断相应的listener是否加入监听set + * 传入url取消订阅，判断相应的listener是否从监听set移除 + */ + @Test + public void testSubscribeAndUnsubscribe() { + + //订阅测试 + final AtomicReference notified = new AtomicReference(false); + NotifyListener listener = urls -> notified.set(Boolean.TRUE); + URL url = new URL("dubbo", "127.0.0.0", 2200); + abstractRegistry.subscribe(url, listener); + Set subscribeListeners = abstractRegistry.getSubscribed().get(url); + Assert.assertThat(true, Matchers.equalTo(subscribeListeners.contains(listener))); + + //取消订阅测试 + abstractRegistry.unsubscribe(url, listener); + Set unsubscribeListeners = abstractRegistry.getSubscribed().get(url); + Assert.assertThat(false, Matchers.equalTo(unsubscribeListeners.contains(listener))); + } + + /** + * 测试recover()方法的逻辑正确性 + *

+ * ZookeeperRegistry: + * zkClient创建成功以后，会监听RECONNECTED事件， + * recover方法主要做一件事：将需要暴露的url放在failedRegistered（Set）中， + * 将需要订阅的服务放在failedSubscribed（Set）中。 + * 说明RECONNECTED时，因为所有需要暴露的服务都需要重新注册， + * 所以其实是将需要暴露、订阅的url都放到failedRegistered、failedSubscribed中。 + */ + @Test + public void testRecover() { + List list = getList(); + try { + abstractRegistry.recover(); + Assert.assertEquals(0, abstractRegistry.getRegistered().size()); + for (URL url : list) { + abstractRegistry.register(url); + } + Assert.assertEquals(3, abstractRegistry.getRegistered().size()); + abstractRegistry.recover(); + Assert.assertEquals(3, abstractRegistry.getRegistered().size()); + } catch (Exception e) { + e.printStackTrace(); + } + } + + + /** + * 测试提醒方法，传入url list + * 判断url list是否在已经提醒的set + */ + @Test + public void testNotifyList() { + final AtomicReference notified = new AtomicReference(false); + NotifyListener listner1 = urls -> notified.set(Boolean.TRUE); + URL url1 = new URL("dubbo", "127.0.0.0", 2200, parametersConsumer); + abstractRegistry.subscribe(url1, listner1); + + NotifyListener listner2 = urls -> notified.set(Boolean.TRUE); + URL url2 = new URL("dubbo", "127.0.0.1", 2201, parametersConsumer); + abstractRegistry.subscribe(url2, listner2); + + NotifyListener listner3 = urls -> notified.set(Boolean.TRUE); + URL url3 = new URL("dubbo", "127.0.0.2", 2202, parametersConsumer); + abstractRegistry.subscribe(url3, listner3); + + List urls = new ArrayList<>(); + urls.add(url1); + urls.add(url2); + urls.add(url3); + abstractRegistry.notify(urls); + Map>> map = abstractRegistry.getNotified(); + + Assert.assertThat(true, Matchers.equalTo(map.containsKey(url1))); + Assert.assertThat(true, Matchers.equalTo(map.containsKey(url2))); + Assert.assertThat(true, Matchers.equalTo(map.containsKey(url3))); + + } + + /** + * 测试提醒方法，传入url，listner，url list参数 + * 判断该url是否在已经提醒的set + */ + @Test + public void testNotify() { + final AtomicReference notified = new AtomicReference(false); + NotifyListener listner1 = urls -> notified.set(Boolean.TRUE); + URL url1 = new URL("dubbo", "127.0.0.0", 2200, parametersConsumer); + abstractRegistry.subscribe(url1, listner1); + + NotifyListener listner2 = urls -> notified.set(Boolean.TRUE); + URL url2 = new URL("dubbo", "127.0.0.1", 2201, parametersConsumer); + abstractRegistry.subscribe(url2, listner2); + + NotifyListener listner3 = urls -> notified.set(Boolean.TRUE); + URL url3 = new URL("dubbo", "127.0.0.2", 2202, parametersConsumer); + abstractRegistry.subscribe(url3, listner3); + + List urls = new ArrayList<>(); + urls.add(url1); + urls.add(url2); + urls.add(url3); + abstractRegistry.notify(url1, listner1, urls); + Map>> map = abstractRegistry.getNotified(); + + Assert.assertThat(true, Matchers.equalTo(map.containsKey(url1))); + Assert.assertThat(false, Matchers.equalTo(map.containsKey(url2))); + Assert.assertThat(false, Matchers.equalTo(map.containsKey(url3))); + } +}