2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.mdsal.dom.broker;
10 import static org.junit.Assert.assertFalse;
11 import static org.junit.Assert.assertNotNull;
12 import static org.junit.Assert.assertTrue;
13 import static org.mockito.Mockito.doReturn;
14 import static org.mockito.Mockito.mock;
16 import com.google.common.collect.Multimap;
17 import java.lang.reflect.Field;
18 import java.util.concurrent.ExecutorService;
19 import java.util.concurrent.TimeUnit;
20 import javax.annotation.Nonnull;
21 import org.junit.Test;
22 import org.opendaylight.mdsal.dom.api.DOMNotification;
23 import org.opendaylight.mdsal.dom.api.DOMNotificationListener;
24 import org.opendaylight.mdsal.dom.spi.DOMNotificationSubscriptionListener;
25 import org.opendaylight.yangtools.concepts.ListenerRegistration;
26 import org.opendaylight.yangtools.util.ListenerRegistry;
27 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
29 public class DOMNotificationRouterTest extends TestUtils {
32 public void create() throws Exception {
33 assertNotNull(DOMNotificationRouter.create(1,1,1,TimeUnit.SECONDS));
34 assertNotNull(DOMNotificationRouter.create(1));
38 public void complexTest() throws Exception {
39 final DOMNotificationSubscriptionListener domNotificationSubscriptionListener =
40 mock(DOMNotificationSubscriptionListener.class);
41 final DOMNotificationListener domNotificationListener = new TestListener();
42 final DOMNotificationRouter domNotificationRouter = DOMNotificationRouter.create(1);
44 final Field listenersField = DOMNotificationRouter.class.getDeclaredField("listeners");
45 listenersField.setAccessible(true);
47 final Field subscriptionListenersField = DOMNotificationRouter.class.getDeclaredField("subscriptionListeners");
48 subscriptionListenersField.setAccessible(true);
50 Multimap<SchemaPath, ListenerRegistration<? extends DOMNotificationListener>> listeners =
51 (Multimap<SchemaPath, ListenerRegistration<? extends DOMNotificationListener>>)
52 listenersField.get(domNotificationRouter);
54 assertTrue(listeners.isEmpty());
55 assertNotNull(domNotificationRouter.registerNotificationListener(domNotificationListener, SchemaPath.ROOT));
56 assertNotNull(domNotificationRouter.registerNotificationListener(domNotificationListener, SchemaPath.SAME));
58 listeners = (Multimap<SchemaPath, ListenerRegistration<? extends DOMNotificationListener>>)
59 listenersField.get(domNotificationRouter);
61 assertFalse(listeners.isEmpty());
63 ListenerRegistry<DOMNotificationSubscriptionListener> subscriptionListeners =
64 (ListenerRegistry<DOMNotificationSubscriptionListener>)
65 subscriptionListenersField.get(domNotificationRouter);
67 assertFalse(subscriptionListeners.iterator().hasNext());
68 assertNotNull(domNotificationRouter.registerSubscriptionListener(domNotificationSubscriptionListener));
70 subscriptionListeners = (ListenerRegistry<DOMNotificationSubscriptionListener>)
71 subscriptionListenersField.get(domNotificationRouter);
72 assertTrue(subscriptionListeners.iterator().hasNext());
74 final DOMNotification domNotification = mock(DOMNotification.class);
75 doReturn("test").when(domNotification).toString();
76 doReturn(SchemaPath.ROOT).when(domNotification).getType();
77 doReturn(TEST_CHILD).when(domNotification).getBody();
79 assertNotNull(domNotificationRouter.offerNotification(domNotification));
82 assertNotNull(domNotificationRouter.offerNotification(domNotification, 1, TimeUnit.SECONDS));
83 assertNotNull(domNotificationRouter.offerNotification(domNotification, 1, TimeUnit.SECONDS));
84 } catch (Exception e) {
85 assertTrue(e instanceof UnsupportedOperationException);
88 assertNotNull(domNotificationRouter.putNotification(domNotification));
92 public void offerNotification() throws Exception {
93 final DOMNotificationRouter domNotificationRouter = DOMNotificationRouter.create(1);
94 final DOMNotification domNotification = mock(DOMNotification.class);
95 doReturn(SchemaPath.ROOT).when(domNotification).getType();
96 doReturn(TEST_CHILD).when(domNotification).getBody();
97 assertNotNull(domNotificationRouter.putNotification(domNotification));
98 assertNotNull(domNotificationRouter.offerNotification(domNotification));
99 assertNotNull(domNotificationRouter.offerNotification(domNotification, 1, TimeUnit.SECONDS));
103 public void close() throws Exception {
104 final DOMNotificationRouter domNotificationRouter = DOMNotificationRouter.create(1);
106 final Field executorField = DOMNotificationRouter.class.getDeclaredField("executor");
107 executorField.setAccessible(true);
109 final ExecutorService executor = (ExecutorService) executorField.get(domNotificationRouter);
111 assertFalse(executor.isShutdown());
112 domNotificationRouter.close();
113 assertTrue(executor.isShutdown());
116 private class TestListener implements DOMNotificationListener {
118 public void onNotification(@Nonnull DOMNotification notification) {}