Bug 3864: Notify netconf monitoring about changes in session
[netconf.git] / netconf / mdsal-netconf-monitoring / src / test / java / org / opendaylight / controller / config / yang / netconf / mdsal / monitoring / MonitoringToMdsalWriterTest.java
1 /*
2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
3  *
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
7  */
8
9 package org.opendaylight.controller.config.yang.netconf.mdsal.monitoring;
10
11 import static org.mockito.Matchers.any;
12 import static org.mockito.Matchers.eq;
13 import static org.mockito.Mockito.doNothing;
14 import static org.mockito.Mockito.doReturn;
15 import static org.mockito.Mockito.inOrder;
16 import static org.mockito.Mockito.verify;
17
18 import com.google.common.util.concurrent.Futures;
19 import java.util.ArrayList;
20 import java.util.Collections;
21 import java.util.List;
22 import org.junit.Before;
23 import org.junit.Test;
24 import org.mockito.InOrder;
25 import org.mockito.Mock;
26 import org.mockito.MockitoAnnotations;
27 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
28 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
29 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
30 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
31 import org.opendaylight.netconf.api.monitoring.NetconfMonitoringService;
32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.NetconfState;
33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Capabilities;
34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.CapabilitiesBuilder;
35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Schemas;
36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.SchemasBuilder;
37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Sessions;
38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.sessions.Session;
39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.sessions.SessionBuilder;
40 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
41
42 public class MonitoringToMdsalWriterTest {
43
44     private static final InstanceIdentifier<NetconfState> INSTANCE_IDENTIFIER = InstanceIdentifier.create(NetconfState.class);
45
46     @Mock
47     private NetconfMonitoringService monitoring;
48     @Mock
49     private BindingAwareBroker.ProviderContext context;
50     @Mock
51     private DataBroker dataBroker;
52     @Mock
53     private WriteTransaction writeTransaction;
54
55     private MonitoringToMdsalWriter writer;
56
57     @Before
58     public void setUp() throws Exception {
59         MockitoAnnotations.initMocks(this);
60
61         doReturn(null).when(monitoring).registerCapabilitiesListener(any());
62         doReturn(null).when(monitoring).registerSessionsListener(any());
63
64         doReturn(dataBroker).when(context).getSALService(DataBroker.class);
65
66         doReturn(writeTransaction).when(dataBroker).newWriteOnlyTransaction();
67
68         doNothing().when(writeTransaction).put(eq(LogicalDatastoreType.OPERATIONAL), any(), any());
69         doNothing().when(writeTransaction).delete(eq(LogicalDatastoreType.OPERATIONAL), any());
70         doReturn(Futures.immediateCheckedFuture(null)).when(writeTransaction).submit();
71
72         writer = new MonitoringToMdsalWriter(monitoring);
73     }
74
75     @Test
76     public void testClose() throws Exception {
77         writer.onSessionInitiated(context);
78         writer.close();
79         InOrder inOrder = inOrder(writeTransaction);
80         inOrder.verify(writeTransaction).delete(LogicalDatastoreType.OPERATIONAL, INSTANCE_IDENTIFIER);
81         inOrder.verify(writeTransaction).submit();
82     }
83
84     @Test
85     public void testOnCapabilityChanged() throws Exception {
86         final InstanceIdentifier<Capabilities> capabilitiesId = InstanceIdentifier.create(NetconfState.class).child(Capabilities.class);
87         writer.onSessionInitiated(context);
88         final Capabilities capabilities = new CapabilitiesBuilder().build();
89         writer.onCapabilitiesChanged(capabilities);
90         InOrder inOrder = inOrder(writeTransaction);
91         inOrder.verify(writeTransaction).put(LogicalDatastoreType.OPERATIONAL, capabilitiesId, capabilities);
92         inOrder.verify(writeTransaction).submit();
93     }
94
95     @Test
96     public void testOnSchemasChanged() throws Exception {
97         final InstanceIdentifier<Schemas> schemasId = InstanceIdentifier.create(NetconfState.class).child(Schemas.class);
98         writer.onSessionInitiated(context);
99         final Schemas schemas = new SchemasBuilder().build();
100         writer.onSchemasChanged(schemas);
101         InOrder inOrder = inOrder(writeTransaction);
102         inOrder.verify(writeTransaction).put(LogicalDatastoreType.OPERATIONAL, schemasId, schemas);
103         inOrder.verify(writeTransaction).submit();
104     }
105
106     @Test
107     public void testOnSessionStart() throws Exception {
108         Session session = new SessionBuilder()
109                 .setSessionId(1L)
110                 .build();
111         final InstanceIdentifier<Session> id = InstanceIdentifier.create(NetconfState.class).child(Sessions.class).child(Session.class, session.getKey());
112         writer.onSessionInitiated(context);
113         writer.onSessionStarted(session);
114         InOrder inOrder = inOrder(writeTransaction);
115         inOrder.verify(writeTransaction).put(LogicalDatastoreType.OPERATIONAL, id, session);
116         inOrder.verify(writeTransaction).submit();
117     }
118
119     @Test
120     public void testOnSessionEnd() throws Exception {
121         Session session = new SessionBuilder()
122                 .setSessionId(1L)
123                 .build();
124         final InstanceIdentifier<Session> id = InstanceIdentifier.create(NetconfState.class).child(Sessions.class).child(Session.class, session.getKey());
125         writer.onSessionInitiated(context);
126         writer.onSessionEnded(session);
127         InOrder inOrder = inOrder(writeTransaction);
128         inOrder.verify(writeTransaction).delete(LogicalDatastoreType.OPERATIONAL, id);
129         inOrder.verify(writeTransaction).submit();
130     }
131
132     @Test
133     public void testOnSessionsUpdated() throws Exception {
134         Session session1 = new SessionBuilder()
135                 .setSessionId(1L)
136                 .build();
137         Session session2 = new SessionBuilder()
138                 .setSessionId(2L)
139                 .build();
140         List<Session> sessions = new ArrayList<>();
141         sessions.add(session1);
142         sessions.add(session2);
143         final InstanceIdentifier<Session> id1 = InstanceIdentifier.create(NetconfState.class).child(Sessions.class).child(Session.class, session1.getKey());
144         final InstanceIdentifier<Session> id2 = InstanceIdentifier.create(NetconfState.class).child(Sessions.class).child(Session.class, session2.getKey());
145         writer.onSessionInitiated(context);
146         writer.onSessionsUpdated(sessions);
147         InOrder inOrder = inOrder(writeTransaction);
148         inOrder.verify(writeTransaction).put(LogicalDatastoreType.OPERATIONAL, id1, session1);
149         inOrder.verify(writeTransaction).put(LogicalDatastoreType.OPERATIONAL, id2, session2);
150         inOrder.verify(writeTransaction).submit();
151     }
152
153     @Test
154     public void testOnSessionInitiated() throws Exception {
155         writer.onSessionInitiated(context);
156         verify(monitoring).registerCapabilitiesListener(writer);
157     }
158 }