2 * Copyright (c) 2020 Pantheon Technologies s.r.o. 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
9 package org.opendaylight.netconf.mdsal.notification.impl;
11 import static org.mockito.ArgumentMatchers.any;
12 import static org.mockito.Mockito.doNothing;
13 import static org.mockito.Mockito.doReturn;
14 import static org.mockito.Mockito.mock;
15 import static org.mockito.Mockito.verify;
17 import java.util.Collections;
18 import org.junit.Before;
19 import org.junit.Test;
20 import org.junit.runner.RunWith;
21 import org.mockito.Mock;
22 import org.mockito.junit.MockitoJUnitRunner;
23 import org.opendaylight.mdsal.binding.api.DataBroker;
24 import org.opendaylight.mdsal.binding.api.DataObjectModification;
25 import org.opendaylight.mdsal.binding.api.DataTreeModification;
26 import org.opendaylight.netconf.notifications.NetconfNotificationCollector;
27 import org.opendaylight.netconf.notifications.YangLibraryPublisherRegistration;
28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Capabilities;
29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.ModulesState;
30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.ModulesStateBuilder;
31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.YangLibraryChange;
32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.YangLibraryChangeBuilder;
34 @RunWith(MockitoJUnitRunner.StrictStubs.class)
35 public class YangLibraryNotificationProducerTest {
37 private YangLibraryNotificationProducer yangLibraryNotificationProducer;
40 private YangLibraryPublisherRegistration yangLibraryPublisherRegistration;
43 private NetconfNotificationCollector netconfNotificationCollector;
45 private DataBroker dataBroker;
49 doNothing().when(yangLibraryPublisherRegistration).onYangLibraryChange(any(YangLibraryChange.class));
50 doReturn(yangLibraryPublisherRegistration).when(netconfNotificationCollector)
51 .registerYangLibraryPublisher();
53 yangLibraryNotificationProducer = new YangLibraryNotificationProducer(netconfNotificationCollector,
58 public void testOnDataTreeChanged() {
59 final String moduleSetId = "1";
60 ModulesState modulesStateAfter = new ModulesStateBuilder().setModuleSetId(moduleSetId).build();
62 final DataTreeModification<ModulesState> treeChange = mock(DataTreeModification.class);
63 final DataObjectModification<Capabilities> objectChange = mock(DataObjectModification.class);
64 doReturn(objectChange).when(treeChange).getRootNode();
65 doReturn(modulesStateAfter).when(objectChange).getDataAfter();
67 YangLibraryChange yangLibraryChange = new YangLibraryChangeBuilder().setModuleSetId(moduleSetId).build();
68 yangLibraryNotificationProducer.onDataTreeChanged(Collections.singleton(treeChange));
70 verify(yangLibraryPublisherRegistration).onYangLibraryChange(yangLibraryChange);