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.Matchers.eq;
14 import static org.mockito.Mockito.doNothing;
15 import static org.mockito.Mockito.mock;
16 import static org.mockito.Mockito.verify;
17 import static org.mockito.Mockito.when;
19 import java.util.HashMap;
20 import java.util.HashSet;
23 import org.junit.Before;
24 import org.junit.Test;
25 import org.junit.runner.RunWith;
26 import org.mockito.Mockito;
27 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
28 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
29 import org.opendaylight.ovsdb.lib.message.TableUpdates;
30 import org.opendaylight.ovsdb.lib.notation.Column;
31 import org.opendaylight.ovsdb.lib.notation.UUID;
32 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
33 import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
34 import org.opendaylight.ovsdb.lib.schema.typed.TyperUtils;
35 import org.opendaylight.ovsdb.schema.openvswitch.Bridge;
36 import org.opendaylight.ovsdb.schema.openvswitch.Port;
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.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
41 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
42 import org.powermock.api.mockito.PowerMockito;
43 import org.powermock.api.support.membermodification.MemberModifier;
44 import org.powermock.core.classloader.annotations.PrepareForTest;
45 import org.powermock.modules.junit4.PowerMockRunner;
46 import org.powermock.reflect.Whitebox;
48 @RunWith(PowerMockRunner.class)
49 @PrepareForTest({TyperUtils.class, SouthboundMapper.class, InstanceIdentifier.class})
50 public class OvsdbPortRemoveCommandTest {
52 private static final String PORT_NAME = "port0";
53 private OvsdbPortRemoveCommand ovsdbPortRemoveCommand;
56 public void setUp() throws Exception {
57 ovsdbPortRemoveCommand = PowerMockito.mock(OvsdbPortRemoveCommand.class, Mockito.CALLS_REAL_METHODS);
61 public void testOvsdbPortRemoveCommandTest() {
62 OvsdbConnectionInstance key = mock(OvsdbConnectionInstance.class);
63 TableUpdates updates = mock(TableUpdates.class);
64 DatabaseSchema dbSchema = mock(DatabaseSchema.class);
65 OvsdbPortRemoveCommand ovsdbPortRemoveCommand1 =
66 new OvsdbPortRemoveCommand(mock(InstanceIdentifierCodec.class), key, updates, dbSchema);
67 assertEquals(key, Whitebox.getInternalState(ovsdbPortRemoveCommand1, "key"));
68 assertEquals(updates, Whitebox.getInternalState(ovsdbPortRemoveCommand1, "updates"));
69 assertEquals(dbSchema, Whitebox.getInternalState(ovsdbPortRemoveCommand1, "dbSchema"));
73 @SuppressWarnings("unchecked")
74 public void testExecute() throws Exception {
75 when(ovsdbPortRemoveCommand.getUpdates()).thenReturn(mock(TableUpdates.class));
76 when(ovsdbPortRemoveCommand.getDbSchema()).thenReturn(mock(DatabaseSchema.class));
77 PowerMockito.mockStatic(TyperUtils.class);
78 UUID uuid = mock(UUID.class);
79 Map<UUID, Port> portRemovedRows = new HashMap<>();
80 Port port = mock(Port.class);
81 portRemovedRows.put(uuid, port);
82 when(TyperUtils.extractRowsRemoved(eq(Port.class), any(TableUpdates.class), any(DatabaseSchema.class)))
83 .thenReturn(portRemovedRows);
84 Map<UUID, Bridge> bridgeUpdatedRows = new HashMap<>();
85 Bridge updatedBridgeData = mock(Bridge.class);
86 bridgeUpdatedRows.put(uuid, updatedBridgeData);
87 when(TyperUtils.extractRowsUpdated(eq(Bridge.class), any(TableUpdates.class), any(DatabaseSchema.class)))
88 .thenReturn(bridgeUpdatedRows);
89 Map<UUID, Bridge> bridgeUpdatedOldRows = new HashMap<>();
90 Bridge oldBridgeData = mock(Bridge.class);
91 bridgeUpdatedOldRows.put(uuid, oldBridgeData);
92 when(TyperUtils.extractRowsOld(eq(Bridge.class), any(TableUpdates.class), any(DatabaseSchema.class)))
93 .thenReturn(bridgeUpdatedOldRows);
95 Column<GenericTableSchema, Set<UUID>> column = mock(Column.class);
96 when(oldBridgeData.getPortsColumn()).thenReturn(column);
97 Set<UUID> uuids = new HashSet<>();
99 when(column.getData()).thenReturn(uuids);
101 Column<GenericTableSchema, UUID> uuidColumn = mock(Column.class);
102 when(port.getUuidColumn()).thenReturn(uuidColumn);
103 when(uuidColumn.getData()).thenReturn(uuid);
105 when(port.getName()).thenReturn(PORT_NAME);
106 PowerMockito.mockStatic(SouthboundMapper.class);
107 InstanceIdentifier<Node> nodeIID = mock(InstanceIdentifier.class);
108 when(ovsdbPortRemoveCommand.getOvsdbConnectionInstance()).thenReturn(mock(OvsdbConnectionInstance.class));
109 when(SouthboundMapper.createInstanceIdentifier(any(InstanceIdentifierCodec.class),
110 any(OvsdbConnectionInstance.class), any(Bridge.class)))
111 .thenReturn(nodeIID);
112 MemberModifier.suppress(MemberModifier.methodsDeclaredIn(InstanceIdentifier.class));
113 ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
114 doNothing().when(transaction).delete(any(LogicalDatastoreType.class), any(InstanceIdentifier.class));
116 ovsdbPortRemoveCommand.execute(transaction);
117 verify(transaction).delete(any(LogicalDatastoreType.class), any(InstanceIdentifier.class));