import static org.mockito.Mockito.verify;
import static org.opendaylight.mdsal.common.api.CommitInfo.emptyFluentFuture;
-import java.util.ArrayList;
import java.util.List;
import org.junit.Before;
import org.junit.Test;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Sessions;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.sessions.Session;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.sessions.SessionBuilder;
+import org.opendaylight.yangtools.concepts.Registration;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.opendaylight.yangtools.yang.common.Uint32;
private DataBroker dataBroker;
@Mock
private WriteTransaction writeTransaction;
+ @Mock
+ private Registration capabilityReg;
+ @Mock
+ private Registration sessionsReg;
private MonitoringToMdsalWriter writer;
@Before
public void setUp() {
- doReturn(null).when(monitoring).registerCapabilitiesListener(any());
- doReturn(null).when(monitoring).registerSessionsListener(any());
+ doReturn(capabilityReg).when(monitoring).registerCapabilitiesListener(any());
+ doReturn(sessionsReg).when(monitoring).registerSessionsListener(any());
doReturn(writeTransaction).when(dataBroker).newWriteOnlyTransaction();
doNothing().when(writeTransaction).delete(eq(LogicalDatastoreType.OPERATIONAL), any());
doReturn(emptyFluentFuture()).when(writeTransaction).commit();
- writer = new MonitoringToMdsalWriter(monitoring, dataBroker);
+ writer = new MonitoringToMdsalWriter(dataBroker, monitoring);
}
@Test
public void testClose() throws Exception {
- writer.start();
+ doNothing().when(capabilityReg).close();
+ doNothing().when(sessionsReg).close();
writer.close();
- InOrder inOrder = inOrder(writeTransaction);
+ InOrder inOrder = inOrder(capabilityReg, sessionsReg, writeTransaction);
+ inOrder.verify(sessionsReg).close();
+ inOrder.verify(capabilityReg).close();
inOrder.verify(writeTransaction).delete(LogicalDatastoreType.OPERATIONAL, INSTANCE_IDENTIFIER);
inOrder.verify(writeTransaction).commit();
}
public void testOnCapabilityChanged() throws Exception {
final InstanceIdentifier<Capabilities> capabilitiesId =
InstanceIdentifier.create(NetconfState.class).child(Capabilities.class);
- writer.start();
final Capabilities capabilities = new CapabilitiesBuilder().build();
writer.onCapabilitiesChanged(capabilities);
InOrder inOrder = inOrder(writeTransaction);
public void testOnSchemasChanged() throws Exception {
final InstanceIdentifier<Schemas> schemasId =
InstanceIdentifier.create(NetconfState.class).child(Schemas.class);
- writer.start();
final Schemas schemas = new SchemasBuilder().build();
writer.onSchemasChanged(schemas);
InOrder inOrder = inOrder(writeTransaction);
InstanceIdentifier.create(NetconfState.class)
.child(Sessions.class)
.child(Session.class, session.key());
- writer.start();
writer.onSessionStarted(session);
InOrder inOrder = inOrder(writeTransaction);
inOrder.verify(writeTransaction).put(LogicalDatastoreType.OPERATIONAL, id, session);
InstanceIdentifier.create(NetconfState.class)
.child(Sessions.class)
.child(Session.class, session.key());
- writer.start();
writer.onSessionEnded(session);
InOrder inOrder = inOrder(writeTransaction);
inOrder.verify(writeTransaction).delete(LogicalDatastoreType.OPERATIONAL, id);
Session session2 = new SessionBuilder()
.setSessionId(Uint32.valueOf(2))
.build();
- List<Session> sessions = new ArrayList<>();
- sessions.add(session1);
- sessions.add(session2);
final InstanceIdentifier<Session> id1 =
InstanceIdentifier.create(NetconfState.class)
.child(Sessions.class)
InstanceIdentifier.create(NetconfState.class)
.child(Sessions.class)
.child(Session.class, session2.key());
- writer.start();
- writer.onSessionsUpdated(sessions);
+ writer.onSessionsUpdated(List.of(session1, session2));
InOrder inOrder = inOrder(writeTransaction);
inOrder.verify(writeTransaction).put(LogicalDatastoreType.OPERATIONAL, id1, session1);
inOrder.verify(writeTransaction).put(LogicalDatastoreType.OPERATIONAL, id2, session2);
@Test
public void testOnSessionInitiated() throws Exception {
- writer.start();
verify(monitoring).registerCapabilitiesListener(writer);
}
}
\ No newline at end of file