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;
10 import static org.mockito.ArgumentMatchers.any;
11 import static org.mockito.Mockito.doNothing;
12 import static org.mockito.Mockito.doReturn;
13 import static org.mockito.Mockito.mock;
14 import static org.mockito.Mockito.verify;
16 import java.util.Collections;
17 import org.junit.Before;
18 import org.junit.Test;
19 import org.mockito.Mock;
20 import org.mockito.MockitoAnnotations;
21 import org.opendaylight.mdsal.binding.api.DataBroker;
22 import org.opendaylight.mdsal.binding.api.DataObjectModification;
23 import org.opendaylight.mdsal.binding.api.DataTreeModification;
24 import org.opendaylight.netconf.notifications.NetconfNotificationCollector;
25 import org.opendaylight.netconf.notifications.YangLibraryPublisherRegistration;
26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Capabilities;
27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev160621.ModulesState;
28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev160621.ModulesStateBuilder;
29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev160621.YangLibraryChange;
30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev160621.YangLibraryChangeBuilder;
32 public class YangLibraryNotificationProducerTest {
34 private YangLibraryNotificationProducer yangLibraryNotificationProducer;
37 private YangLibraryPublisherRegistration yangLibraryPublisherRegistration;
40 private NetconfNotificationCollector netconfNotificationCollector;
42 private DataBroker dataBroker;
45 public void setUp() throws Exception {
46 MockitoAnnotations.initMocks(this);
48 doNothing().when(yangLibraryPublisherRegistration).onYangLibraryChange(any(YangLibraryChange.class));
49 doReturn(yangLibraryPublisherRegistration).when(netconfNotificationCollector)
50 .registerYangLibraryPublisher();
52 yangLibraryNotificationProducer = new YangLibraryNotificationProducer(netconfNotificationCollector,
57 public void testOnDataTreeChanged() {
58 final String moduleSetId = "1";
59 ModulesState modulesStateAfter = new ModulesStateBuilder().setModuleSetId(moduleSetId).build();
61 final DataTreeModification<ModulesState> treeChange = mock(DataTreeModification.class);
62 final DataObjectModification<Capabilities> objectChange = mock(DataObjectModification.class);
63 doReturn(objectChange).when(treeChange).getRootNode();
64 doReturn(modulesStateAfter).when(objectChange).getDataAfter();
66 YangLibraryChange yangLibraryChange = new YangLibraryChangeBuilder().setModuleSetId(moduleSetId).build();
67 yangLibraryNotificationProducer.onDataTreeChanged(Collections.singleton(treeChange));
69 verify(yangLibraryPublisherRegistration).onYangLibraryChange(yangLibraryChange);