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.util.concurrent.FluentFuture;
  21 import java.util.HashMap;
  22 import java.util.Map;
  23 import java.util.Optional;
  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.junit.MockitoJUnitRunner;
  29 import org.opendaylight.mdsal.binding.api.ReadWriteTransaction;
  30 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  31 import org.opendaylight.ovsdb.lib.message.TableUpdates;
  32 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
  33 import org.opendaylight.ovsdb.southbound.OvsdbConnectionInstance;
  34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
  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.OvsdbNodeAugmentationBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagedNodeEntry;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagedNodeEntryBuilder;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagerEntry;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagerEntryKey;
  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.opendaylight.yangtools.yang.common.Uint32;
  45 import org.powermock.reflect.Whitebox;
  46
  47 @RunWith(MockitoJUnitRunner.class)
  48 public class OvsdbNodeRemoveCommandTest {
  49
  50     private static final long ONE_CONNECTED_MANAGER = 1;
  51     private static final long ONE_ACTIVE_CONNECTION_IN_PASSIVE_MODE = 1;
  52     private OvsdbNodeRemoveCommand ovsdbNodeRemoveCommand;
  53
  54     @Before
  55     public void setUp() throws Exception {
  56         ovsdbNodeRemoveCommand = mock(OvsdbNodeRemoveCommand.class, Mockito.CALLS_REAL_METHODS);
  57     }
  58
  59     @Test
  60     public void testOvsdbControllerRemovedCommand() {
  61         OvsdbConnectionInstance key = mock(OvsdbConnectionInstance.class);
  62         TableUpdates updates = mock(TableUpdates.class);
  63         DatabaseSchema dbSchema = mock(DatabaseSchema.class);
  64         OvsdbNodeRemoveCommand ovsdbNodeRemoveCommand1 = new OvsdbNodeRemoveCommand(key, updates, dbSchema);
  65         assertEquals(key, Whitebox.getInternalState(ovsdbNodeRemoveCommand1, "key"));
  66         assertEquals(updates, Whitebox.getInternalState(ovsdbNodeRemoveCommand1, "updates"));
  67         assertEquals(dbSchema, Whitebox.getInternalState(ovsdbNodeRemoveCommand1, "dbSchema"));
  68     }
  69
  70     @SuppressWarnings("unchecked")
  71     @Test
  72     public void testExecute() throws Exception {
  73         ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
  74         FluentFuture<Optional<Node>> ovsdbNodeFuture = mock(FluentFuture.class);
  75         OvsdbConnectionInstance ovsdbConnectionInstance = mock(OvsdbConnectionInstance.class);
  76         when(ovsdbNodeRemoveCommand.getOvsdbConnectionInstance()).thenReturn(ovsdbConnectionInstance);
  77         when(ovsdbConnectionInstance.getInstanceIdentifier()).thenReturn(mock(InstanceIdentifier.class));
  78         when(transaction.read(any(LogicalDatastoreType.class), any(InstanceIdentifier.class)))
  79                 .thenReturn(ovsdbNodeFuture);
  80
  81         Node ovsdbNode = mock(Node.class);
  82         Optional<Node> ovsdbNodeOptional = Optional.of(ovsdbNode);
  83         when(ovsdbNodeFuture.get()).thenReturn(ovsdbNodeOptional);
  84         OvsdbNodeAugmentation ovsdbNodeAugmentation = mock(OvsdbNodeAugmentation.class);
  85         when(ovsdbNode.augmentation(OvsdbNodeAugmentation.class)).thenReturn(ovsdbNodeAugmentation);
  86
  87         doReturn(true).when(ovsdbNodeRemoveCommand).checkIfOnlyConnectedManager(any(OvsdbNodeAugmentation.class));
  88
  89         ManagedNodeEntry managedNode = new ManagedNodeEntryBuilder()
  90                 .setBridgeRef(new OvsdbBridgeRef(mock(InstanceIdentifier.class)))
  91                 .build();
  92
  93         when(ovsdbNodeAugmentation.getManagedNodeEntry()).thenReturn(Map.of(managedNode.key(), managedNode));
  94
  95         doNothing().when(transaction).delete(any(LogicalDatastoreType.class), any(InstanceIdentifier.class));
  96
  97         ovsdbNodeRemoveCommand.execute(transaction);
  98         verify(ovsdbNodeAugmentation).getManagedNodeEntry();
  99         verify(transaction, times(2)).delete(any(LogicalDatastoreType.class), any(InstanceIdentifier.class));
 100     }
 101
 102     @Test
 103     public void testCheckIfOnlyConnectedManager() throws Exception {
 104         ManagerEntry manager = mock(ManagerEntry.class);
 105         ManagerEntry manager1 = mock(ManagerEntry.class);
 106         doReturn(new ManagerEntryKey(new Uri("manager"))).when(manager).key();
 107         doReturn(new ManagerEntryKey(new Uri("manager1"))).when(manager1).key();
 108
 109         when(manager.isConnected()).thenReturn(true, false, true);
 110         when(manager1.isConnected()).thenReturn(true, false, true);
 111         when(manager.getNumberOfConnections()).thenReturn(Uint32.ZERO);
 112
 113         Map<ManagerEntryKey, ManagerEntry> listManagerEntry = new HashMap<>();
 114         listManagerEntry.put(manager.key(), manager);
 115         listManagerEntry.put(manager1.key(), manager1);
 116
 117         OvsdbNodeAugmentation ovsdbNodeAugmentation = new OvsdbNodeAugmentationBuilder()
 118                 .setManagerEntry(listManagerEntry)
 119                 .build();
 120
 121         //case 1: connectedManager > ONE_CONNECTED_MANAGER
 122         assertEquals(false,
 123                 Whitebox.invokeMethod(ovsdbNodeRemoveCommand, "checkIfOnlyConnectedManager", ovsdbNodeAugmentation));
 124
 125         // case 2: connectedManager == 0
 126         assertEquals(true,
 127                 Whitebox.invokeMethod(ovsdbNodeRemoveCommand, "checkIfOnlyConnectedManager", ovsdbNodeAugmentation));
 128
 129         // case 3: onlyConnectedManager.getNumberOfConnections().longValue() > ONE_ACTIVE_CONNECTION_IN_PASSIVE_MODE
 130         ManagerEntry onlyConnectedManager = mock(ManagerEntry.class);
 131         assertEquals(false,
 132                 Whitebox.invokeMethod(ovsdbNodeRemoveCommand, "checkIfOnlyConnectedManager", ovsdbNodeAugmentation));
 133
 134         // case 4: when all the above don't apply
 135         listManagerEntry.remove(manager1.key());
 136         assertEquals(true,
 137                 Whitebox.invokeMethod(ovsdbNodeRemoveCommand, "checkIfOnlyConnectedManager", ovsdbNodeAugmentation));
 138     }
 139 }