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.GenericTableSchema;
37 import org.opendaylight.ovsdb.schema.openvswitch.OpenVSwitch;
38 import org.opendaylight.ovsdb.southbound.InstanceIdentifierCodec;
39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeAugmentation;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ConnectionInfo;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.OpenvswitchExternalIds;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.OpenvswitchOtherConfigs;
45 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
46 import org.powermock.api.mockito.PowerMockito;
47 import org.powermock.api.support.membermodification.MemberMatcher;
48 import org.powermock.core.classloader.annotations.PrepareForTest;
49 import org.powermock.modules.junit4.PowerMockRunner;
51 @RunWith(PowerMockRunner.class)
52 @PrepareForTest({TransactUtils.class, OvsdbNodeUpdateCommand.class, InstanceIdentifier.class, Operations.class})
53 public class OvsdbNodeUpdateCommandTest {
55 private static final String EXTERNAL_ID_KEY = "external id key";
56 private static final String EXTERNAL_ID_VALUE = "external id value";
57 private static final String OTHER_CONFIG_KEY = "other config key";
58 private static final String OTHER_CONFIG_VALUE = "other config value";
60 @Mock private DataChangeEvent changes;
61 private OvsdbNodeUpdateCommand ovsdbNodeUpdateCommand;
65 ovsdbNodeUpdateCommand = mock(OvsdbNodeUpdateCommand.class, Mockito.CALLS_REAL_METHODS);
68 @SuppressWarnings("unchecked")
70 public void testExecute() throws Exception {
71 Map<InstanceIdentifier<OvsdbNodeAugmentation>, OvsdbNodeAugmentation> updated = new HashMap<>();
72 InstanceIdentifier<OvsdbNodeAugmentation> iid = mock(InstanceIdentifier.class);
73 OvsdbNodeAugmentation ovsdbNode = mock(OvsdbNodeAugmentation.class);
74 updated.put(iid, ovsdbNode);
75 PowerMockito.mockStatic(TransactUtils.class);
77 TransactUtils.extractCreatedOrUpdated(any(DataChangeEvent.class), eq(OvsdbNodeAugmentation.class)))
80 ConnectionInfo connectionInfo = mock(ConnectionInfo.class);
81 when(ovsdbNode.getConnectionInfo()).thenReturn(connectionInfo);
82 when(connectionInfo.getRemoteIp()).thenReturn(mock(IpAddress.class));
83 when(connectionInfo.getRemotePort()).thenReturn(mock(PortNumber.class));
85 OpenVSwitch ovs = mock(OpenVSwitch.class);
86 TransactionBuilder transaction = mock(TransactionBuilder.class);
87 when(transaction.getTypedRowWrapper(eq(OpenVSwitch.class))).thenReturn(ovs);
89 List<OpenvswitchExternalIds> externalIds = new ArrayList<>();
90 OpenvswitchExternalIds externalId = mock(OpenvswitchExternalIds.class);
91 externalIds.add(externalId);
92 when(externalId.getExternalIdKey()).thenReturn(EXTERNAL_ID_KEY);
93 when(externalId.getExternalIdValue()).thenReturn(EXTERNAL_ID_VALUE);
94 when(ovsdbNode.getOpenvswitchExternalIds()).thenReturn(externalIds);
95 PowerMockito.suppress(MemberMatcher.method(OvsdbNodeUpdateCommand.class, "stampInstanceIdentifier",
96 TransactionBuilder.class, InstanceIdentifier.class, InstanceIdentifierCodec.class));
97 PowerMockito.suppress(MemberMatcher.methodsDeclaredIn(InstanceIdentifier.class));
98 doNothing().when(ovs).setExternalIds(any(ImmutableMap.class));
100 Mutate<GenericTableSchema> mutate = mock(Mutate.class);
101 Operations op = setOpField();
102 Column<GenericTableSchema, Map<String, String>> column = mock(Column.class);
103 when(ovs.getExternalIdsColumn()).thenReturn(column);
104 when(column.getSchema()).thenReturn(mock(ColumnSchema.class));
105 when(column.getData()).thenReturn(new HashMap<>());
106 when(op.mutate(any(OpenVSwitch.class))).thenReturn(mutate);
107 when(transaction.add(any(Operation.class))).thenReturn(transaction);
109 List<OpenvswitchOtherConfigs> otherConfigs = new ArrayList<>();
110 OpenvswitchOtherConfigs otherConfig = mock(OpenvswitchOtherConfigs.class);
111 otherConfigs.add(otherConfig);
112 when(ovsdbNode.getOpenvswitchOtherConfigs()).thenReturn(otherConfigs);
113 when(otherConfig.getOtherConfigKey()).thenReturn(OTHER_CONFIG_KEY);
114 when(otherConfig.getOtherConfigValue()).thenReturn(OTHER_CONFIG_VALUE);
115 doNothing().when(ovs).setOtherConfig(any(ImmutableMap.class));
116 when(ovs.getOtherConfigColumn()).thenReturn(column);
118 ovsdbNodeUpdateCommand.execute(transaction, mock(BridgeOperationalState.class), changes,
119 mock(InstanceIdentifierCodec.class));
120 verify(externalId).getExternalIdKey();
121 verify(otherConfig).getOtherConfigKey();
122 verify(ovs, times(2)).getExternalIdsColumn();
123 verify(transaction, times(2)).add(eq(null));
126 static Operations setOpField() throws Exception {
127 Field opField = PowerMockito.field(Operations.class, "op");
128 Operations mockOp = mock(Operations.class);
129 opField.set(Operations.class, mockOp);