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
8 package org.opendaylight.netconf.server.mdsal.notifications;
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.verify;
15 import org.junit.Before;
16 import org.junit.Test;
17 import org.junit.runner.RunWith;
18 import org.mockito.Mock;
19 import org.mockito.junit.MockitoJUnitRunner;
20 import org.opendaylight.mdsal.binding.api.DataBroker;
21 import org.opendaylight.netconf.server.api.notifications.NetconfNotificationCollector;
22 import org.opendaylight.netconf.server.api.notifications.YangLibraryPublisherRegistration;
23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.ModulesStateBuilder;
24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.YangLibraryChange;
25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.YangLibraryChangeBuilder;
26 import org.opendaylight.yangtools.concepts.Registration;
28 @RunWith(MockitoJUnitRunner.StrictStubs.class)
29 public class YangLibraryNotificationProducerTest {
31 private YangLibraryPublisherRegistration yangLibraryPublisherRegistration;
33 private NetconfNotificationCollector netconfNotificationCollector;
35 private DataBroker dataBroker;
37 private Registration registration;
39 private YangLibraryNotificationProducer yangLibraryNotificationProducer;
43 doNothing().when(yangLibraryPublisherRegistration).onYangLibraryChange(any(YangLibraryChange.class));
44 doReturn(yangLibraryPublisherRegistration).when(netconfNotificationCollector)
45 .registerYangLibraryPublisher();
46 doReturn(registration).when(dataBroker).registerDataListener(any(), any());
48 yangLibraryNotificationProducer = new YangLibraryNotificationProducer(netconfNotificationCollector, dataBroker);
52 public void testOnDataTreeChanged() {
53 final String moduleSetId = "1";
54 final var modulesStateAfter = new ModulesStateBuilder().setModuleSetId(moduleSetId).build();
56 yangLibraryNotificationProducer.dataChangedTo(modulesStateAfter);
58 final var yangLibraryChange = new YangLibraryChangeBuilder().setModuleSetId(moduleSetId).build();
59 verify(yangLibraryPublisherRegistration).onYangLibraryChange(yangLibraryChange);