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.groupbasedpolicy.renderer.vpp.manager;
10 import java.util.ArrayList;
11 import java.util.List;
13 import org.junit.After;
14 import org.junit.Assert;
15 import org.junit.Before;
16 import org.junit.Test;
17 import org.junit.runner.RunWith;
18 import org.mockito.Matchers;
19 import org.mockito.Mock;
20 import org.mockito.Mockito;
21 import org.mockito.runners.MockitoJUnitRunner;
22 import org.opendaylight.controller.config.yang.config.vpp_provider.impl.VppRenderer;
23 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
24 import org.opendaylight.controller.md.sal.binding.api.MountPoint;
25 import org.opendaylight.controller.md.sal.binding.api.MountPointService;
26 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
27 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
28 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
29 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
30 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
31 import org.opendaylight.groupbasedpolicy.renderer.vpp.VppRendererDataBrokerTest;
32 import org.opendaylight.groupbasedpolicy.renderer.vpp.listener.VppNodeListener;
33 import org.opendaylight.groupbasedpolicy.renderer.vpp.util.VppIidFactory;
34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Host;
35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.Renderer;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.RendererKey;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNode;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeBuilder;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeConnectionStatus;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.connection.status.AvailableCapabilitiesBuilder;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.connection.status.UnavailableCapabilitiesBuilder;
45 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
46 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
47 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
48 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
49 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
50 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
51 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
53 import com.google.common.base.Optional;
54 import com.google.common.eventbus.EventBus;
55 import com.google.common.util.concurrent.CheckedFuture;
58 * Test for {@link VppNodeManager} and {@link VppNodeListener}.
60 @RunWith(MockitoJUnitRunner.class)
61 public class VppManagerDataStoreTest extends VppRendererDataBrokerTest {
63 private static final String V3PO_CAPABILITY = "(urn:opendaylight:params:xml:ns:yang:v3po?revision=2016-12-14)v3po";
64 private static final String INTERFACES_CAPABILITY =
65 "(urn:ietf:params:xml:ns:yang:ietf-interfaces?revision=2014-05-08)ietf-interfaces";
66 private static final String NODE_NAME = "testVpp";
67 private static final InstanceIdentifier<Node> NODE_IID = VppIidFactory
68 .getNodeIid(new TopologyKey(new TopologyId("topology-netconf")), new NodeKey(new NodeId(NODE_NAME)));
71 BindingAwareBroker.ProviderContext providerContext;
73 MountPointService mountPointService;
75 MountPoint mountPoint;
77 DataBroker dataBroker2;
79 private DataBroker dataBroker;
80 private VppNodeListener vppNodeListener;
81 private VppNodeManager vppNodeManager;
84 public void setUp() throws Exception {
85 Mockito.when(providerContext.getSALService(Matchers.<Class<MountPointService>>any()))
86 .thenReturn(mountPointService);
87 Mockito.when(mountPointService.getMountPoint(Matchers.<InstanceIdentifier<Node>>any()))
88 .thenReturn(Optional.of(mountPoint));
89 Mockito.when(mountPoint.getService(Matchers.<Class<DataBroker>>any())).thenReturn(Optional.of(dataBroker2));
90 dataBroker = getDataBroker();
91 vppNodeManager = new VppNodeManager(dataBroker, providerContext);
92 vppNodeListener = new VppNodeListener(dataBroker, vppNodeManager, new EventBus());
95 private Node createNode(final String name, NetconfNodeConnectionStatus.ConnectionStatus status) {
96 Host host = new Host(new IpAddress(new Ipv4Address("192.168.255.101")));
97 PortNumber portNumber = new PortNumber(2830);
99 List<String> avaibleCapabilitiesList = new ArrayList<>();
100 avaibleCapabilitiesList.add(V3PO_CAPABILITY);
101 avaibleCapabilitiesList.add(INTERFACES_CAPABILITY);
103 NetconfNode netconfNode = new NetconfNodeBuilder().setHost(host)
105 .setUnavailableCapabilities(new UnavailableCapabilitiesBuilder().build())
106 .setAvailableCapabilities(
107 new AvailableCapabilitiesBuilder().setAvailableCapability(avaibleCapabilitiesList).build())
108 .setConnectionStatus(status)
111 return new NodeBuilder().setNodeId(new NodeId(name)).addAugmentation(NetconfNode.class, netconfNode).build();
115 public void connectNode() throws ReadFailedException {
116 WriteTransaction writeTransaction = dataBroker.newWriteOnlyTransaction();
117 Node testVppNode = createNode(NODE_NAME, NetconfNodeConnectionStatus.ConnectionStatus.Connected);
119 writeTransaction.put(LogicalDatastoreType.OPERATIONAL, NODE_IID, testVppNode, true);
121 writeTransaction.submit();
123 ReadOnlyTransaction readOnlyTransaction = dataBroker.newReadOnlyTransaction();
124 CheckedFuture<Optional<Renderer>, ReadFailedException> future =
125 readOnlyTransaction.read(LogicalDatastoreType.OPERATIONAL,
126 VppIidFactory.getRendererIID(new RendererKey(VppRenderer.NAME)));
127 Optional<Renderer> rendererOptional = future.checkedGet();
129 Assert.assertTrue(rendererOptional.isPresent());
130 Assert.assertEquals(1, rendererOptional.get().getRendererNodes().getRendererNode().size());
131 Assert.assertEquals(NODE_IID, rendererOptional.get().getRendererNodes().getRendererNode().get(0).getNodePath());
135 public void disconnectNode() throws ReadFailedException, InterruptedException {
136 WriteTransaction writeTransaction = dataBroker.newWriteOnlyTransaction();
137 Node testVppNode = createNode(NODE_NAME, NetconfNodeConnectionStatus.ConnectionStatus.Connected);
139 writeTransaction.put(LogicalDatastoreType.OPERATIONAL, NODE_IID, testVppNode, true);
141 writeTransaction.submit();
143 ReadOnlyTransaction readOnlyTransaction = dataBroker.newReadOnlyTransaction();
144 CheckedFuture<Optional<Renderer>, ReadFailedException> future =
145 readOnlyTransaction.read(LogicalDatastoreType.OPERATIONAL,
146 VppIidFactory.getRendererIID(new RendererKey(VppRenderer.NAME)));
147 Optional<Renderer> rendererOptional = future.checkedGet();
149 Assert.assertTrue(rendererOptional.isPresent());
150 Assert.assertEquals(1, rendererOptional.get().getRendererNodes().getRendererNode().size());
151 Assert.assertEquals(NODE_IID, rendererOptional.get().getRendererNodes().getRendererNode().get(0).getNodePath());
153 WriteTransaction writeTransaction2 = dataBroker.newWriteOnlyTransaction();
154 Node testVppNode2 = createNode(NODE_NAME, NetconfNodeConnectionStatus.ConnectionStatus.Connecting);
156 writeTransaction2.put(LogicalDatastoreType.OPERATIONAL, NODE_IID, testVppNode2, true);
158 writeTransaction2.submit();
160 ReadOnlyTransaction readOnlyTransaction2 = dataBroker.newReadOnlyTransaction();
161 CheckedFuture<Optional<Renderer>, ReadFailedException> future2 =
162 readOnlyTransaction2.read(LogicalDatastoreType.OPERATIONAL,
163 VppIidFactory.getRendererIID(new RendererKey(VppRenderer.NAME)));
164 Optional<Renderer> rendererOptional2 = future2.checkedGet();
166 Assert.assertTrue(rendererOptional2.isPresent());
167 Assert.assertEquals(0, rendererOptional2.get().getRendererNodes().getRendererNode().size());
171 public void cleanUp() throws Exception {
172 vppNodeListener.close();