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.netconf.messagebus.eventsources.netconf;
10 import static org.hamcrest.CoreMatchers.hasItems;
11 import static org.hamcrest.MatcherAssert.assertThat;
12 import static org.junit.Assert.assertFalse;
13 import static org.junit.Assert.assertNotNull;
14 import static org.junit.Assert.assertTrue;
15 import static org.mockito.ArgumentMatchers.any;
16 import static org.mockito.Mockito.doReturn;
17 import static org.mockito.Mockito.verify;
18 import static org.mockito.Mockito.when;
19 import static org.opendaylight.yangtools.util.concurrent.FluentFutures.immediateNullFluentFuture;
21 import java.time.Instant;
22 import java.util.Optional;
24 import org.junit.Before;
25 import org.junit.Test;
26 import org.junit.runner.RunWith;
27 import org.mockito.Mock;
28 import org.mockito.junit.MockitoJUnitRunner;
29 import org.opendaylight.mdsal.dom.api.DOMNotificationListener;
30 import org.opendaylight.mdsal.dom.api.DOMNotificationService;
31 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventaggregator.rev141202.TopicId;
32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netconf.notification._1._0.rev080714.StreamNameType;
33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netmod.notification.rev080714.netconf.streams.Stream;
34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netmod.notification.rev080714.netconf.streams.StreamBuilder;
35 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
36 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
37 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
38 import org.opendaylight.yangtools.concepts.ListenerRegistration;
40 @RunWith(MockitoJUnitRunner.StrictStubs.class)
41 public class StreamNotificationTopicRegistrationTest {
43 private static final String STREAM_NAME = "stream-1";
44 private static final String PREFIX = ConnectionNotificationTopicRegistration.EVENT_SOURCE_STATUS_PATH
45 .lastNodeIdentifier().getNamespace().toString();
48 private NetconfEventSource source;
50 private NetconfEventSourceMount mount;
52 private DOMNotificationService reference;
54 private ListenerRegistration<DOMNotificationListener> listenerRegistration;
56 private StreamNotificationTopicRegistration registration;
57 private Stream stream;
60 public void setUp() throws Exception {
61 Node node = new NodeBuilder().setNodeId(NodeId.getDefaultInstance("node-id")).build();
62 when(mount.getNode()).thenReturn(node);
63 when(mount.registerNotificationListener(source,
64 ConnectionNotificationTopicRegistration.EVENT_SOURCE_STATUS_PATH.asSchemaPath()))
65 .thenReturn(listenerRegistration);
66 doReturn(immediateNullFluentFuture()).when(mount).invokeCreateSubscription(any(), any());
67 doReturn(immediateNullFluentFuture()).when(mount).invokeCreateSubscription(any());
69 when(source.getMount()).thenReturn(mount);
70 stream = new StreamBuilder().setName(StreamNameType.getDefaultInstance(STREAM_NAME)).setReplaySupport(true)
73 registration = new StreamNotificationTopicRegistration(stream, PREFIX, source);
77 public void testActivateNotificationSource() throws Exception {
78 registration.activateNotificationSource();
79 assertTrue(registration.isActive());
80 verify(mount).invokeCreateSubscription(stream);
85 public void testReActivateNotificationSource() throws Exception {
86 registration.setActive(true);
87 registration.reActivateNotificationSource();
89 assertTrue(registration.isActive());
90 verify(mount).invokeCreateSubscription(stream, Optional.empty());
94 public void testReActivateNotificationSourceWithReplay() throws Exception {
95 final Instant lastEventTime = Instant.now();
96 registration.setActive(true);
97 registration.setLastEventTime(lastEventTime);
98 registration.reActivateNotificationSource();
100 assertTrue(registration.isActive());
101 verify(mount).invokeCreateSubscription(stream, Optional.of(lastEventTime));
105 public void testClose() throws Exception {
106 registration.setActive(true);
107 registration.close();
108 assertFalse(registration.isActive());
112 public void testRegisterAndUnregisterNotificationTopic() throws Exception {
113 final TopicId topic1 = registerTopic("topic1");
114 final TopicId topic2 = registerTopic("topic2");
115 final TopicId topic3 = registerTopic("topic3");
116 final Set<TopicId> notificationTopicIds = registration.getTopicsForNotification(
117 ConnectionNotificationTopicRegistration.EVENT_SOURCE_STATUS_PATH.asSchemaPath());
118 assertNotNull(notificationTopicIds);
119 assertThat(notificationTopicIds, hasItems(topic1, topic2, topic3));
121 registration.unRegisterNotificationTopic(topic3);
122 final Set<TopicId> afterUnregister = registration.getTopicsForNotification(
123 ConnectionNotificationTopicRegistration.EVENT_SOURCE_STATUS_PATH.asSchemaPath());
124 assertNotNull(afterUnregister);
125 assertThat(afterUnregister, hasItems(topic1, topic2));
126 assertFalse(afterUnregister.contains(topic3));
129 private TopicId registerTopic(final String value) {
130 final TopicId topic = TopicId.getDefaultInstance(value);
131 registration.registerNotificationTopic(
132 ConnectionNotificationTopicRegistration.EVENT_SOURCE_STATUS_PATH.asSchemaPath(), topic);