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 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;
43 @RunWith(MockitoJUnitRunner.class)
44 public class OvsdbNodeRemoveCommandTest {
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;
51 public void setUp() throws Exception {
52 ovsdbNodeRemoveCommand = mock(OvsdbNodeRemoveCommand.class, Mockito.CALLS_REAL_METHODS);
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"));
66 @SuppressWarnings("unchecked")
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);
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);
83 doReturn(true).when(ovsdbNodeRemoveCommand).checkIfOnlyConnectedManager(any(OvsdbNodeAugmentation.class));
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));
93 doNothing().when(transaction).delete(any(LogicalDatastoreType.class), any(InstanceIdentifier.class));
95 ovsdbNodeRemoveCommand.execute(transaction);
96 verify(ovsdbNodeAugmentation, times(2)).getManagedNodeEntry();
97 verify(transaction, times(2)).delete(any(LogicalDatastoreType.class), any(InstanceIdentifier.class));
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);
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);
114 Whitebox.invokeMethod(ovsdbNodeRemoveCommand, "checkIfOnlyConnectedManager", ovsdbNodeAugmentation));
116 // case 2: connectedManager == 0
118 Whitebox.invokeMethod(ovsdbNodeRemoveCommand, "checkIfOnlyConnectedManager", ovsdbNodeAugmentation));
120 // case 3: onlyConnectedManager.getNumberOfConnections().longValue() > ONE_ACTIVE_CONNECTION_IN_PASSIVE_MODE
121 ManagerEntry onlyConnectedManager = mock(ManagerEntry.class);
123 Whitebox.invokeMethod(ovsdbNodeRemoveCommand, "checkIfOnlyConnectedManager", ovsdbNodeAugmentation));
125 // case 4: when all the above don't apply
126 listManagerEntry.remove(manager1);
128 Whitebox.invokeMethod(ovsdbNodeRemoveCommand, "checkIfOnlyConnectedManager", ovsdbNodeAugmentation));