southbound/southbound-impl/src/test/java/org/opendaylight/ovsdb/southbound/transactions/md/OvsdbControllerRemovedCommandTest.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 java.util.ArrayList;
  20 import java.util.HashMap;
  21 import java.util.HashSet;
  22 import java.util.List;
  23 import java.util.Map;
  24 import java.util.Set;
  25 import org.junit.Before;
  26 import org.junit.Test;
  27 import org.junit.runner.RunWith;
  28 import org.mockito.Mockito;
  29 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
  30 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  31 import org.opendaylight.ovsdb.lib.message.TableUpdates;
  32 import org.opendaylight.ovsdb.lib.notation.Column;
  33 import org.opendaylight.ovsdb.lib.notation.UUID;
  34 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
  35 import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
  36 import org.opendaylight.ovsdb.schema.openvswitch.Bridge;
  37 import org.opendaylight.ovsdb.southbound.InstanceIdentifierCodec;
  38 import org.opendaylight.ovsdb.southbound.OvsdbConnectionInstance;
  39 import org.opendaylight.ovsdb.southbound.SouthboundMapper;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.BridgeOtherConfigs;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.ControllerEntry;
  42 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  43 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  44 import org.powermock.api.mockito.PowerMockito;
  45 import org.powermock.api.support.membermodification.MemberMatcher;
  46 import org.powermock.api.support.membermodification.MemberModifier;
  47 import org.powermock.core.classloader.annotations.PrepareForTest;
  48 import org.powermock.modules.junit4.PowerMockRunner;
  49 import org.powermock.reflect.Whitebox;
  50
  51 @RunWith(PowerMockRunner.class)
  52 @PrepareForTest({SouthboundMapper.class, OvsdbControllerRemovedCommand.class})
  53 public class OvsdbControllerRemovedCommandTest {
  54
  55     private Map<UUID, Bridge> oldBridgeRows;
  56     private Map<UUID, Bridge> updatedBridgeRows;
  57     private OvsdbControllerRemovedCommand ovsdbControllerRemovedCommand;
  58
  59     @Before
  60     public void setUp() throws Exception {
  61         ovsdbControllerRemovedCommand = PowerMockito.mock(OvsdbControllerRemovedCommand.class,
  62                 Mockito.CALLS_REAL_METHODS);
  63     }
  64
  65     @Test
  66     public void testOvsdbControllerRemovedCommand() {
  67         OvsdbConnectionInstance key = mock(OvsdbConnectionInstance.class);
  68         TableUpdates updates = mock(TableUpdates.class);
  69         DatabaseSchema dbSchema = mock(DatabaseSchema.class);
  70         OvsdbControllerRemovedCommand ovsdbControllerRemovedCommand1 =
  71                 new OvsdbControllerRemovedCommand(mock(InstanceIdentifierCodec.class), key, updates, dbSchema);
  72         assertEquals(key, Whitebox.getInternalState(ovsdbControllerRemovedCommand1, "key"));
  73         assertEquals(updates, Whitebox.getInternalState(ovsdbControllerRemovedCommand1, "updates"));
  74         assertEquals(dbSchema, Whitebox.getInternalState(ovsdbControllerRemovedCommand1, "dbSchema"));
  75     }
  76
  77     @SuppressWarnings("unchecked")
  78     @Test
  79     public void testExecute() throws Exception {
  80         updatedBridgeRows = new HashMap<>();
  81         UUID uuid = mock(UUID.class);
  82         Bridge bridge = mock(Bridge.class);
  83         updatedBridgeRows.put(uuid, bridge);
  84         MemberModifier.field(OvsdbControllerRemovedCommand.class, "updatedBridgeRows")
  85                 .set(ovsdbControllerRemovedCommand, updatedBridgeRows);
  86         PowerMockito.mockStatic(SouthboundMapper.class);
  87         when(SouthboundMapper.createInstanceIdentifier(any(InstanceIdentifierCodec.class),
  88                 any(OvsdbConnectionInstance.class), any(Bridge.class)))
  89                 .thenReturn(mock(InstanceIdentifier.class));
  90         MemberModifier.suppress(MemberMatcher.method(OvsdbControllerRemovedCommand.class, "deleteControllers",
  91                 ReadWriteTransaction.class, List.class));
  92         MemberModifier.suppress(MemberMatcher.method(OvsdbControllerRemovedCommand.class, "controllerEntriesToRemove",
  93                 InstanceIdentifier.class, Bridge.class));
  94         ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
  95         ovsdbControllerRemovedCommand.execute(transaction);
  96         PowerMockito.verifyPrivate(ovsdbControllerRemovedCommand).invoke("deleteControllers",
  97                 any(ReadWriteTransaction.class), any(List.class));
  98         PowerMockito.verifyPrivate(ovsdbControllerRemovedCommand).invoke("controllerEntriesToRemove",
  99                 any(ReadWriteTransaction.class), any(Bridge.class));
 100     }
 101
 102     @SuppressWarnings("unchecked")
 103     @Test
 104     public void testDeleteControllers() throws Exception {
 105         ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
 106         List<InstanceIdentifier<ControllerEntry>> controllerEntryIids = new ArrayList<>();
 107
 108         doNothing().when(transaction).delete(any(LogicalDatastoreType.class), any(InstanceIdentifier.class));
 109         Whitebox.invokeMethod(ovsdbControllerRemovedCommand, "deleteControllers", transaction, controllerEntryIids);
 110         verify(transaction, times(0)).delete(any(LogicalDatastoreType.class), any(InstanceIdentifier.class));
 111
 112         InstanceIdentifier<ControllerEntry> controllerEntry = mock(InstanceIdentifier.class);
 113         controllerEntryIids.add(controllerEntry);
 114         Whitebox.invokeMethod(ovsdbControllerRemovedCommand, "deleteControllers", transaction, controllerEntryIids);
 115         verify(transaction).delete(any(LogicalDatastoreType.class), any(InstanceIdentifier.class));
 116     }
 117
 118     @SuppressWarnings("unchecked")
 119     @Test
 120     public void testControllerEntriesToRemove() throws Exception {
 121         Bridge bridge = mock(Bridge.class);
 122
 123         UUID uuid = mock(UUID.class);
 124         Bridge oldBridgeNode = mock(Bridge.class);
 125         oldBridgeRows = new HashMap<>();
 126         oldBridgeRows.put(uuid, oldBridgeNode);
 127         when(bridge.getUuid()).thenReturn(uuid);
 128         MemberModifier.field(OvsdbControllerRemovedCommand.class, "oldBridgeRows").set(ovsdbControllerRemovedCommand,
 129                 oldBridgeRows);
 130         Column<GenericTableSchema, Set<UUID>> column = mock(Column.class);
 131         Set<UUID> set = new HashSet<>();
 132         UUID controllerUuid = mock(UUID.class);
 133         set.add(controllerUuid);
 134         when(column.getData()).thenReturn(set);
 135         when(oldBridgeNode.getControllerColumn()).thenReturn(column);
 136         when(column.getData()).thenReturn(set);
 137         when(bridge.getControllerColumn()).thenReturn(column);
 138         InstanceIdentifier<Node> bridgeIid = mock(InstanceIdentifier.class);
 139         List<InstanceIdentifier<BridgeOtherConfigs>> resultControllerEntries = Whitebox
 140                 .invokeMethod(ovsdbControllerRemovedCommand, "controllerEntriesToRemove", bridgeIid, bridge);
 141         assertEquals(ArrayList.class, resultControllerEntries.getClass());
 142         verify(oldBridgeNode, times(2)).getControllerColumn();
 143     }
 144 }