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.northbound.monitoring;
10 import static org.mockito.ArgumentMatchers.any;
11 import static org.mockito.ArgumentMatchers.eq;
12 import static org.mockito.Mockito.doNothing;
13 import static org.mockito.Mockito.doReturn;
14 import static org.mockito.Mockito.inOrder;
15 import static org.mockito.Mockito.verify;
17 import java.util.List;
18 import org.junit.jupiter.api.BeforeEach;
19 import org.junit.jupiter.api.Test;
20 import org.junit.jupiter.api.extension.ExtendWith;
21 import org.mockito.InOrder;
22 import org.mockito.Mock;
23 import org.mockito.junit.jupiter.MockitoExtension;
24 import org.opendaylight.mdsal.binding.api.DataBroker;
25 import org.opendaylight.mdsal.binding.api.WriteTransaction;
26 import org.opendaylight.mdsal.common.api.CommitInfo;
27 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
28 import org.opendaylight.netconf.server.api.monitoring.NetconfMonitoringService;
29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.NetconfState;
30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Capabilities;
31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.CapabilitiesBuilder;
32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Schemas;
33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.SchemasBuilder;
34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Sessions;
35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.sessions.Session;
36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.sessions.SessionBuilder;
37 import org.opendaylight.yangtools.concepts.Registration;
38 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
39 import org.opendaylight.yangtools.yang.common.Uint32;
41 @ExtendWith(MockitoExtension.class)
42 class MonitoringToMdsalWriterTest {
44 private NetconfMonitoringService monitoring;
46 private DataBroker dataBroker;
48 private WriteTransaction writeTransaction;
50 private Registration capabilityReg;
52 private Registration sessionsReg;
54 private MonitoringToMdsalWriter writer;
58 doReturn(capabilityReg).when(monitoring).registerCapabilitiesListener(any());
59 doReturn(sessionsReg).when(monitoring).registerSessionsListener(any());
61 writer = new MonitoringToMdsalWriter(dataBroker, monitoring);
66 doNothing().when(capabilityReg).close();
67 doNothing().when(sessionsReg).close();
68 doReturn(writeTransaction).when(dataBroker).newWriteOnlyTransaction();
69 doReturn(CommitInfo.emptyFluentFuture()).when(writeTransaction).commit();
72 final var inOrder = inOrder(capabilityReg, sessionsReg, writeTransaction);
73 inOrder.verify(sessionsReg).close();
74 inOrder.verify(capabilityReg).close();
75 inOrder.verify(writeTransaction).delete(LogicalDatastoreType.OPERATIONAL,
76 InstanceIdentifier.create(NetconfState.class));
77 inOrder.verify(writeTransaction).commit();
81 void testOnCapabilityChanged() {
82 final var capabilitiesId = InstanceIdentifier.create(NetconfState.class).child(Capabilities.class);
83 final var capabilities = new CapabilitiesBuilder().build();
84 doReturn(writeTransaction).when(dataBroker).newWriteOnlyTransaction();
85 doNothing().when(writeTransaction).put(eq(LogicalDatastoreType.OPERATIONAL), any(), any());
86 doReturn(CommitInfo.emptyFluentFuture()).when(writeTransaction).commit();
88 writer.onCapabilitiesChanged(capabilities);
89 InOrder inOrder = inOrder(writeTransaction);
90 inOrder.verify(writeTransaction).put(LogicalDatastoreType.OPERATIONAL, capabilitiesId, capabilities);
91 inOrder.verify(writeTransaction).commit();
95 void testOnSchemasChanged() {
96 final var schemasId = InstanceIdentifier.create(NetconfState.class).child(Schemas.class);
97 final var schemas = new SchemasBuilder().build();
98 doReturn(writeTransaction).when(dataBroker).newWriteOnlyTransaction();
99 doNothing().when(writeTransaction).put(eq(LogicalDatastoreType.OPERATIONAL), any(), any());
100 doReturn(CommitInfo.emptyFluentFuture()).when(writeTransaction).commit();
102 writer.onSchemasChanged(schemas);
103 final var inOrder = inOrder(writeTransaction);
104 inOrder.verify(writeTransaction).put(LogicalDatastoreType.OPERATIONAL, schemasId, schemas);
105 inOrder.verify(writeTransaction).commit();
109 void testOnSessionStart() {
110 final var session = new SessionBuilder().setSessionId(Uint32.ONE).build();
111 doReturn(writeTransaction).when(dataBroker).newWriteOnlyTransaction();
112 doNothing().when(writeTransaction).put(eq(LogicalDatastoreType.OPERATIONAL), any(), any());
113 doReturn(CommitInfo.emptyFluentFuture()).when(writeTransaction).commit();
115 writer.onSessionStarted(session);
116 final var inOrder = inOrder(writeTransaction);
117 inOrder.verify(writeTransaction).put(LogicalDatastoreType.OPERATIONAL,
118 InstanceIdentifier.create(NetconfState.class).child(Sessions.class).child(Session.class, session.key()),
120 inOrder.verify(writeTransaction).commit();
124 void testOnSessionEnd() {
125 final var session = new SessionBuilder().setSessionId(Uint32.ONE).build();
126 doReturn(writeTransaction).when(dataBroker).newWriteOnlyTransaction();
127 doNothing().when(writeTransaction).delete(eq(LogicalDatastoreType.OPERATIONAL), any());
128 doReturn(CommitInfo.emptyFluentFuture()).when(writeTransaction).commit();
130 writer.onSessionEnded(session);
131 final var inOrder = inOrder(writeTransaction);
132 inOrder.verify(writeTransaction).delete(LogicalDatastoreType.OPERATIONAL,
133 InstanceIdentifier.create(NetconfState.class).child(Sessions.class).child(Session.class, session.key()));
134 inOrder.verify(writeTransaction).commit();
138 void testOnSessionsUpdated() {
139 final var session1 = new SessionBuilder().setSessionId(Uint32.ONE).build();
140 final var session2 = new SessionBuilder().setSessionId(Uint32.TWO).build();
141 doReturn(writeTransaction).when(dataBroker).newWriteOnlyTransaction();
142 doNothing().when(writeTransaction).put(eq(LogicalDatastoreType.OPERATIONAL), any(), any());
143 doReturn(CommitInfo.emptyFluentFuture()).when(writeTransaction).commit();
145 writer.onSessionsUpdated(List.of(session1, session2));
146 final var inOrder = inOrder(writeTransaction);
147 inOrder.verify(writeTransaction).put(LogicalDatastoreType.OPERATIONAL,
148 InstanceIdentifier.create(NetconfState.class).child(Sessions.class).child(Session.class, session1.key()),
150 inOrder.verify(writeTransaction).put(LogicalDatastoreType.OPERATIONAL,
151 InstanceIdentifier.create(NetconfState.class).child(Sessions.class).child(Session.class, session2.key()),
153 inOrder.verify(writeTransaction).commit();
157 void testOnSessionInitiated() {
158 verify(monitoring).registerCapabilitiesListener(writer);