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