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.MdsalUtils;
27 import org.opendaylight.unimgr.utils.OvsdbUtils;
28 import org.opendaylight.unimgr.utils.UniUtils;
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.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeAugmentation;
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.ovsdb.rev150105.ovsdb.node.attributes.QosEntries;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.QosEntriesKey;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.Queues;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.QueuesKey;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.unimgr.rev151012.UniAugmentation;
37 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
38 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
39 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
40 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
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({UniUtils.class, OvsdbUtils.class, MdsalUtils.class,
53 public class UniRemoveCommandTest {
55 private static final NodeId OVSDB_NODE_ID = new NodeId("ovsdb://7011db35-f44b-4aab-90f6-d89088caf9d8");
56 private UniRemoveCommand uniRemoveCommand;
57 private DataTreeModification<Node> uni;
58 private DataBroker dataBroker;
61 @SuppressWarnings("unchecked")
64 PowerMockito.mockStatic(UniUtils.class);
65 PowerMockito.mockStatic(OvsdbUtils.class);
66 PowerMockito.mockStatic(MdsalUtils.class);
67 PowerMockito.mockStatic(UnimgrMapper.class);
68 dataBroker = mock(DataBroker.class);
69 uniNode = mock(Node.class);
70 uni = DataTreeModificationHelper.getUniNode(uniNode);
71 uniRemoveCommand = new UniRemoveCommand(dataBroker, uni);
75 * Test method for {@link org.opendaylight.unimgr.command.UniRemoveCommand#execute()}.
77 @SuppressWarnings({ "unchecked", "rawtypes"})
79 public void testExecute() {
80 final UniAugmentation uniAugmentation = mock(UniAugmentation.class);
81 final OvsdbNodeRef ovsNodedRef = mock(OvsdbNodeRef.class);
82 final Optional<Node> optionalNode = mock(Optional.class);
83 final InstanceIdentifier instanceOfNode = mock(InstanceIdentifier.class);
84 final InstanceIdentifier uniKey = InstanceIdentifier
85 .create(NetworkTopology.class)
86 .child(Topology.class, new TopologyKey(UnimgrConstants.UNI_TOPOLOGY_ID))
87 .child(Node.class, new NodeKey(OVSDB_NODE_ID));
88 final Node ovsdbNode = mock(Node.class);
89 final OvsdbNodeAugmentation ovsdbNodeAugmentation = mock(OvsdbNodeAugmentation.class);
90 final QosEntries qosEntries = mock(QosEntries.class);
91 final Queues queues = mock(Queues.class);
92 when(uniNode.getAugmentation(UniAugmentation.class)).thenReturn(uniAugmentation);
93 when(uniAugmentation.getIpAddress()).thenReturn(mock(IpAddress.class));
94 when(uniAugmentation.getOvsdbNodeRef()).thenReturn(ovsNodedRef);
95 when(ovsNodedRef.getValue()).thenReturn(uniKey);
96 when(optionalNode.isPresent()).thenReturn(true);
97 when(optionalNode.get()).thenReturn(ovsdbNode);
98 when(ovsdbNode.getAugmentation(OvsdbNodeAugmentation.class)).thenReturn(ovsdbNodeAugmentation);
99 List<QosEntries> qosEntriesList = new ArrayList<>();
100 when(qosEntries.getKey()).thenReturn(mock(QosEntriesKey.class));
101 qosEntriesList.add(qosEntries);
102 when(ovsdbNodeAugmentation.getQosEntries()).thenReturn(qosEntriesList);
103 List<Queues> queuesList = new ArrayList<>();
104 when(queues.getKey()).thenReturn(mock(QueuesKey.class));
105 queuesList.add(queues);
106 when(ovsdbNodeAugmentation.getQueues()).thenReturn(queuesList);
107 when(MdsalUtils.read(any(DataBroker.class), any(LogicalDatastoreType.class),
108 any(InstanceIdentifier.class))).thenReturn(uniAugmentation);
109 when(MdsalUtils.readNode(any(DataBroker.class), any(LogicalDatastoreType.class),
110 any(InstanceIdentifier.class))).thenReturn(optionalNode);
111 when(MdsalUtils.readNode(any(DataBroker.class), any(InstanceIdentifier.class)))
112 .thenReturn(optionalNode);
113 when(UnimgrMapper.getOvsdbBridgeNodeIid(any(Node.class))).thenReturn(instanceOfNode);
114 when(UnimgrMapper.getTerminationPointIid(any(Node.class), any(String.class)))
115 .thenReturn(instanceOfNode);
116 when(UnimgrMapper.getUniIid(any(DataBroker.class), any(IpAddress.class),
117 any(LogicalDatastoreType.class))).thenReturn(instanceOfNode);
118 when(UnimgrMapper.getOvsdbQoSEntriesIid(any(Node.class), any(QosEntriesKey.class))).thenReturn(instanceOfNode);
119 when(UnimgrMapper.getOvsdbQueuesIid(any(Node.class), any(QueuesKey.class))).thenReturn(instanceOfNode);
120 when(MdsalUtils.deleteNode(any(DataBroker.class), any(InstanceIdentifier.class),
121 any(LogicalDatastoreType.class))).thenReturn(true);
122 uniRemoveCommand.execute();
123 PowerMockito.verifyStatic(times(4));
124 MdsalUtils.deleteNode(any(DataBroker.class), any(InstanceIdentifier.class),
125 any(LogicalDatastoreType.class));