southbound/southbound-impl/src/test/java/org/opendaylight/ovsdb/southbound/transactions/md/OvsdbNodeRemoveCommandTest.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.ArgumentMatchers.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;
  19
  20 import com.google.common.base.Optional;
  21 import com.google.common.util.concurrent.CheckedFuture;
  22 import java.util.ArrayList;
  23 import java.util.List;
  24 import org.junit.Before;
  25 import org.junit.Test;
  26 import org.junit.runner.RunWith;
  27 import org.mockito.Mockito;
  28 import org.mockito.runners.MockitoJUnitRunner;
  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.controller.md.sal.common.api.data.ReadFailedException;
  32 import org.opendaylight.ovsdb.lib.message.TableUpdates;
  33 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
  34 import org.opendaylight.ovsdb.southbound.OvsdbConnectionInstance;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeRef;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeAugmentation;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagedNodeEntry;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagerEntry;
  39 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  40 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  41 import org.powermock.reflect.Whitebox;
  42
  43 @RunWith(MockitoJUnitRunner.class)
  44 public class OvsdbNodeRemoveCommandTest {
  45
  46     private static final long ONE_CONNECTED_MANAGER = 1;
  47     private static final long ONE_ACTIVE_CONNECTION_IN_PASSIVE_MODE = 1;
  48     private OvsdbNodeRemoveCommand ovsdbNodeRemoveCommand;
  49
  50     @Before
  51     public void setUp() throws Exception {
  52         ovsdbNodeRemoveCommand = mock(OvsdbNodeRemoveCommand.class, Mockito.CALLS_REAL_METHODS);
  53     }
  54
  55     @Test
  56     public void testOvsdbControllerRemovedCommand() {
  57         OvsdbConnectionInstance key = mock(OvsdbConnectionInstance.class);
  58         TableUpdates updates = mock(TableUpdates.class);
  59         DatabaseSchema dbSchema = mock(DatabaseSchema.class);
  60         OvsdbNodeRemoveCommand ovsdbNodeRemoveCommand1 = new OvsdbNodeRemoveCommand(key, updates, dbSchema);
  61         assertEquals(key, Whitebox.getInternalState(ovsdbNodeRemoveCommand1, "key"));
  62         assertEquals(updates, Whitebox.getInternalState(ovsdbNodeRemoveCommand1, "updates"));
  63         assertEquals(dbSchema, Whitebox.getInternalState(ovsdbNodeRemoveCommand1, "dbSchema"));
  64     }
  65
  66     @SuppressWarnings("unchecked")
  67     @Test
  68     public void testExecute() throws Exception {
  69         ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
  70         CheckedFuture<Optional<Node>, ReadFailedException> ovsdbNodeFuture = mock(CheckedFuture.class);
  71         OvsdbConnectionInstance ovsdbConnectionInstance = mock(OvsdbConnectionInstance.class);
  72         when(ovsdbNodeRemoveCommand.getOvsdbConnectionInstance()).thenReturn(ovsdbConnectionInstance);
  73         when(ovsdbConnectionInstance.getInstanceIdentifier()).thenReturn(mock(InstanceIdentifier.class));
  74         when(transaction.read(any(LogicalDatastoreType.class), any(InstanceIdentifier.class)))
  75                 .thenReturn(ovsdbNodeFuture);
  76
  77         Node ovsdbNode = mock(Node.class);
  78         Optional<Node> ovsdbNodeOptional = Optional.of(ovsdbNode);
  79         when(ovsdbNodeFuture.get()).thenReturn(ovsdbNodeOptional);
  80         OvsdbNodeAugmentation ovsdbNodeAugmentation = mock(OvsdbNodeAugmentation.class);
  81         when(ovsdbNode.augmentation(OvsdbNodeAugmentation.class)).thenReturn(ovsdbNodeAugmentation);
  82
  83         doReturn(true).when(ovsdbNodeRemoveCommand).checkIfOnlyConnectedManager(any(OvsdbNodeAugmentation.class));
  84
  85         List<ManagedNodeEntry> listManagedNodeEntry = new ArrayList<>();
  86         ManagedNodeEntry managedNode = mock(ManagedNodeEntry.class);
  87         listManagedNodeEntry.add(managedNode);
  88         when(ovsdbNodeAugmentation.getManagedNodeEntry()).thenReturn(listManagedNodeEntry);
  89         OvsdbBridgeRef ovsdbBridgeRef = mock(OvsdbBridgeRef.class);
  90         when(managedNode.getBridgeRef()).thenReturn(ovsdbBridgeRef);
  91         when(ovsdbBridgeRef.getValue()).thenReturn(mock(InstanceIdentifier.class));
  92
  93         doNothing().when(transaction).delete(any(LogicalDatastoreType.class), any(InstanceIdentifier.class));
  94
  95         ovsdbNodeRemoveCommand.execute(transaction);
  96         verify(ovsdbNodeAugmentation, times(2)).getManagedNodeEntry();
  97         verify(transaction, times(2)).delete(any(LogicalDatastoreType.class), any(InstanceIdentifier.class));
  98     }
  99
 100     @Test
 101     public void testCheckIfOnlyConnectedManager() throws Exception {
 102         OvsdbNodeAugmentation ovsdbNodeAugmentation = mock(OvsdbNodeAugmentation.class);
 103         ManagerEntry manager = mock(ManagerEntry.class);
 104         List<ManagerEntry> listManagerEntry = new ArrayList<>();
 105         listManagerEntry.add(manager);
 106
 107         //case 1: connectedManager > ONE_CONNECTED_MANAGER
 108         ManagerEntry manager1 = mock(ManagerEntry.class);
 109         listManagerEntry.add(manager1);
 110         when(ovsdbNodeAugmentation.getManagerEntry()).thenReturn(listManagerEntry);
 111         when(manager.isConnected()).thenReturn(true, false, true);
 112         when(manager1.isConnected()).thenReturn(true, false, true);
 113         assertEquals(false,
 114                 Whitebox.invokeMethod(ovsdbNodeRemoveCommand, "checkIfOnlyConnectedManager", ovsdbNodeAugmentation));
 115
 116         // case 2: connectedManager == 0
 117         assertEquals(true,
 118                 Whitebox.invokeMethod(ovsdbNodeRemoveCommand, "checkIfOnlyConnectedManager", ovsdbNodeAugmentation));
 119
 120         // case 3: onlyConnectedManager.getNumberOfConnections().longValue() > ONE_ACTIVE_CONNECTION_IN_PASSIVE_MODE
 121         ManagerEntry onlyConnectedManager = mock(ManagerEntry.class);
 122         assertEquals(false,
 123                 Whitebox.invokeMethod(ovsdbNodeRemoveCommand, "checkIfOnlyConnectedManager", ovsdbNodeAugmentation));
 124
 125         // case 4: when all the above don't apply
 126         listManagerEntry.remove(manager1);
 127         assertEquals(true,
 128                 Whitebox.invokeMethod(ovsdbNodeRemoveCommand, "checkIfOnlyConnectedManager", ovsdbNodeAugmentation));
 129     }
 130 }