southbound/southbound-impl/src/test/java/org/opendaylight/ovsdb/southbound/transactions/md/OvsdbManagersUpdateCommandTest.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.Mockito.doNothing;
  14 import static org.mockito.Mockito.mock;
  15 import static org.mockito.Mockito.times;
  16 import static org.mockito.Mockito.verify;
  17 import static org.mockito.Mockito.when;
  18
  19 import com.google.common.base.Optional;
  20 import java.util.ArrayList;
  21 import java.util.HashMap;
  22 import java.util.List;
  23 import java.util.Map;
  24 import org.junit.Before;
  25 import org.junit.Test;
  26 import org.junit.runner.RunWith;
  27 import org.mockito.Mockito;
  28 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
  29 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  30 import org.opendaylight.ovsdb.lib.message.TableUpdates;
  31 import org.opendaylight.ovsdb.lib.notation.Column;
  32 import org.opendaylight.ovsdb.lib.notation.UUID;
  33 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
  34 import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
  35 import org.opendaylight.ovsdb.schema.openvswitch.Manager;
  36 import org.opendaylight.ovsdb.schema.openvswitch.OpenVSwitch;
  37 import org.opendaylight.ovsdb.southbound.OvsdbConnectionInstance;
  38 import org.opendaylight.ovsdb.southbound.SouthboundMapper;
  39 import org.opendaylight.ovsdb.southbound.SouthboundUtil;
  40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagerEntry;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagerEntryKey;
  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.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
  46 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  47 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
  48 import org.powermock.api.mockito.PowerMockito;
  49 import org.powermock.api.support.membermodification.MemberModifier;
  50 import org.powermock.core.classloader.annotations.PrepareForTest;
  51 import org.powermock.modules.junit4.PowerMockRunner;
  52 import org.powermock.reflect.Whitebox;
  53
  54 @RunWith(PowerMockRunner.class)
  55 @PrepareForTest({ OvsdbManagersUpdateCommand.class, SouthboundMapper.class, SouthboundUtil.class,
  56         InstanceIdentifier.class })
  57 public class OvsdbManagersUpdateCommandTest {
  58
  59     private static final String TARGET_COLUMN_DATA = "Manager Column";
  60     private static final String NODE_ID = "Node ID String";
  61     private Map<UUID, Manager> updatedManagerRows;
  62     private Map<UUID, OpenVSwitch> updatedOpenVSwitchRows;
  63     private OvsdbManagersUpdateCommand ovsdbManagersUpdateCommand;
  64
  65     @Before
  66     public void setUp() {
  67         ovsdbManagersUpdateCommand = PowerMockito.mock(OvsdbManagersUpdateCommand.class, Mockito.CALLS_REAL_METHODS);
  68     }
  69
  70     @Test
  71     public void testOvsdbManagersUpdateCommand() {
  72         OvsdbConnectionInstance key = mock(OvsdbConnectionInstance.class);
  73         TableUpdates updates = mock(TableUpdates.class);
  74         DatabaseSchema dbSchema = mock(DatabaseSchema.class);
  75         OvsdbManagersUpdateCommand ovsdbManagersUpdateCommand1 = new OvsdbManagersUpdateCommand(key, updates, dbSchema);
  76         assertEquals(key, Whitebox.getInternalState(ovsdbManagersUpdateCommand1, "key"));
  77         assertEquals(updates, Whitebox.getInternalState(ovsdbManagersUpdateCommand1, "updates"));
  78         assertEquals(dbSchema, Whitebox.getInternalState(ovsdbManagersUpdateCommand1, "dbSchema"));
  79     }
  80
  81     @Test
  82     public void testExecute() throws Exception {
  83         updatedManagerRows = new HashMap<>();
  84         updatedManagerRows.put(mock(UUID.class), mock(Manager.class));
  85         MemberModifier.field(OvsdbManagersUpdateCommand.class, "updatedManagerRows").set(ovsdbManagersUpdateCommand,
  86                 updatedManagerRows);
  87         Map<Uri, Manager> updatedManagerRowsWithUri = new HashMap<>();
  88         PowerMockito.doReturn(updatedManagerRowsWithUri).when(ovsdbManagersUpdateCommand, "getUriManagerMap",
  89                 any(Map.class));
  90
  91         updatedOpenVSwitchRows = new HashMap<>();
  92         updatedOpenVSwitchRows.put(mock(UUID.class), mock(OpenVSwitch.class));
  93         MemberModifier.field(OvsdbManagersUpdateCommand.class, "updatedOpenVSwitchRows").set(ovsdbManagersUpdateCommand,
  94                 updatedOpenVSwitchRows);
  95
  96         // mock updateManagers()
  97         PowerMockito.doNothing().when(ovsdbManagersUpdateCommand, "updateManagers", any(ReadWriteTransaction.class),
  98                 any(Map.class), any(Map.class));
  99
 100         ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
 101         ovsdbManagersUpdateCommand.execute(transaction);
 102         PowerMockito.verifyPrivate(ovsdbManagersUpdateCommand).invoke("getUriManagerMap", any(Map.class));
 103         PowerMockito.verifyPrivate(ovsdbManagersUpdateCommand).invoke("updateManagers", any(ReadWriteTransaction.class),
 104                 any(Map.class), any(Map.class));
 105     }
 106
 107     @SuppressWarnings("unchecked")
 108     @Test
 109     public void testUpdateManagers() throws Exception {
 110         Map<UUID, OpenVSwitch> updatedOpenVSwitchRows = new HashMap<>();
 111         OpenVSwitch openVSwitch = mock(OpenVSwitch.class);
 112         updatedOpenVSwitchRows.put(mock(UUID.class), openVSwitch);
 113
 114         PowerMockito.mockStatic(SouthboundMapper.class);
 115         List<ManagerEntry> managerEntries = new ArrayList<>();
 116         managerEntries.add(mock(ManagerEntry.class));
 117         when(SouthboundMapper.createManagerEntries(any(OpenVSwitch.class), any(Map.class))).thenReturn(managerEntries);
 118
 119         // mock getManagerEntryIid()
 120         ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
 121         Map<UUID, Manager> updatedManagerRows = new HashMap<>();
 122         PowerMockito.doReturn(mock(InstanceIdentifier.class)).when(ovsdbManagersUpdateCommand, "getManagerEntryIid",
 123                 any(ManagerEntry.class));
 124         doNothing().when(transaction).merge(any(LogicalDatastoreType.class), any(InstanceIdentifier.class),
 125                 any(ManagerEntry.class));
 126         Whitebox.invokeMethod(ovsdbManagersUpdateCommand, "updateManagers", transaction, updatedManagerRows,
 127                 updatedOpenVSwitchRows);
 128         verify(transaction).merge(any(LogicalDatastoreType.class), any(InstanceIdentifier.class),
 129                 any(ManagerEntry.class));
 130     }
 131
 132     @SuppressWarnings("unchecked")
 133     @Test
 134     public void testUpdateManagers1() throws Exception {
 135         OvsdbConnectionInstance ovsdbConnectionInstance = mock(OvsdbConnectionInstance.class);
 136         when(ovsdbManagersUpdateCommand.getOvsdbConnectionInstance()).thenReturn(ovsdbConnectionInstance);
 137         InstanceIdentifier<Node> connectionIId = mock(InstanceIdentifier.class);
 138         when(ovsdbConnectionInstance.getInstanceIdentifier()).thenReturn(connectionIId);
 139
 140         Optional<Node> ovsdbNode = mock(Optional.class);
 141         PowerMockito.mockStatic(SouthboundUtil.class);
 142         when(SouthboundUtil.readNode(any(ReadWriteTransaction.class), any(InstanceIdentifier.class)))
 143                 .thenReturn(ovsdbNode);
 144         when(ovsdbNode.isPresent()).thenReturn(true);
 145         when(ovsdbNode.get()).thenReturn(mock(Node.class));
 146         ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
 147         Map<Uri, Manager> updatedManagerRows = new HashMap<>();
 148         Whitebox.invokeMethod(ovsdbManagersUpdateCommand, "updateManagers", transaction, updatedManagerRows);
 149         verify(ovsdbNode, times(2)).get();
 150     }
 151
 152     @Test
 153     public void testGetManagerEntryIid() throws Exception {
 154         ManagerEntry managerEntry = mock(ManagerEntry.class);
 155         OvsdbConnectionInstance client = mock(OvsdbConnectionInstance.class, Mockito.RETURNS_DEEP_STUBS);
 156         when(ovsdbManagersUpdateCommand.getOvsdbConnectionInstance()).thenReturn(client);
 157         when(client.getNodeKey().getNodeId().getValue()).thenReturn(NODE_ID);
 158         PowerMockito.whenNew(Uri.class).withAnyArguments().thenReturn(mock(Uri.class));
 159
 160         NodeId nodeId = mock(NodeId.class);
 161         PowerMockito.whenNew(NodeId.class).withAnyArguments().thenReturn(nodeId);
 162         NodeKey nodeKey = mock(NodeKey.class);
 163         PowerMockito.whenNew(NodeKey.class).withAnyArguments().thenReturn(nodeKey);
 164         when(managerEntry.getKey()).thenReturn(mock(ManagerEntryKey.class));
 165         assertEquals(KeyedInstanceIdentifier.class,
 166                 Whitebox.invokeMethod(ovsdbManagersUpdateCommand, "getManagerEntryIid", managerEntry).getClass());
 167     }
 168
 169     @SuppressWarnings("unchecked")
 170     @Test
 171     public void testGetUriManagerMap() throws Exception {
 172         Map<UUID,Manager> uuidManagerMap = new HashMap<>();
 173         Manager manager = mock(Manager.class);
 174         uuidManagerMap.put(mock(UUID.class), manager);
 175
 176         Column<GenericTableSchema, String> column = mock(Column.class);
 177         when(manager.getTargetColumn()).thenReturn(column);
 178         when(column.getData()).thenReturn(TARGET_COLUMN_DATA);
 179
 180         Uri uri = mock(Uri.class);
 181         PowerMockito.whenNew(Uri.class).withAnyArguments().thenReturn(uri);
 182         Map<Uri, Manager> testUriManagerMap = new HashMap<>();
 183         testUriManagerMap.put(uri, manager);
 184         assertEquals(testUriManagerMap,
 185                 Whitebox.invokeMethod(ovsdbManagersUpdateCommand, "getUriManagerMap", uuidManagerMap));
 186     }
 187 }