2 * Copyright (c) 2015 Inocybe Technologies 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
9 package org.opendaylight.ovsdb.southbound.transactions.md;
11 import static org.junit.Assert.assertEquals;
12 import static org.mockito.Matchers.any;
13 import static org.mockito.Mockito.doNothing;
14 import static org.mockito.Mockito.mock;
15 import static org.mockito.Mockito.verify;
16 import static org.mockito.Mockito.when;
18 import com.google.common.base.Optional;
19 import java.util.ArrayList;
20 import java.util.HashMap;
21 import java.util.List;
23 import org.junit.Before;
24 import org.junit.Test;
25 import org.junit.runner.RunWith;
26 import org.mockito.Mockito;
27 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
28 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
29 import org.opendaylight.ovsdb.lib.message.TableUpdates;
30 import org.opendaylight.ovsdb.lib.notation.Column;
31 import org.opendaylight.ovsdb.lib.notation.UUID;
32 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
33 import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
34 import org.opendaylight.ovsdb.schema.openvswitch.Bridge;
35 import org.opendaylight.ovsdb.schema.openvswitch.Controller;
36 import org.opendaylight.ovsdb.southbound.OvsdbConnectionInstance;
37 import org.opendaylight.ovsdb.southbound.SouthboundMapper;
38 import org.opendaylight.ovsdb.southbound.SouthboundUtil;
39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeRef;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeAugmentation;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.ControllerEntry;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.ControllerEntryKey;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagedNodeEntry;
45 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
46 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
47 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
48 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
49 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
50 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
51 import org.powermock.api.mockito.PowerMockito;
52 import org.powermock.api.support.membermodification.MemberMatcher;
53 import org.powermock.api.support.membermodification.MemberModifier;
54 import org.powermock.core.classloader.annotations.PrepareForTest;
55 import org.powermock.modules.junit4.PowerMockRunner;
56 import org.powermock.reflect.Whitebox;
58 @RunWith(PowerMockRunner.class)
59 @PrepareForTest({ OvsdbControllerUpdateCommand.class, SouthboundMapper.class, SouthboundUtil.class,
60 InstanceIdentifier.class })
61 public class OvsdbControllerUpdateCommandTest {
62 private Map<UUID, Controller> updatedControllerRows;
63 private Map<UUID, Bridge> updatedBridgeRows;
64 private OvsdbControllerUpdateCommand ovsdbControllerUpdateCommand;
66 private static final String BRIDGE_NAME = "br-int";
67 private static final String NODE_ID = "OF|00:00:00:0c:29:70:45:9b";
71 ovsdbControllerUpdateCommand = PowerMockito.mock(OvsdbControllerUpdateCommand.class,
72 Mockito.CALLS_REAL_METHODS);
76 public void testOvsdbControllerUpdateCommand() {
77 OvsdbConnectionInstance key = mock(OvsdbConnectionInstance.class);
78 TableUpdates updates = mock(TableUpdates.class);
79 DatabaseSchema dbSchema = mock(DatabaseSchema.class);
80 OvsdbControllerUpdateCommand ovsdbControllerUpdateCommand1 = new OvsdbControllerUpdateCommand(key, updates,
82 assertEquals(key, Whitebox.getInternalState(ovsdbControllerUpdateCommand1, "key"));
83 assertEquals(updates, Whitebox.getInternalState(ovsdbControllerUpdateCommand1, "updates"));
84 assertEquals(dbSchema, Whitebox.getInternalState(ovsdbControllerUpdateCommand1, "dbSchema"));
88 public void testExecute() throws Exception {
89 updatedControllerRows = new HashMap<>();
90 updatedControllerRows.put(mock(UUID.class), mock(Controller.class));
91 MemberModifier.field(OvsdbControllerUpdateCommand.class, "updatedControllerRows")
92 .set(ovsdbControllerUpdateCommand, updatedControllerRows);
93 MemberModifier.suppress(MemberMatcher.method(OvsdbControllerUpdateCommand.class, "updateController",
94 ReadWriteTransaction.class, Map.class, Map.class));
95 MemberModifier.suppress(MemberMatcher.method(OvsdbControllerUpdateCommand.class, "updateController",
96 ReadWriteTransaction.class, Map.class));
98 // updatedBridgeRows null case
99 ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
100 ovsdbControllerUpdateCommand.execute(transaction);
101 PowerMockito.verifyPrivate(ovsdbControllerUpdateCommand).invoke("updateController",
102 any(ReadWriteTransaction.class), any(Map.class));
104 // updatedBridgeRows not null case
105 updatedBridgeRows = new HashMap<>();
106 updatedBridgeRows.put(mock(UUID.class), mock(Bridge.class));
107 MemberModifier.field(OvsdbControllerUpdateCommand.class, "updatedBridgeRows").set(ovsdbControllerUpdateCommand,
111 @SuppressWarnings("unchecked")
113 public void testUpdateController1() throws Exception {
114 Map<UUID, Bridge> updatedBridgeRows = new HashMap<>();
115 Bridge bridge = mock(Bridge.class);
116 updatedBridgeRows.put(mock(UUID.class), bridge);
118 PowerMockito.mockStatic(SouthboundMapper.class);
119 List<ControllerEntry> controllerEntries = new ArrayList<>();
120 controllerEntries.add(mock(ControllerEntry.class));
121 when(SouthboundMapper.createControllerEntries(any(Bridge.class), any(Map.class))).thenReturn(controllerEntries);
122 Column<GenericTableSchema, String> column = mock(Column.class);
123 when(bridge.getNameColumn()).thenReturn(column);
124 when(column.getData()).thenReturn(BRIDGE_NAME);
126 // suppress call to getControllerEntryIid()
127 ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
128 Map<UUID, Controller> updatedControllerRows = new HashMap<>();
129 MemberModifier.suppress(MemberMatcher.method(OvsdbControllerUpdateCommand.class, "getControllerEntryIid",
130 ControllerEntry.class, String.class));
131 doNothing().when(transaction).merge(any(LogicalDatastoreType.class), any(InstanceIdentifier.class),
132 any(ControllerEntry.class));
133 Whitebox.invokeMethod(ovsdbControllerUpdateCommand, "updateController", transaction, updatedControllerRows,
135 verify(transaction).merge(any(LogicalDatastoreType.class), any(InstanceIdentifier.class),
136 any(ControllerEntry.class));
137 verify(bridge).getNameColumn();
140 @SuppressWarnings("unchecked")
142 public void testUpdateController2() throws Exception {
143 ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
144 Map<UUID, Controller> updatedControllerRows = new HashMap<>();
145 Map<InstanceIdentifier<Node>, Node> bridgeNodes = new HashMap<>();
146 Node node = mock(Node.class);
147 InstanceIdentifier<Node> bridgeIid = mock(InstanceIdentifier.class);
148 bridgeNodes.put(bridgeIid, node);
149 PowerMockito.doReturn(bridgeNodes).when(ovsdbControllerUpdateCommand, "getBridgeNodes",
150 any(ReadWriteTransaction.class));
152 Whitebox.invokeMethod(ovsdbControllerUpdateCommand, "updateController", transaction, updatedControllerRows);
153 PowerMockito.verifyPrivate(ovsdbControllerUpdateCommand).invoke("getBridgeNodes",
154 any(ReadWriteTransaction.class));
157 @SuppressWarnings("unchecked")
159 public void testGetBridgeNodes() throws Exception {
160 OvsdbConnectionInstance ovsdbConnectionInstance = mock(OvsdbConnectionInstance.class);
161 when(ovsdbControllerUpdateCommand.getOvsdbConnectionInstance()).thenReturn(ovsdbConnectionInstance);
162 InstanceIdentifier<Node> connectionIId = mock(InstanceIdentifier.class);
163 when(ovsdbConnectionInstance.getInstanceIdentifier()).thenReturn(connectionIId);
164 PowerMockito.mockStatic(SouthboundUtil.class);
165 Optional<Node> ovsdbNode = mock(Optional.class);
166 ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
167 when(SouthboundUtil.readNode(transaction, connectionIId)).thenReturn(ovsdbNode);
168 when(ovsdbNode.isPresent()).thenReturn(true);
169 Node node = mock(Node.class);
170 when(ovsdbNode.get()).thenReturn(node);
171 OvsdbNodeAugmentation ovsdbNodeAugmentation = mock(OvsdbNodeAugmentation.class);
172 when(node.getAugmentation(OvsdbNodeAugmentation.class)).thenReturn(ovsdbNodeAugmentation);
174 List<ManagedNodeEntry> managedNodeEntries = new ArrayList<>();
175 ManagedNodeEntry managedNodeEntry = mock(ManagedNodeEntry.class);
176 managedNodeEntries.add(managedNodeEntry);
177 when(ovsdbNodeAugmentation.getManagedNodeEntry()).thenReturn(managedNodeEntries);
178 InstanceIdentifier<Node> bridgeIid = mock(InstanceIdentifier.class);
179 OvsdbBridgeRef ovsdbBridgeRef = mock(OvsdbBridgeRef.class);
180 when(managedNodeEntry.getBridgeRef()).thenReturn(ovsdbBridgeRef);
181 when((InstanceIdentifier<Node>) ovsdbBridgeRef.getValue()).thenReturn(bridgeIid);
182 Optional<Node> bridgeNode = mock(Optional.class);
183 when(SouthboundUtil.readNode(transaction, bridgeIid)).thenReturn(bridgeNode);
184 when(bridgeNode.isPresent()).thenReturn(true);
185 when(bridgeNode.get()).thenReturn(node);
187 Map<InstanceIdentifier<Node>, Node> testBridgeNodes = new HashMap<>();
188 testBridgeNodes.put(bridgeIid, node);
190 //verify if getBridgeNodes() returns expected value
191 assertEquals(testBridgeNodes,
192 Whitebox.invokeMethod(ovsdbControllerUpdateCommand, "getBridgeNodes", transaction));
196 public void testGetControllerEntryIid() throws Exception {
197 ControllerEntry controllerEntry = mock(ControllerEntry.class);
198 OvsdbConnectionInstance client = mock(OvsdbConnectionInstance.class);
199 when(ovsdbControllerUpdateCommand.getOvsdbConnectionInstance()).thenReturn(client);
200 NodeKey nodeKey = mock(NodeKey.class);
201 when(client.getNodeKey()).thenReturn(nodeKey);
202 NodeId nodeId = mock(NodeId.class);
203 when(nodeKey.getNodeId()).thenReturn(nodeId);
204 when(nodeId.getValue()).thenReturn(NODE_ID);
205 PowerMockito.whenNew(Uri.class).withAnyArguments().thenReturn(mock(Uri.class));
206 PowerMockito.whenNew(NodeId.class).withAnyArguments().thenReturn(nodeId);
207 PowerMockito.whenNew(NodeKey.class).withAnyArguments().thenReturn(nodeKey);
208 PowerMockito.whenNew(TopologyKey.class).withAnyArguments().thenReturn(mock(TopologyKey.class));
209 //PowerMockito.suppress(MemberMatcher.methodsDeclaredIn(InstanceIdentifier.class));
210 when(controllerEntry.getKey()).thenReturn(mock(ControllerEntryKey.class));
211 assertEquals(KeyedInstanceIdentifier.class, (Whitebox
212 .invokeMethod(ovsdbControllerUpdateCommand, "getControllerEntryIid", controllerEntry, BRIDGE_NAME)