2 * Copyright (c) 2016 CableLabs 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.unimgr.command;
10 import static org.mockito.Matchers.any;
11 import static org.mockito.Mockito.mock;
12 import static org.mockito.Mockito.times;
13 import static org.mockito.Mockito.when;
15 import java.util.ArrayList;
16 import java.util.List;
18 import org.junit.Before;
19 import org.junit.Test;
20 import org.junit.runner.RunWith;
21 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
22 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
23 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
24 import org.opendaylight.unimgr.impl.UnimgrConstants;
25 import org.opendaylight.unimgr.impl.UnimgrMapper;
26 import org.opendaylight.unimgr.utils.EvcUtils;
27 import org.opendaylight.unimgr.utils.MdsalUtils;
28 import org.opendaylight.unimgr.utils.OvsdbUtils;
29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeRef;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.unimgr.rev151012.EvcAugmentation;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.unimgr.rev151012.UniAugmentation;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.unimgr.rev151012.evc.UniDest;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.unimgr.rev151012.evc.UniSource;
36 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
37 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
38 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
39 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
40 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Link;
41 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
42 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
43 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
44 import org.powermock.api.mockito.PowerMockito;
45 import org.powermock.core.classloader.annotations.PrepareForTest;
46 import org.powermock.modules.junit4.PowerMockRunner;
48 import com.google.common.base.Optional;
50 @RunWith(PowerMockRunner.class)
51 @PrepareForTest({EvcUtils.class, MdsalUtils.class, OvsdbUtils.class, UnimgrMapper.class})
52 public class EvcAddCommandTest {
54 private static final NodeId OVSDB_NODE_ID = new NodeId("ovsdb://7011db35-f44b-4aab-90f6-d89088caf9d8");
56 private EvcAddCommand evcAddCommand;
57 private DataTreeModification<Link> evcLink;
58 private DataBroker dataBroker;
60 private Optional<Link> optLinks;
62 @SuppressWarnings("unchecked")
65 PowerMockito.mockStatic(MdsalUtils.class);
66 PowerMockito.mockStatic(EvcUtils.class);
67 PowerMockito.mockStatic(OvsdbUtils.class);
68 PowerMockito.mockStatic(UnimgrMapper.class);
69 dataBroker = mock(DataBroker.class);
70 link = mock(Link.class);
71 optLinks = mock(Optional.class);
72 evcLink = DataTreeModificationHelper.getEvcLink(link);
73 evcAddCommand = new EvcAddCommand(dataBroker, evcLink);
77 * Test method for {@link org.opendaylight.unimgr.command.evcAddCommand#execute()}.
80 @SuppressWarnings({ "rawtypes", "unchecked" })
82 public void testExecute() throws Exception {
83 final UniAugmentation uniAugmentation = mock(UniAugmentation.class);
84 final EvcAugmentation evcAugmentation = mock(EvcAugmentation.class);
85 final List<UniSource> unisSource = new ArrayList<UniSource>();
86 final UniSource uniSource = mock(UniSource.class);
87 final List<UniDest> unisDest = new ArrayList<UniDest>();
88 final UniDest uniDest = mock(UniDest.class);
89 final IpAddress ipAddress = mock(IpAddress.class);
90 final Ipv4Address ipv4Address = mock(Ipv4Address.class);
91 final Optional<Node> optionalOvsdbNode = mock(Optional.class);
92 final Node node = mock(Node.class);
93 final OvsdbNodeRef ovsNodedRef = mock(OvsdbNodeRef.class);
94 final InstanceIdentifier evcKey = InstanceIdentifier
95 .create(NetworkTopology.class)
96 .child(Topology.class, new TopologyKey(UnimgrConstants.EVC_TOPOLOGY_ID))
97 .child(Node.class, new NodeKey(OVSDB_NODE_ID));
98 unisSource.add(uniSource);
99 unisDest.add(uniDest);
101 when(link.getAugmentation(EvcAugmentation.class)).thenReturn(evcAugmentation);
102 when(optLinks.isPresent()).thenReturn(false);
103 when(uniAugmentation.getOvsdbNodeRef()).thenReturn(ovsNodedRef);
104 when(evcAugmentation.getUniSource()).thenReturn(unisSource);
105 when(evcAugmentation.getUniDest()).thenReturn(unisDest);
106 when(uniSource.getIpAddress()).thenReturn(ipAddress);
107 when(uniDest.getIpAddress()).thenReturn(ipAddress);
108 when(uniSource.getUni()).thenReturn(evcKey);
109 when(uniDest.getUni()).thenReturn(evcKey);
110 when(ipAddress.getIpv4Address()).thenReturn(ipv4Address);
111 when(ipv4Address.toString()).thenReturn("ipv4Address_test");
112 when(ipv4Address.getValue()).thenReturn("ipv4AddressValue_test");
113 when(optionalOvsdbNode.isPresent()).thenReturn(true);
114 when(optionalOvsdbNode.get()).thenReturn(node);
115 when(node.getAugmentation(any(Class.class))).thenReturn(uniAugmentation);
116 when(ovsNodedRef.getValue()).thenReturn(evcKey);
118 when(MdsalUtils.readLink(any(DataBroker.class), any(LogicalDatastoreType.class),
119 any(InstanceIdentifier.class))).thenReturn(optLinks);
120 when(MdsalUtils.readNode(any(DataBroker.class), any(LogicalDatastoreType.class),
121 any(InstanceIdentifier.class))).thenReturn(optionalOvsdbNode);
122 PowerMockito.doNothing().when(OvsdbUtils.class, "updateMaxRate", dataBroker, uniAugmentation,
123 uniAugmentation, evcAugmentation);
124 PowerMockito.doNothing().when(OvsdbUtils.class, "createTerminationPointNode", dataBroker,
125 uniAugmentation, node, UnimgrConstants.DEFAULT_BRIDGE_NAME, UnimgrConstants.DEFAULT_TUNNEL_IFACE);
126 PowerMockito.doNothing().when(OvsdbUtils.class, "createGreTunnel", dataBroker, uniAugmentation,
127 uniAugmentation, node, UnimgrConstants.DEFAULT_BRIDGE_NAME, UnimgrConstants.DEFAULT_GRE_TUNNEL_NAME);
128 when(OvsdbUtils.createOvsdbNode(any(DataBroker.class), any(UniAugmentation.class)))
130 when(UnimgrMapper.getOvsdbNodeIid(any(NodeId.class))).thenReturn(evcKey);
131 when(UnimgrMapper.getUniIid(any(DataBroker.class), any(IpAddress.class),
132 any(LogicalDatastoreType.class))).thenReturn(evcKey);
134 evcAddCommand.execute();
135 PowerMockito.verifyStatic(times(2));
136 OvsdbUtils.createTerminationPointNode(any(DataBroker.class), any(UniAugmentation.class),
137 any(Node.class), any(String.class), any(String.class));
138 PowerMockito.verifyStatic(times(2));
139 OvsdbUtils.createGreTunnel(any(DataBroker.class), any(UniAugmentation.class),
140 any(UniAugmentation.class), any(Node.class), any(String.class), any(String.class));
141 PowerMockito.verifyStatic(times(2));
142 EvcUtils.updateEvcNode(any(LogicalDatastoreType.class), any(InstanceIdentifier.class),
143 any(EvcAugmentation.class), any(InstanceIdentifier.class),
144 any(InstanceIdentifier.class), any(DataBroker.class));