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.ovsdb.transact;
11 import static org.mockito.Matchers.any;
12 import static org.mockito.Matchers.eq;
13 import static org.mockito.Mockito.doNothing;
14 import static org.mockito.Mockito.mock;
15 import static org.mockito.Mockito.verify;
16 import static org.mockito.Mockito.when;
18 import java.lang.reflect.Field;
19 import java.util.ArrayList;
20 import java.util.HashSet;
21 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.opendaylight.ovsdb.lib.notation.Column;
29 import org.opendaylight.ovsdb.lib.notation.Mutator;
30 import org.opendaylight.ovsdb.lib.notation.UUID;
31 import org.opendaylight.ovsdb.lib.operations.Insert;
32 import org.opendaylight.ovsdb.lib.operations.Mutate;
33 import org.opendaylight.ovsdb.lib.operations.Operation;
34 import org.opendaylight.ovsdb.lib.operations.Operations;
35 import org.opendaylight.ovsdb.lib.operations.TransactionBuilder;
36 import org.opendaylight.ovsdb.lib.schema.ColumnSchema;
37 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
38 import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
39 import org.opendaylight.ovsdb.lib.schema.typed.TyperUtils;
40 import org.opendaylight.ovsdb.schema.openvswitch.Bridge;
41 import org.opendaylight.ovsdb.schema.openvswitch.OpenVSwitch;
42 import org.powermock.api.mockito.PowerMockito;
43 import org.powermock.core.classloader.annotations.PrepareForTest;
44 import org.powermock.modules.junit4.PowerMockRunner;
46 @RunWith(PowerMockRunner.class)
47 @PrepareForTest({TyperUtils.class, TransactUtils.class})
48 public class OpenVSwitchBridgeAddCommandTest {
49 private OpenVSwitchBridgeAddCommand openVSwitchBridgeAddCommand;
53 openVSwitchBridgeAddCommand = mock(OpenVSwitchBridgeAddCommand.class, Mockito.CALLS_REAL_METHODS);
56 @SuppressWarnings({ "unchecked", "rawtypes" })
58 public void testExecute() throws Exception {
59 TransactionBuilder transaction = mock(TransactionBuilder.class);
60 List<Operation> operations = new ArrayList<>();
61 when(transaction.getOperations()).thenReturn(operations);
63 Bridge bridge = mock(Bridge.class);
64 when(transaction.getDatabaseSchema()).thenReturn(mock(DatabaseSchema.class));
65 PowerMockito.mockStatic(TyperUtils.class);
66 PowerMockito.when(TyperUtils.getTypedRowWrapper(any(DatabaseSchema.class), eq(Bridge.class))).thenReturn(bridge);
68 List<Insert> inserts = new ArrayList<>();
69 Insert insert = mock(Insert.class);
71 PowerMockito.mockStatic(TransactUtils.class);
72 when(bridge.getSchema()).thenReturn(mock(GenericTableSchema.class));
73 PowerMockito.when(TransactUtils.extractInsert(any(TransactionBuilder.class), any(GenericTableSchema.class))).thenReturn(inserts);
75 OpenVSwitch ovs = mock(OpenVSwitch.class);
76 PowerMockito.when(TyperUtils.getTypedRowWrapper(any(DatabaseSchema.class), eq(OpenVSwitch.class))).thenReturn(ovs);
77 PowerMockito.when(TransactUtils.extractNamedUuid(any(Insert.class))).thenReturn(mock(UUID.class));
78 doNothing().when(ovs).setBridges(any(Set.class));
80 Mutate<GenericTableSchema> mutate = mock(Mutate.class);
81 Operations op = (Operations) setField("op");
82 when(op.mutate(any(OpenVSwitch.class))).thenReturn(mutate);
83 Column<GenericTableSchema, Set<UUID>> column = mock(Column.class);
84 when(ovs.getBridgesColumn()).thenReturn(column);
85 when(column.getSchema()).thenReturn(mock(ColumnSchema.class));
86 when(column.getData()).thenReturn(new HashSet<UUID>());
87 when(mutate.addMutation(any(ColumnSchema.class), any(Mutator.class), any(Set.class))).thenReturn(mutate);
88 when(transaction.add(any(Operation.class))).thenReturn(transaction);
90 openVSwitchBridgeAddCommand.execute(transaction);
91 verify(transaction).add(any(Operation.class));
94 private Object setField(String fieldName) throws Exception {
95 Field field = Operations.class.getDeclaredField(fieldName);
96 field.setAccessible(true);
97 field.set(field.get(Operations.class), mock(Operations.class));
98 return field.get(Operations.class);