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.doReturn;
15 import static org.mockito.Mockito.mock;
16 import static org.mockito.Mockito.times;
17 import static org.mockito.Mockito.verify;
18 import static org.mockito.Mockito.when;
20 import java.util.ArrayList;
21 import java.util.HashMap;
22 import java.util.HashSet;
23 import java.util.List;
26 import org.junit.Before;
27 import org.junit.Test;
28 import org.junit.runner.RunWith;
29 import org.mockito.Mockito;
30 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
31 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
32 import org.opendaylight.ovsdb.lib.message.TableUpdates;
33 import org.opendaylight.ovsdb.lib.notation.Column;
34 import org.opendaylight.ovsdb.lib.notation.UUID;
35 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
36 import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
37 import org.opendaylight.ovsdb.schema.openvswitch.Manager;
38 import org.opendaylight.ovsdb.schema.openvswitch.OpenVSwitch;
39 import org.opendaylight.ovsdb.southbound.OvsdbConnectionInstance;
40 import org.opendaylight.ovsdb.southbound.SouthboundMapper;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.BridgeOtherConfigs;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagerEntry;
43 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
44 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
45 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
46 import org.powermock.api.mockito.PowerMockito;
47 import org.powermock.api.support.membermodification.MemberModifier;
48 import org.powermock.core.classloader.annotations.PrepareForTest;
49 import org.powermock.modules.junit4.PowerMockRunner;
50 import org.powermock.reflect.Whitebox;
53 @PrepareForTest({SouthboundMapper.class, OvsdbManagersRemovedCommand.class})
54 @RunWith(PowerMockRunner.class)
55 public class OvsdbManagersRemovedCommandTest {
57 private static final String TARGET_COLUMN_DATA = "Target Column Data";
58 private OvsdbManagersRemovedCommand ovsdbManagersRemovedCommand;
59 private Map<UUID, OpenVSwitch> updatedOpenVSwitchRows;
60 private Map<UUID, OpenVSwitch> oldOpenVSwitchRows;
61 private Map<UUID, Manager> updatedManagerRows;
65 ovsdbManagersRemovedCommand = PowerMockito.mock(OvsdbManagersRemovedCommand.class, Mockito.CALLS_REAL_METHODS);
69 public void testOvsdbManagersRemovedCommand() {
70 OvsdbConnectionInstance key = mock(OvsdbConnectionInstance.class);
71 TableUpdates updates = mock(TableUpdates.class);
72 DatabaseSchema dbSchema = mock(DatabaseSchema.class);
73 OvsdbManagersRemovedCommand ovsdbManagersRemovedCommand1 = new OvsdbManagersRemovedCommand(key, updates,
75 assertEquals(key, Whitebox.getInternalState(ovsdbManagersRemovedCommand1, "key"));
76 assertEquals(updates, Whitebox.getInternalState(ovsdbManagersRemovedCommand1, "updates"));
77 assertEquals(dbSchema, Whitebox.getInternalState(ovsdbManagersRemovedCommand1, "dbSchema"));
80 @SuppressWarnings("unchecked")
82 public void testExecute() throws Exception {
83 updatedOpenVSwitchRows = new HashMap<>();
84 UUID uuid = mock(UUID.class);
85 OpenVSwitch openVSwitch = mock(OpenVSwitch.class);
86 updatedOpenVSwitchRows.put(uuid, openVSwitch);
87 MemberModifier.field(OvsdbManagersRemovedCommand.class, "updatedOpenVSwitchRows")
88 .set(ovsdbManagersRemovedCommand, updatedOpenVSwitchRows);
89 PowerMockito.mockStatic(SouthboundMapper.class);
90 OvsdbConnectionInstance ovsdbConnectionInstance = mock(OvsdbConnectionInstance.class);
91 when(ovsdbManagersRemovedCommand.getOvsdbConnectionInstance()).thenReturn(ovsdbConnectionInstance);
92 when(ovsdbConnectionInstance.getNodeId()).thenReturn(mock(NodeId.class));
93 when(SouthboundMapper.createInstanceIdentifier(any(NodeId.class))).thenReturn(mock(InstanceIdentifier.class));
95 doNothing().when(ovsdbManagersRemovedCommand).deleteManagers(any(ReadWriteTransaction.class), any(List.class));
96 doReturn(mock(List.class)).when(ovsdbManagersRemovedCommand).managerEntriesToRemove(
97 any(InstanceIdentifier.class), any(OpenVSwitch.class));
99 ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
100 ovsdbManagersRemovedCommand.execute(transaction);
101 verify(ovsdbManagersRemovedCommand).deleteManagers(any(ReadWriteTransaction.class), any(List.class));
102 verify(ovsdbManagersRemovedCommand).managerEntriesToRemove(any(InstanceIdentifier.class),
103 any(OpenVSwitch.class));
106 @SuppressWarnings("unchecked")
108 public void testDeleteManagers() throws Exception {
109 ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
110 List<InstanceIdentifier<ManagerEntry>> managerEntryIids = new ArrayList<>();
111 managerEntryIids.add(mock(InstanceIdentifier.class));
112 doNothing().when(transaction).delete(any(LogicalDatastoreType.class), any(InstanceIdentifier.class));
113 Whitebox.invokeMethod(ovsdbManagersRemovedCommand, "deleteManagers", transaction, managerEntryIids);
114 verify(transaction).delete(any(LogicalDatastoreType.class), any(InstanceIdentifier.class));
117 @SuppressWarnings("unchecked")
119 public void testManagerEntriesToRemove() throws Exception {
120 OpenVSwitch openVSwitch = mock(OpenVSwitch.class);
122 UUID uuid = mock(UUID.class);
123 OpenVSwitch oldOvsdbNode = mock(OpenVSwitch.class);
124 oldOpenVSwitchRows = new HashMap<>();
125 oldOpenVSwitchRows.put(uuid, oldOvsdbNode);
126 when(openVSwitch.getUuid()).thenReturn(uuid);
127 MemberModifier.field(OvsdbManagersRemovedCommand.class, "oldOpenVSwitchRows").set(ovsdbManagersRemovedCommand,
129 Column<GenericTableSchema, Set<UUID>> column = mock(Column.class);
130 Set<UUID> set = new HashSet<>();
131 UUID controllerUuid = mock(UUID.class);
132 set.add(controllerUuid);
133 when(column.getData()).thenReturn(set);
134 when(oldOvsdbNode.getManagerOptionsColumn()).thenReturn(column);
135 when(column.getData()).thenReturn(set);
136 when(openVSwitch.getManagerOptionsColumn()).thenReturn(column);
137 InstanceIdentifier<Node> bridgeIid = mock(InstanceIdentifier.class);
138 List<InstanceIdentifier<BridgeOtherConfigs>> resultManagerEntries = Whitebox
139 .invokeMethod(ovsdbManagersRemovedCommand, "managerEntriesToRemove", bridgeIid, openVSwitch);
140 assertEquals(ArrayList.class, resultManagerEntries.getClass());
141 verify(oldOvsdbNode, times(2)).getManagerOptionsColumn();
144 @SuppressWarnings("unchecked")
146 public void testCheckIfManagerPresentInUpdatedManagersList() throws Exception {
147 Manager updatedManager = mock(Manager.class);
148 updatedManagerRows = new HashMap<>();
149 UUID uuid = mock(UUID.class);
150 updatedManagerRows.put(uuid, updatedManager);
151 MemberModifier.field(OvsdbManagersRemovedCommand.class, "updatedManagerRows").set(ovsdbManagersRemovedCommand,
153 Column<GenericTableSchema, String> column = mock(Column.class);
154 Manager removedManager = mock(Manager.class);
155 when(removedManager.getTargetColumn()).thenReturn(column);
156 when(updatedManager.getTargetColumn()).thenReturn(column);
157 when(column.getData()).thenReturn(TARGET_COLUMN_DATA);
159 assertEquals(true, Whitebox.invokeMethod(ovsdbManagersRemovedCommand,
160 "checkIfManagerPresentInUpdatedManagersList", removedManager));