2 * Copyright © 2015, 2017 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.ArgumentMatchers.any;
12 import static org.mockito.ArgumentMatchers.eq;
13 import static org.mockito.Mockito.doNothing;
14 import static org.mockito.Mockito.mock;
15 import static org.mockito.Mockito.times;
16 import static org.mockito.Mockito.verify;
17 import static org.mockito.Mockito.when;
19 import com.google.common.collect.ImmutableMap;
20 import java.lang.reflect.Field;
21 import java.util.ArrayList;
22 import java.util.HashMap;
23 import java.util.List;
25 import org.junit.Before;
26 import org.junit.Test;
27 import org.junit.runner.RunWith;
28 import org.mockito.Mock;
29 import org.mockito.Mockito;
30 import org.opendaylight.ovsdb.lib.notation.Column;
31 import org.opendaylight.ovsdb.lib.operations.Mutate;
32 import org.opendaylight.ovsdb.lib.operations.Operation;
33 import org.opendaylight.ovsdb.lib.operations.Operations;
34 import org.opendaylight.ovsdb.lib.operations.TransactionBuilder;
35 import org.opendaylight.ovsdb.lib.schema.ColumnSchema;
36 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
37 import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
38 import org.opendaylight.ovsdb.lib.schema.typed.TyperUtils;
39 import org.opendaylight.ovsdb.schema.openvswitch.OpenVSwitch;
40 import org.opendaylight.ovsdb.southbound.InstanceIdentifierCodec;
41 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
42 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeAugmentation;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ConnectionInfo;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.OpenvswitchExternalIds;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.OpenvswitchOtherConfigs;
47 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
48 import org.powermock.api.mockito.PowerMockito;
49 import org.powermock.api.support.membermodification.MemberMatcher;
50 import org.powermock.core.classloader.annotations.PrepareForTest;
51 import org.powermock.modules.junit4.PowerMockRunner;
53 @RunWith(PowerMockRunner.class)
54 @PrepareForTest({TransactUtils.class, TyperUtils.class, OvsdbNodeUpdateCommand.class, InstanceIdentifier.class,
56 public class OvsdbNodeUpdateCommandTest {
58 private static final String EXTERNAL_ID_KEY = "external id key";
59 private static final String EXTERNAL_ID_VALUE = "external id value";
60 private static final String OTHER_CONFIG_KEY = "other config key";
61 private static final String OTHER_CONFIG_VALUE = "other config value";
63 @Mock private DataChangeEvent changes;
64 private OvsdbNodeUpdateCommand ovsdbNodeUpdateCommand;
68 ovsdbNodeUpdateCommand = mock(OvsdbNodeUpdateCommand.class, Mockito.CALLS_REAL_METHODS);
71 @SuppressWarnings("unchecked")
73 public void testExecute() throws Exception {
74 Map<InstanceIdentifier<OvsdbNodeAugmentation>, OvsdbNodeAugmentation> updated = new HashMap<>();
75 InstanceIdentifier<OvsdbNodeAugmentation> iid = mock(InstanceIdentifier.class);
76 OvsdbNodeAugmentation ovsdbNode = mock(OvsdbNodeAugmentation.class);
77 updated.put(iid, ovsdbNode);
78 PowerMockito.mockStatic(TransactUtils.class);
80 TransactUtils.extractCreatedOrUpdated(any(DataChangeEvent.class), eq(OvsdbNodeAugmentation.class)))
83 ConnectionInfo connectionInfo = mock(ConnectionInfo.class);
84 when(ovsdbNode.getConnectionInfo()).thenReturn(connectionInfo);
85 when(connectionInfo.getRemoteIp()).thenReturn(mock(IpAddress.class));
86 when(connectionInfo.getRemotePort()).thenReturn(mock(PortNumber.class));
88 OpenVSwitch ovs = mock(OpenVSwitch.class);
89 TransactionBuilder transaction = mock(TransactionBuilder.class);
90 when(transaction.getDatabaseSchema()).thenReturn(mock(DatabaseSchema.class));
91 PowerMockito.mockStatic(TyperUtils.class);
92 PowerMockito.when(TyperUtils.getTypedRowWrapper(any(DatabaseSchema.class), eq(OpenVSwitch.class)))
95 List<OpenvswitchExternalIds> externalIds = new ArrayList<>();
96 OpenvswitchExternalIds externalId = mock(OpenvswitchExternalIds.class);
97 externalIds.add(externalId);
98 when(externalId.getExternalIdKey()).thenReturn(EXTERNAL_ID_KEY);
99 when(externalId.getExternalIdValue()).thenReturn(EXTERNAL_ID_VALUE);
100 when(ovsdbNode.getOpenvswitchExternalIds()).thenReturn(externalIds);
101 PowerMockito.suppress(MemberMatcher.method(OvsdbNodeUpdateCommand.class, "stampInstanceIdentifier",
102 TransactionBuilder.class, InstanceIdentifier.class, InstanceIdentifierCodec.class));
103 PowerMockito.suppress(MemberMatcher.methodsDeclaredIn(InstanceIdentifier.class));
104 doNothing().when(ovs).setExternalIds(any(ImmutableMap.class));
106 Mutate<GenericTableSchema> mutate = mock(Mutate.class);
107 Operations op = setOpField();
108 Column<GenericTableSchema, Map<String, String>> column = mock(Column.class);
109 when(ovs.getExternalIdsColumn()).thenReturn(column);
110 when(column.getSchema()).thenReturn(mock(ColumnSchema.class));
111 when(column.getData()).thenReturn(new HashMap<>());
112 when(op.mutate(any(OpenVSwitch.class))).thenReturn(mutate);
113 when(transaction.add(any(Operation.class))).thenReturn(transaction);
115 List<OpenvswitchOtherConfigs> otherConfigs = new ArrayList<>();
116 OpenvswitchOtherConfigs otherConfig = mock(OpenvswitchOtherConfigs.class);
117 otherConfigs.add(otherConfig);
118 when(ovsdbNode.getOpenvswitchOtherConfigs()).thenReturn(otherConfigs);
119 when(otherConfig.getOtherConfigKey()).thenReturn(OTHER_CONFIG_KEY);
120 when(otherConfig.getOtherConfigValue()).thenReturn(OTHER_CONFIG_VALUE);
121 doNothing().when(ovs).setOtherConfig(any(ImmutableMap.class));
122 when(ovs.getOtherConfigColumn()).thenReturn(column);
124 ovsdbNodeUpdateCommand.execute(transaction, mock(BridgeOperationalState.class), changes,
125 mock(InstanceIdentifierCodec.class));
126 verify(externalId).getExternalIdKey();
127 verify(otherConfig).getOtherConfigKey();
128 verify(ovs, times(2)).getExternalIdsColumn();
129 verify(transaction, times(2)).add(eq(null));
132 static Operations setOpField() throws Exception {
133 Field opField = PowerMockito.field(Operations.class, "op");
134 Operations mockOp = mock(Operations.class);
135 opField.set(Operations.class, mockOp);