Upgrade powermock to 2.0.0
[ovsdb.git] / southbound / southbound-impl / src / test / java / org / opendaylight / ovsdb / southbound / transactions / md / OvsdbPortUpdateCommandTest.java
1 /*
2  * Copyright © 2015, 2017 Inocybe Technologies 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.ovsdb.southbound.transactions.md;
10
11 import static org.junit.Assert.assertEquals;
12 import static org.mockito.ArgumentMatchers.any;
13 import static org.mockito.ArgumentMatchers.anyString;
14 import static org.mockito.ArgumentMatchers.eq;
15 import static org.mockito.Mockito.doNothing;
16 import static org.mockito.Mockito.mock;
17 import static org.mockito.Mockito.times;
18 import static org.mockito.Mockito.verify;
19 import static org.mockito.Mockito.when;
20 import static org.powermock.api.support.membermodification.MemberMatcher.field;
21 import static org.powermock.api.support.membermodification.MemberMatcher.method;
22 import static org.powermock.api.support.membermodification.MemberModifier.suppress;
23
24 import com.google.common.base.Optional;
25 import com.google.common.util.concurrent.CheckedFuture;
26 import java.util.AbstractMap.SimpleEntry;
27 import java.util.ArrayList;
28 import java.util.HashMap;
29 import java.util.HashSet;
30 import java.util.List;
31 import java.util.Map;
32 import java.util.Map.Entry;
33 import java.util.Set;
34 import org.junit.Before;
35 import org.junit.Ignore;
36 import org.junit.Test;
37 import org.junit.runner.RunWith;
38 import org.mockito.Mockito;
39 import org.mockito.stubbing.Answer;
40 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
41 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
42 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
43 import org.opendaylight.ovsdb.lib.message.TableUpdates;
44 import org.opendaylight.ovsdb.lib.notation.Column;
45 import org.opendaylight.ovsdb.lib.notation.UUID;
46 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
47 import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
48 import org.opendaylight.ovsdb.lib.schema.typed.TyperUtils;
49 import org.opendaylight.ovsdb.schema.openvswitch.Bridge;
50 import org.opendaylight.ovsdb.schema.openvswitch.Interface;
51 import org.opendaylight.ovsdb.schema.openvswitch.Port;
52 import org.opendaylight.ovsdb.southbound.InstanceIdentifierCodec;
53 import org.opendaylight.ovsdb.southbound.OvsdbConnectionInstance;
54 import org.opendaylight.ovsdb.southbound.SouthboundConstants;
55 import org.opendaylight.ovsdb.southbound.SouthboundMapper;
56 import org.opendaylight.ovsdb.southbound.SouthboundUtil;
57 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
58 import org.opendaylight.yang.gen.v1.urn.opendaylight.l2.types.rev130827.VlanId;
59 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.InterfaceTypeBase;
60 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.InterfaceTypeInternal;
61 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeAugmentation;
62 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeRef;
63 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeAugmentation;
64 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbPortInterfaceAttributes;
65 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbPortInterfaceAttributes.VlanMode;
66 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbTerminationPointAugmentation;
67 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbTerminationPointAugmentationBuilder;
68 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagedNodeEntry;
69 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.port._interface.attributes.InterfaceExternalIds;
70 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.port._interface.attributes.InterfaceExternalIdsBuilder;
71 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.port._interface.attributes.InterfaceOtherConfigs;
72 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.port._interface.attributes.InterfaceOtherConfigsBuilder;
73 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.port._interface.attributes.PortExternalIds;
74 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.port._interface.attributes.PortExternalIdsBuilder;
75 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.port._interface.attributes.PortOtherConfigs;
76 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.port._interface.attributes.PortOtherConfigsBuilder;
77 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.port._interface.attributes.TrunksBuilder;
78 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
79 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TpId;
80 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
81 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
82 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
83 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPoint;
84 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPointBuilder;
85 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPointKey;
86 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
87 import org.powermock.api.mockito.PowerMockito;
88 import org.powermock.api.support.membermodification.MemberMatcher;
89 import org.powermock.api.support.membermodification.MemberModifier;
90 import org.powermock.core.classloader.annotations.PrepareForTest;
91 import org.powermock.modules.junit4.PowerMockRunner;
92 import org.powermock.reflect.Whitebox;
93
94 @RunWith(PowerMockRunner.class)
95 @PrepareForTest({TyperUtils.class, OvsdbPortUpdateCommand.class, SouthboundUtil.class, SouthboundMapper.class})
96 public class OvsdbPortUpdateCommandTest {
97
98     private static final String OTHER_CONFIG_KEY = "key";
99     private static final String OTHER_CONFIG_VALUE = "value";
100     private static final String EXTERNAL_ID_KEY = "key";
101     private static final String EXTERNAL_ID_VALUE = "value";
102     private static final String INTERFACE_NAME = "interface_name";
103     private static final String VLAN_MODE_ACCESS = "access";
104     private static final String OVSDB_INTERFACE_TYPE = "internal";
105     private static final String PORT_NAME = "port_name";
106     private static final String TP_NAME = "tp_name";
107     private static final String TERMINATION_POINT_NAME = "termination_point_name";
108
109     private Map<UUID, Port> portUpdatedRows;
110     private Map<UUID, Port> portOldRows;
111     private Map<UUID, Interface> ifUpdatedRows;
112     private Map<UUID, Interface> interfaceOldRows;
113     private Map<UUID, Bridge> bridgeUpdatedRows;
114     private OvsdbPortUpdateCommand ovsdbPortUpdateCommand;
115
116     @Before
117     public void setUp() throws Exception {
118         ovsdbPortUpdateCommand = PowerMockito.mock(OvsdbPortUpdateCommand.class, Mockito.CALLS_REAL_METHODS);
119     }
120
121     @Test
122     public void testOvsdbPortUpdateCommand() throws Exception {
123         TableUpdates updates = mock(TableUpdates.class);
124         DatabaseSchema dbSchema = mock(DatabaseSchema.class);
125
126         PowerMockito.mockStatic(TyperUtils.class);
127         PowerMockito.when(TyperUtils.extractRowsUpdated(Port.class, updates, dbSchema)).thenReturn(portUpdatedRows);
128         PowerMockito.when(TyperUtils.extractRowsOld(Port.class, updates, dbSchema)).thenReturn(portOldRows);
129         PowerMockito.when(TyperUtils.extractRowsUpdated(Interface.class, updates, dbSchema)).thenReturn(ifUpdatedRows);
130         PowerMockito.when(TyperUtils.extractRowsOld(Interface.class, updates, dbSchema)).thenReturn(interfaceOldRows);
131         PowerMockito.when(TyperUtils.extractRowsUpdated(Bridge.class, updates, dbSchema)).thenReturn(bridgeUpdatedRows);
132
133         OvsdbConnectionInstance key = mock(OvsdbConnectionInstance.class);
134         OvsdbPortUpdateCommand ovsdbPortUpdateCommand1 =
135                 new OvsdbPortUpdateCommand(mock(InstanceIdentifierCodec.class), key, updates, dbSchema);
136         assertEquals(portUpdatedRows, Whitebox.getInternalState(ovsdbPortUpdateCommand1, "portUpdatedRows"));
137         assertEquals(portOldRows, Whitebox.getInternalState(ovsdbPortUpdateCommand1, "portOldRows"));
138         assertEquals(dbSchema, Whitebox.getInternalState(ovsdbPortUpdateCommand1, "dbSchema"));
139     }
140
141     @SuppressWarnings("unchecked")
142     @Test
143     public void testExecute() throws Exception {
144         portUpdatedRows = new HashMap<>();
145         interfaceOldRows = new HashMap<>();
146         portUpdatedRows.put(mock(UUID.class), mock(Port.class));
147         interfaceOldRows.put(mock(UUID.class), mock(Interface.class));
148         field(OvsdbPortUpdateCommand.class, "portUpdatedRows").set(ovsdbPortUpdateCommand, portUpdatedRows);
149         field(OvsdbPortUpdateCommand.class, "interfaceOldRows").set(ovsdbPortUpdateCommand, interfaceOldRows);
150
151         OvsdbConnectionInstance ovsdbConnectionInstance = mock(OvsdbConnectionInstance.class);
152         when(ovsdbPortUpdateCommand.getOvsdbConnectionInstance()).thenReturn(ovsdbConnectionInstance);
153         InstanceIdentifier<Node> connectionIId = mock(InstanceIdentifier.class);
154         when(ovsdbConnectionInstance.getInstanceIdentifier()).thenReturn(connectionIId);
155
156         //case 1: portUpdatedRows & interfaceOldRows not null, not empty
157         Optional<Node> node = Optional.of(mock(Node.class));
158         PowerMockito.doReturn(node).when(ovsdbPortUpdateCommand, "readNode", any(ReadWriteTransaction.class),
159                 any(InstanceIdentifier.class));
160         doNothing().when(ovsdbPortUpdateCommand).updateTerminationPoints(any(ReadWriteTransaction.class),
161             any(Node.class));
162         ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
163         ovsdbPortUpdateCommand.execute(transaction);
164         verify(ovsdbConnectionInstance).getInstanceIdentifier();
165         verify(ovsdbPortUpdateCommand).updateTerminationPoints(any(ReadWriteTransaction.class), any(Node.class));
166     }
167
168     @Test
169     @SuppressWarnings("unchecked")
170     public void testUpdateTerminationPoints() throws Exception {
171         portUpdatedRows = new HashMap<>();
172         Port port = mock(Port.class);
173         UUID uuid = mock(UUID.class);
174         portUpdatedRows.put(uuid, port);
175         field(OvsdbPortUpdateCommand.class, "portUpdatedRows").set(ovsdbPortUpdateCommand, portUpdatedRows);
176         Column<GenericTableSchema, String> bridgeColumn = mock(Column.class);
177         when(port.getNameColumn()).thenReturn(bridgeColumn);
178         when(bridgeColumn.getData()).thenReturn(TERMINATION_POINT_NAME);
179
180         Optional<InstanceIdentifier<Node>> bridgeIid = Optional.of(mock(InstanceIdentifier.class));
181         PowerMockito.doReturn(bridgeIid).when(ovsdbPortUpdateCommand, "getTerminationPointBridge", any(UUID.class));
182
183         NodeId bridgeId = mock(NodeId.class);
184         PowerMockito.mockStatic(SouthboundMapper.class);
185         PowerMockito.when(SouthboundMapper.createManagedNodeId(any(InstanceIdentifier.class))).thenReturn(bridgeId);
186
187         PowerMockito.whenNew(TpId.class).withAnyArguments().thenReturn(mock(TpId.class));
188         TerminationPointKey tpKey = mock(TerminationPointKey.class);
189         PowerMockito.whenNew(TerminationPointKey.class).withAnyArguments().thenReturn(tpKey);
190         TerminationPointBuilder tpBuilder = mock(TerminationPointBuilder.class);
191         PowerMockito.whenNew(TerminationPointBuilder.class).withNoArguments().thenReturn(tpBuilder);
192         when(tpBuilder.withKey(any(TerminationPointKey.class))).thenReturn(tpBuilder);
193         when(tpKey.getTpId()).thenReturn(mock(TpId.class));
194         when(tpBuilder.setTpId(any(TpId.class))).thenReturn(tpBuilder);
195         InstanceIdentifier<TerminationPoint> tpPath = mock(InstanceIdentifier.class);
196         PowerMockito.doReturn(tpPath).when(ovsdbPortUpdateCommand, "getInstanceIdentifier",
197                 any(InstanceIdentifier.class), any(Port.class));
198
199         OvsdbTerminationPointAugmentationBuilder tpAugmentationBuilder = mock(
200                 OvsdbTerminationPointAugmentationBuilder.class);
201         PowerMockito.whenNew(OvsdbTerminationPointAugmentationBuilder.class).withNoArguments()
202                 .thenReturn(tpAugmentationBuilder);
203         PowerMockito.suppress(MemberMatcher.method(OvsdbPortUpdateCommand.class, "buildTerminationPoint",
204                 ReadWriteTransaction.class, InstanceIdentifier.class, OvsdbTerminationPointAugmentationBuilder.class,
205                 Node.class, Entry.class));
206
207         Column<GenericTableSchema, Set<UUID>> interfacesColumn = mock(Column.class);
208         when(port.getInterfacesColumn()).thenReturn(interfacesColumn);
209         Set<UUID> uuids = new HashSet<>();
210         UUID uuid2 = mock(UUID.class);
211         uuids.add(uuid2);
212         when(interfacesColumn.getData()).thenReturn(uuids);
213
214         ifUpdatedRows = new HashMap<>();
215         interfaceOldRows = new HashMap<>();
216         Interface iface = mock(Interface.class);
217         ifUpdatedRows.put(uuid2, iface);
218         Interface interfaceUpdate = mock(Interface.class);
219         ifUpdatedRows.put(uuid, interfaceUpdate);
220         interfaceOldRows.put(uuid2, iface);
221         field(OvsdbPortUpdateCommand.class, "interfaceUpdatedRows").set(ovsdbPortUpdateCommand, ifUpdatedRows);
222         field(OvsdbPortUpdateCommand.class, "interfaceOldRows").set(ovsdbPortUpdateCommand, interfaceOldRows);
223         PowerMockito.suppress(MemberMatcher.method(OvsdbPortUpdateCommand.class, "buildTerminationPoint",
224                 OvsdbTerminationPointAugmentationBuilder.class, Interface.class));
225
226         when(tpAugmentationBuilder.build()).thenReturn(mock(OvsdbTerminationPointAugmentation.class));
227         when(tpBuilder.addAugmentation(eq(OvsdbTerminationPointAugmentation.class),
228                 any(OvsdbTerminationPointAugmentation.class))).thenReturn(tpBuilder);
229         when(tpBuilder.build()).thenReturn(mock(TerminationPoint.class));
230         portOldRows = new HashMap<>();
231         portOldRows.put(uuid, port);
232         MemberModifier.field(OvsdbPortUpdateCommand.class, "portOldRows").set(ovsdbPortUpdateCommand, portOldRows);
233         ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
234         doNothing().when(transaction).merge(any(LogicalDatastoreType.class), any(InstanceIdentifier.class),
235                 any(TerminationPoint.class));
236         doNothing().when(transaction).put(any(LogicalDatastoreType.class), any(InstanceIdentifier.class),
237                 any(TerminationPoint.class));
238
239         Column<GenericTableSchema, String> interfaceColumn = mock(Column.class);
240         when(interfaceUpdate.getNameColumn()).thenReturn(interfaceColumn);
241         when(interfaceColumn.getData()).thenReturn(INTERFACE_NAME);
242
243         PowerMockito.doReturn(bridgeIid).when(ovsdbPortUpdateCommand, "getTerminationPointBridge",
244                 any(ReadWriteTransaction.class), any(Node.class), anyString());
245         PowerMockito.when(SouthboundMapper.createManagedNodeId(any(InstanceIdentifier.class))).thenReturn(bridgeId);
246         PowerMockito.whenNew(TopologyKey.class).withAnyArguments().thenReturn(mock(TopologyKey.class));
247         PowerMockito.whenNew(NodeKey.class).withAnyArguments().thenReturn(mock(NodeKey.class));
248
249         Node node = mock(Node.class);
250         Whitebox.invokeMethod(ovsdbPortUpdateCommand, "updateTerminationPoints", transaction, node);
251         verify(ovsdbPortUpdateCommand).getInstanceIdentifier(any(InstanceIdentifier.class),
252             any(Port.class));
253         verify(transaction, times(2)).merge(any(LogicalDatastoreType.class), any(InstanceIdentifier.class),
254                 any(TerminationPoint.class));
255     }
256
257     @Test
258     @SuppressWarnings("unchecked")
259     public void testBuildTerminationPoint() throws Exception {
260         Port portUpdate = mock(Port.class);
261         Entry<UUID,Port> portEntry = mock(Entry.class);
262         when(portEntry.getValue()).thenReturn(mock(Port.class));
263         when(portEntry.getValue().getName()).thenReturn(PORT_NAME);
264         when(portEntry.getValue().getUuid()).thenReturn(mock(UUID.class));
265         when(portUpdate.getName()).thenReturn(PORT_NAME);
266         when(portUpdate.getUuid()).thenReturn(mock(UUID.class));
267         PowerMockito.whenNew(Uuid.class).withAnyArguments().thenReturn(mock(Uuid.class));
268
269         OvsdbTerminationPointAugmentationBuilder tpAugmentationBuilder = mock(
270                 OvsdbTerminationPointAugmentationBuilder.class);
271
272         when(tpAugmentationBuilder.setName(anyString())).thenReturn(tpAugmentationBuilder);
273         when(tpAugmentationBuilder.setPortUuid(any(Uuid.class))).thenReturn(tpAugmentationBuilder);
274         doNothing().when(ovsdbPortUpdateCommand).updatePort(any(ReadWriteTransaction.class),
275             any(Node.class), any(InstanceIdentifier.class), any(Entry.class),
276             any(OvsdbTerminationPointAugmentationBuilder.class));
277
278         Node node = mock(Node.class);
279         ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
280         InstanceIdentifier<TerminationPoint> tpPath = mock(InstanceIdentifier.class);
281
282         Whitebox.invokeMethod(ovsdbPortUpdateCommand, "buildTerminationPoint", transaction, tpPath,
283                 tpAugmentationBuilder, node, portEntry);
284         verify(tpAugmentationBuilder).setName(anyString());
285         verify(tpAugmentationBuilder).setPortUuid(any(Uuid.class));
286         verify(ovsdbPortUpdateCommand).updatePort(any(ReadWriteTransaction.class),
287                 any(Node.class), any(InstanceIdentifier.class), any(Entry.class),
288                 any(OvsdbTerminationPointAugmentationBuilder.class));
289     }
290
291     @Test
292     public void testBuildTerminationPoint1() throws Exception {
293         Interface interfaceUpdate = mock(Interface.class);
294         when(interfaceUpdate.getName()).thenReturn(INTERFACE_NAME);
295         when(interfaceUpdate.getUuid()).thenReturn(mock(UUID.class));
296         PowerMockito.whenNew(Uuid.class).withAnyArguments().thenReturn(mock(Uuid.class));
297         OvsdbTerminationPointAugmentationBuilder tpAugmentationBuilder = mock(
298                 OvsdbTerminationPointAugmentationBuilder.class);
299         when(tpAugmentationBuilder.setName(anyString())).thenReturn(tpAugmentationBuilder);
300         when(tpAugmentationBuilder.setInterfaceUuid(any(Uuid.class))).thenReturn(tpAugmentationBuilder);
301
302         doNothing().when(ovsdbPortUpdateCommand).updateInterfaces(any(Interface.class),
303             any(OvsdbTerminationPointAugmentationBuilder.class));
304
305         Whitebox.invokeMethod(ovsdbPortUpdateCommand, "buildTerminationPoint", tpAugmentationBuilder, interfaceUpdate);
306         verify(tpAugmentationBuilder).setName(anyString());
307         verify(tpAugmentationBuilder).setInterfaceUuid(any(Uuid.class));
308         verify(ovsdbPortUpdateCommand).updateInterfaces(any(Interface.class),
309                 any(OvsdbTerminationPointAugmentationBuilder.class));
310     }
311
312     @SuppressWarnings("unchecked")
313     @Test
314     public void testReadNode() throws Exception {
315         ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
316         InstanceIdentifier<Node> nodePath = mock(InstanceIdentifier.class);
317         Optional<Node> node = Optional.of(mock(Node.class));
318         CheckedFuture<Optional<Node>, ReadFailedException> checkedFuture = mock(CheckedFuture.class);
319         when(transaction.read(any(LogicalDatastoreType.class), any(InstanceIdentifier.class)))
320                 .thenReturn(checkedFuture);
321         when(checkedFuture.checkedGet()).thenReturn(node);
322         assertEquals(node, Whitebox.invokeMethod(ovsdbPortUpdateCommand, "readNode", transaction, nodePath));
323     }
324
325     @SuppressWarnings("unchecked")
326     @Test
327     public void testGetTerminationPointBridge1() throws Exception {
328         Node node = mock(Node.class);
329         OvsdbNodeAugmentation ovsdbNode = mock(OvsdbNodeAugmentation.class);
330         when(node.augmentation(OvsdbNodeAugmentation.class)).thenReturn(ovsdbNode);
331         List<ManagedNodeEntry> managedNodes = new ArrayList<>();
332         ManagedNodeEntry managedNodeEntry = mock(ManagedNodeEntry.class);
333         managedNodes.add(managedNodeEntry);
334         when(ovsdbNode.getManagedNodeEntry()).thenReturn(managedNodes);
335
336         Node managedNode = mock(Node.class);
337         OvsdbBridgeRef ovsdbBridgeRef = mock(OvsdbBridgeRef.class);
338         when(managedNodeEntry.getBridgeRef()).thenReturn(ovsdbBridgeRef);
339         InstanceIdentifier<Node> iidNode = mock(InstanceIdentifier.class);
340         when((InstanceIdentifier<Node>) ovsdbBridgeRef.getValue()).thenReturn(iidNode);
341         Optional<Node> optionalNode = Optional.of(managedNode);
342         PowerMockito.doReturn(optionalNode).when(ovsdbPortUpdateCommand, "readNode", any(ReadWriteTransaction.class),
343                 any(InstanceIdentifier.class));
344
345         TerminationPointBuilder tpBuilder = mock(TerminationPointBuilder.class);
346         PowerMockito.whenNew(TerminationPointBuilder.class).withNoArguments().thenReturn(tpBuilder);
347         PowerMockito.whenNew(TpId.class).withAnyArguments().thenReturn(mock(TpId.class));
348         PowerMockito.whenNew(TerminationPointKey.class).withAnyArguments().thenReturn(mock(TerminationPointKey.class));
349         when(tpBuilder.withKey(any(TerminationPointKey.class))).thenReturn(tpBuilder);
350
351         List<TerminationPoint> terminationPointList = new ArrayList<>();
352         TerminationPoint terminationPoint = mock(TerminationPoint.class);
353         terminationPointList.add(terminationPoint);
354         when(tpBuilder.build()).thenReturn(terminationPoint);
355         when(managedNode.getTerminationPoint()).thenReturn(terminationPointList);
356
357         when(managedNode.augmentation(OvsdbBridgeAugmentation.class))
358                 .thenReturn(mock(OvsdbBridgeAugmentation.class));
359         TpId tpId = new TpId(TP_NAME);
360         when(terminationPoint.getTpId()).thenReturn(tpId);
361
362         Optional<InstanceIdentifier<Node>> testResult = Optional.of(iidNode);
363         ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
364         Optional<InstanceIdentifier<Node>> result = Whitebox.invokeMethod(ovsdbPortUpdateCommand,
365                 "getTerminationPointBridge", transaction, node, TP_NAME);
366
367         assertEquals(testResult, result);
368     }
369
370     @SuppressWarnings("unchecked")
371     @Test
372     public void testUpdateInterfaces() throws Exception {
373         Interface interfaceUpdate = mock(Interface.class);
374         Column<GenericTableSchema, String> typeColumn = mock(Column.class);
375         when(interfaceUpdate.getTypeColumn()).thenReturn(typeColumn);
376         when(typeColumn.getData()).thenReturn(OVSDB_INTERFACE_TYPE);
377         doNothing().when(ovsdbPortUpdateCommand).updateInterface(any(Interface.class), anyString(),
378             any(OvsdbTerminationPointAugmentationBuilder.class));
379
380         OvsdbTerminationPointAugmentationBuilder ovsdbTerminationPointBuilder = mock(
381                 OvsdbTerminationPointAugmentationBuilder.class);
382         Whitebox.invokeMethod(ovsdbPortUpdateCommand, "updateInterfaces", interfaceUpdate,
383                 ovsdbTerminationPointBuilder);
384         verify(ovsdbPortUpdateCommand).updateInterface(any(Interface.class), anyString(),
385                 any(OvsdbTerminationPointAugmentationBuilder.class));
386     }
387
388     @Test
389     @SuppressWarnings("unchecked")
390     public void testUpdateInterface() throws Exception {
391         Interface interf = mock(Interface.class);
392         OvsdbTerminationPointAugmentationBuilder ovsdbTerminationPointBuilder = mock(
393                 OvsdbTerminationPointAugmentationBuilder.class);
394         when(interf.getUuid()).thenReturn(mock(UUID.class));
395         PowerMockito.whenNew(Uuid.class).withAnyArguments().thenReturn(mock(Uuid.class));
396         when(ovsdbTerminationPointBuilder.setInterfaceUuid(any(Uuid.class))).thenReturn(ovsdbTerminationPointBuilder);
397         PowerMockito.mockStatic(SouthboundMapper.class);
398         PowerMockito.when(SouthboundMapper.createInterfaceType(anyString()))
399                 .thenAnswer((Answer<Class<? extends InterfaceTypeBase>>) invocation -> InterfaceTypeInternal.class);
400         when(ovsdbTerminationPointBuilder.setInterfaceType(any(Class.class))).thenReturn(ovsdbTerminationPointBuilder);
401         suppress(method(OvsdbPortUpdateCommand.class, "updateOfPort", Interface.class,
402                 OvsdbTerminationPointAugmentationBuilder.class));
403         suppress(method(OvsdbPortUpdateCommand.class, "updateOfPortRequest", Interface.class,
404                 OvsdbTerminationPointAugmentationBuilder.class));
405         suppress(method(OvsdbPortUpdateCommand.class, "updateInterfaceExternalIds", Interface.class,
406                 OvsdbTerminationPointAugmentationBuilder.class));
407         suppress(method(OvsdbPortUpdateCommand.class, "updateOptions", Interface.class,
408                 OvsdbTerminationPointAugmentationBuilder.class));
409         suppress(method(OvsdbPortUpdateCommand.class, "updateInterfaceOtherConfig", Interface.class,
410                 OvsdbTerminationPointAugmentationBuilder.class));
411         suppress(method(OvsdbPortUpdateCommand.class, "updateInterfaceLldp", Interface.class,
412                 OvsdbTerminationPointAugmentationBuilder.class));
413         suppress(method(OvsdbPortUpdateCommand.class, "updateInterfaceBfd", Interface.class,
414                 OvsdbTerminationPointAugmentationBuilder.class));
415         suppress(method(OvsdbPortUpdateCommand.class, "updateInterfaceBfdStatus", Interface.class,
416                 OvsdbTerminationPointAugmentationBuilder.class));
417         suppress(method(OvsdbPortUpdateCommand.class, "updateIfIndex", Interface.class,
418                 OvsdbTerminationPointAugmentationBuilder.class));
419
420         Whitebox.invokeMethod(ovsdbPortUpdateCommand, "updateInterface", interf, OVSDB_INTERFACE_TYPE,
421                 ovsdbTerminationPointBuilder);
422         verify(ovsdbTerminationPointBuilder).setInterfaceUuid(any(Uuid.class));
423         verify(ovsdbTerminationPointBuilder).setInterfaceType(any(Class.class));
424         verify(ovsdbPortUpdateCommand).updateOfPort(any(Interface.class),
425                 any(OvsdbTerminationPointAugmentationBuilder.class));
426         verify(ovsdbPortUpdateCommand).updateOfPortRequest(any(Interface.class),
427                 any(OvsdbTerminationPointAugmentationBuilder.class));
428         verify(ovsdbPortUpdateCommand).updateInterfaceExternalIds(any(Interface.class),
429                 any(OvsdbTerminationPointAugmentationBuilder.class));
430         verify(ovsdbPortUpdateCommand).updateOptions(any(Interface.class),
431                 any(OvsdbTerminationPointAugmentationBuilder.class));
432         verify(ovsdbPortUpdateCommand).updateInterfaceOtherConfig(any(Interface.class),
433                 any(OvsdbTerminationPointAugmentationBuilder.class));
434     }
435
436     @Test
437     @SuppressWarnings("unchecked")
438     public void testUpdatePort() throws Exception {
439         suppress(method(OvsdbPortUpdateCommand.class, "updateVlan", Port.class,
440             OvsdbTerminationPointAugmentationBuilder.class));
441         suppress(method(OvsdbPortUpdateCommand.class, "updateVlanTrunks", Port.class,
442             OvsdbTerminationPointAugmentationBuilder.class));
443         suppress(method(OvsdbPortUpdateCommand.class, "updateVlanMode", Port.class,
444             OvsdbTerminationPointAugmentationBuilder.class));
445         suppress(method(OvsdbPortUpdateCommand.class, "updatePortExternalIds", Port.class,
446             OvsdbTerminationPointAugmentationBuilder.class));
447         suppress(method(OvsdbPortUpdateCommand.class, "updatePortOtherConfig", Port.class,
448             OvsdbTerminationPointAugmentationBuilder.class));
449         suppress(method(OvsdbPortUpdateCommand.class, "updatePortOtherConfig", Port.class,
450             OvsdbTerminationPointAugmentationBuilder.class));
451         suppress(method(OvsdbPortUpdateCommand.class, "updateQos", ReadWriteTransaction.class, Node.class,
452                 InstanceIdentifier.class, Entry.class, OvsdbTerminationPointAugmentationBuilder.class));
453
454         Node node = mock(Node.class);
455         Entry<UUID, Port> port = new SimpleEntry<>(mock(UUID.class), mock(Port.class));
456         OvsdbTerminationPointAugmentationBuilder ovsdbTerminationPointBuilder = mock(
457                 OvsdbTerminationPointAugmentationBuilder.class);
458         ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
459         InstanceIdentifier<TerminationPoint> tpPath = mock(InstanceIdentifier.class);
460         Whitebox.invokeMethod(ovsdbPortUpdateCommand, "updatePort", transaction, node, tpPath, port,
461                 ovsdbTerminationPointBuilder);
462
463         verify(ovsdbPortUpdateCommand).updateVlan(any(Port.class),
464                 any(OvsdbTerminationPointAugmentationBuilder.class));
465         verify(ovsdbPortUpdateCommand).updateVlanTrunks(any(Port.class),
466                 any(OvsdbTerminationPointAugmentationBuilder.class));
467         verify(ovsdbPortUpdateCommand).updateVlanMode(any(Port.class),
468                 any(OvsdbTerminationPointAugmentationBuilder.class));
469         verify(ovsdbPortUpdateCommand).updatePortExternalIds(any(Port.class),
470                 any(OvsdbTerminationPointAugmentationBuilder.class));
471         verify(ovsdbPortUpdateCommand).updatePortOtherConfig(any(Port.class),
472                 any(OvsdbTerminationPointAugmentationBuilder.class));
473     }
474
475     @SuppressWarnings("unchecked")
476     @Test
477     public void testUpdateVlan() throws Exception {
478         Port port = mock(Port.class);
479         Column<GenericTableSchema, Set<Long>> column = mock(Column.class);
480         when(port.getTagColumn()).thenReturn(column);
481         Set<Long> vlanId = new HashSet<>();
482         vlanId.add((long) 808);
483         when(column.getData()).thenReturn(vlanId);
484         PowerMockito.whenNew(VlanId.class).withAnyArguments().thenReturn(mock(VlanId.class));
485         OvsdbTerminationPointAugmentationBuilder ovsdbTerminationPointBuilder = mock(
486                 OvsdbTerminationPointAugmentationBuilder.class);
487         when(ovsdbTerminationPointBuilder.setVlanTag(any(VlanId.class))).thenReturn(ovsdbTerminationPointBuilder);
488         Whitebox.invokeMethod(ovsdbPortUpdateCommand, "updateVlan", port, ovsdbTerminationPointBuilder);
489         verify(ovsdbTerminationPointBuilder).setVlanTag(any(VlanId.class));
490     }
491
492     @SuppressWarnings("unchecked")
493     @Test
494     public void testUpdateVlanTrunks() throws Exception {
495         Port port = mock(Port.class);
496         Column<GenericTableSchema, Set<Long>> column = mock(Column.class);
497         when(port.getTrunksColumn()).thenReturn(column);
498         Set<Long> portTrunks = new HashSet<>();
499         portTrunks.add((long) 300);
500         when(column.getData()).thenReturn(portTrunks);
501
502         TrunksBuilder trunksBuilder = mock(TrunksBuilder.class);
503         PowerMockito.whenNew(TrunksBuilder.class).withNoArguments().thenReturn(trunksBuilder);
504         PowerMockito.whenNew(VlanId.class).withAnyArguments().thenReturn(mock(VlanId.class));
505         when(trunksBuilder.setTrunk(any(VlanId.class))).thenReturn(trunksBuilder);
506         OvsdbTerminationPointAugmentationBuilder ovsdbTerminationPointBuilder = mock(
507                 OvsdbTerminationPointAugmentationBuilder.class);
508         when(ovsdbTerminationPointBuilder.setTrunks(any(List.class))).thenReturn(ovsdbTerminationPointBuilder);
509         Whitebox.invokeMethod(ovsdbPortUpdateCommand, "updateVlanTrunks", port, ovsdbTerminationPointBuilder);
510         verify(trunksBuilder).setTrunk(any(VlanId.class));
511         verify(ovsdbTerminationPointBuilder).setTrunks(any(List.class));
512     }
513
514     @SuppressWarnings("unchecked")
515     @Test
516     public void testUpdateVlanMode() throws Exception {
517         Port port = mock(Port.class);
518         Column<GenericTableSchema, Set<String>> column = mock(Column.class);
519         when(port.getVlanModeColumn()).thenReturn(column);
520         Set<String> set = new HashSet<>();
521         set.add(VLAN_MODE_ACCESS);
522         when(column.getData()).thenReturn(set);
523         OvsdbTerminationPointAugmentationBuilder ovsdbTerminationPointBuilder = mock(
524                 OvsdbTerminationPointAugmentationBuilder.class);
525         when(ovsdbTerminationPointBuilder.setVlanMode(OvsdbPortInterfaceAttributes.VlanMode.Access))
526                 .thenReturn(ovsdbTerminationPointBuilder);
527         Whitebox.invokeMethod(ovsdbPortUpdateCommand, "updateVlanMode", port, ovsdbTerminationPointBuilder);
528         verify(ovsdbTerminationPointBuilder).setVlanMode(any(VlanMode.class));
529     }
530
531     @SuppressWarnings("unchecked")
532     @Test
533     public void testUpdateOfPort() throws Exception {
534         Interface interf = mock(Interface.class);
535         Set<Long> ofPorts = new HashSet<>();
536         ofPorts.add((long) 10000);
537         Column<GenericTableSchema, Set<Long>> column = mock(Column.class);
538         when(interf.getOpenFlowPortColumn()).thenReturn(column);
539         when(column.getData()).thenReturn(ofPorts);
540         OvsdbTerminationPointAugmentationBuilder ovsdbTerminationPointBuilder = mock(
541                 OvsdbTerminationPointAugmentationBuilder.class);
542         when(ovsdbTerminationPointBuilder.setOfport(any(Long.class))).thenReturn(ovsdbTerminationPointBuilder);
543         when(interf.getName()).thenReturn(INTERFACE_NAME);
544         Whitebox.invokeMethod(ovsdbPortUpdateCommand, "updateOfPort", interf, ovsdbTerminationPointBuilder);
545         verify(ovsdbTerminationPointBuilder).setOfport(any(Long.class));
546     }
547
548     @SuppressWarnings("unchecked")
549     @Test
550     public void testUpdateOfPortRequest() throws Exception {
551         Interface interf = mock(Interface.class);
552         Set<Long> ofPortRequests = new HashSet<>();
553         ofPortRequests.add((long) 10000);
554         Column<GenericTableSchema, Set<Long>> column = mock(Column.class);
555         when(interf.getOpenFlowPortRequestColumn()).thenReturn(column);
556         when(column.getData()).thenReturn(ofPortRequests);
557         OvsdbTerminationPointAugmentationBuilder ovsdbTerminationPointBuilder = mock(
558                 OvsdbTerminationPointAugmentationBuilder.class);
559         when(ovsdbTerminationPointBuilder.setOfportRequest(any(Integer.class)))
560                 .thenReturn(ovsdbTerminationPointBuilder);
561         when(interf.getName()).thenReturn(INTERFACE_NAME);
562         Whitebox.invokeMethod(ovsdbPortUpdateCommand, "updateOfPortRequest", interf, ovsdbTerminationPointBuilder);
563         verify(ovsdbTerminationPointBuilder).setOfportRequest(any(Integer.class));
564     }
565
566     @SuppressWarnings("unchecked")
567     @Test
568     public void testUpdateInterfaceExternalIds() throws Exception {
569         Interface interf = mock(Interface.class);
570         Column<GenericTableSchema, Map<String, String>> column = mock(Column.class);
571         when(interf.getExternalIdsColumn()).thenReturn(column);
572         Map<String, String> map = new HashMap<>();
573         when(column.getData()).thenReturn(map);
574         map.put(EXTERNAL_ID_KEY, EXTERNAL_ID_VALUE);
575         when(column.getData()).thenReturn(map);
576
577         InterfaceExternalIdsBuilder interfaceExternalIdsBuilder = mock(InterfaceExternalIdsBuilder.class);
578         PowerMockito.whenNew(InterfaceExternalIdsBuilder.class).withNoArguments()
579                 .thenReturn(interfaceExternalIdsBuilder);
580
581         when(interfaceExternalIdsBuilder.setExternalIdKey(anyString())).thenReturn(interfaceExternalIdsBuilder);
582         when(interfaceExternalIdsBuilder.setExternalIdValue(anyString())).thenReturn(interfaceExternalIdsBuilder);
583         when(interfaceExternalIdsBuilder.build()).thenReturn(mock(InterfaceExternalIds.class));
584         OvsdbTerminationPointAugmentationBuilder ovsdbTerminationPointBldr = mock(
585                 OvsdbTerminationPointAugmentationBuilder.class);
586         when(ovsdbTerminationPointBldr.setInterfaceExternalIds(any(List.class))).thenReturn(ovsdbTerminationPointBldr);
587
588         Whitebox.invokeMethod(ovsdbPortUpdateCommand, "updateInterfaceExternalIds", interf, ovsdbTerminationPointBldr);
589         verify(interfaceExternalIdsBuilder).setExternalIdKey(anyString());
590         verify(interfaceExternalIdsBuilder).setExternalIdValue(anyString());
591
592     }
593
594     @SuppressWarnings("unchecked")
595     @Test
596     public void testUpdatePortExternalIds() throws Exception {
597         Port port = mock(Port.class);
598         Column<GenericTableSchema, Map<String, String>> column = mock(Column.class);
599         when(port.getExternalIdsColumn()).thenReturn(column);
600         Map<String, String> map = new HashMap<>();
601         when(column.getData()).thenReturn(map);
602         map.put(EXTERNAL_ID_KEY, EXTERNAL_ID_VALUE);
603         when(column.getData()).thenReturn(map);
604
605         PortExternalIdsBuilder portExternalIdsBuilder = mock(PortExternalIdsBuilder.class);
606         PowerMockito.whenNew(PortExternalIdsBuilder.class).withNoArguments().thenReturn(portExternalIdsBuilder);
607
608         when(portExternalIdsBuilder.setExternalIdKey(anyString())).thenReturn(portExternalIdsBuilder);
609         when(portExternalIdsBuilder.setExternalIdValue(anyString())).thenReturn(portExternalIdsBuilder);
610         when(portExternalIdsBuilder.build()).thenReturn(mock(PortExternalIds.class));
611         OvsdbTerminationPointAugmentationBuilder ovsdbTerminationPointBuilder = mock(
612                 OvsdbTerminationPointAugmentationBuilder.class);
613         when(ovsdbTerminationPointBuilder.setPortExternalIds(any(List.class))).thenReturn(ovsdbTerminationPointBuilder);
614
615         Whitebox.invokeMethod(ovsdbPortUpdateCommand, "updatePortExternalIds", port, ovsdbTerminationPointBuilder);
616         verify(portExternalIdsBuilder).setExternalIdKey(anyString());
617         verify(portExternalIdsBuilder).setExternalIdValue(anyString());
618     }
619
620     @SuppressWarnings("unchecked")
621     @Test
622     public void testUpdatePortOtherConfig() throws Exception {
623         Port port = mock(Port.class);
624         Column<GenericTableSchema, Map<String, String>> column = mock(Column.class);
625         when(port.getOtherConfigColumn()).thenReturn(column);
626         Map<String, String> map = new HashMap<>();
627         map.put(OTHER_CONFIG_KEY, OTHER_CONFIG_VALUE);
628         when(column.getData()).thenReturn(map);
629
630         PortOtherConfigsBuilder portOtherConfigsBuilder = mock(PortOtherConfigsBuilder.class);
631         PowerMockito.whenNew(PortOtherConfigsBuilder.class).withNoArguments().thenReturn(portOtherConfigsBuilder);
632
633         when(portOtherConfigsBuilder.setOtherConfigKey(anyString())).thenReturn(portOtherConfigsBuilder);
634         when(portOtherConfigsBuilder.setOtherConfigValue(anyString())).thenReturn(portOtherConfigsBuilder);
635         when(portOtherConfigsBuilder.build()).thenReturn(mock(PortOtherConfigs.class));
636         OvsdbTerminationPointAugmentationBuilder ovsdbTerminationPointBuilder = mock(
637                 OvsdbTerminationPointAugmentationBuilder.class);
638         when(ovsdbTerminationPointBuilder.setInterfaceOtherConfigs(any(List.class)))
639                 .thenReturn(ovsdbTerminationPointBuilder);
640
641         Whitebox.invokeMethod(ovsdbPortUpdateCommand, "updatePortOtherConfig", port, ovsdbTerminationPointBuilder);
642         verify(portOtherConfigsBuilder).setOtherConfigKey(anyString());
643         verify(portOtherConfigsBuilder).setOtherConfigValue(anyString());
644     }
645
646     @SuppressWarnings("unchecked")
647     @Test
648     public void testUpdateInterfaceOtherConfig() throws Exception {
649         Interface interf = mock(Interface.class);
650         Map<String, String> interfaceOtherConfigMap = new HashMap<>();
651         interfaceOtherConfigMap.put(OTHER_CONFIG_KEY, OTHER_CONFIG_VALUE);
652         Column<GenericTableSchema, Map<String, String>> column = mock(Column.class);
653         when(interf.getOtherConfigColumn()).thenReturn(column);
654         when(column.getData()).thenReturn(interfaceOtherConfigMap);
655
656         InterfaceOtherConfigsBuilder interfaceOtherConfigsBuilder = mock(InterfaceOtherConfigsBuilder.class);
657         PowerMockito.whenNew(InterfaceOtherConfigsBuilder.class).withNoArguments()
658                 .thenReturn(interfaceOtherConfigsBuilder);
659
660         when(interfaceOtherConfigsBuilder.setOtherConfigKey(anyString())).thenReturn(interfaceOtherConfigsBuilder);
661         when(interfaceOtherConfigsBuilder.setOtherConfigValue(anyString())).thenReturn(interfaceOtherConfigsBuilder);
662         when(interfaceOtherConfigsBuilder.build()).thenReturn(mock(InterfaceOtherConfigs.class));
663         OvsdbTerminationPointAugmentationBuilder ovsdbTerminationPointBuilder = mock(
664                 OvsdbTerminationPointAugmentationBuilder.class);
665         when(ovsdbTerminationPointBuilder.setInterfaceOtherConfigs(any(List.class)))
666                 .thenReturn(ovsdbTerminationPointBuilder);
667
668         Whitebox.invokeMethod(ovsdbPortUpdateCommand, "updateInterfaceOtherConfig", interf,
669                 ovsdbTerminationPointBuilder);
670         verify(interfaceOtherConfigsBuilder).setOtherConfigKey(anyString());
671         verify(interfaceOtherConfigsBuilder).setOtherConfigValue(anyString());
672     }
673
674     @SuppressWarnings("unchecked")
675     @Test
676     // TODO This test needs to be re-done
677     @Ignore("Broken mock-based test")
678     public void testGetInstanceIdentifier() throws Exception {
679         Port port = mock(Port.class);
680         Column<GenericTableSchema, Map<String, String>> column = mock(Column.class);
681         when(port.getExternalIdsColumn()).thenReturn(column);
682         Map<String, String> map = new HashMap<>();
683         map.put(SouthboundConstants.IID_EXTERNAL_ID_KEY, "opendaylight-iid");
684         when(column.getData()).thenReturn(map);
685
686         PowerMockito.mockStatic(SouthboundUtil.class);
687         InstanceIdentifier<TerminationPoint> terminationPointIId = mock(InstanceIdentifier.class);
688 //        PowerMockito
689 //                .when((InstanceIdentifier<TerminationPoint>) SouthboundUtil.deserializeInstanceIdentifier(
690 //                        any(InstanceIdentifierCodec.class), anyString()))
691 //                .thenReturn(terminationPointIId);
692         InstanceIdentifier<Node> bridgeIid = mock(InstanceIdentifier.class);
693         assertEquals(terminationPointIId,
694                 Whitebox.invokeMethod(ovsdbPortUpdateCommand, "getInstanceIdentifier", bridgeIid, port));
695     }
696 }