Make TransactionBuilder type-aware
[ovsdb.git] / southbound / southbound-impl / src / test / java / org / opendaylight / ovsdb / southbound / ovsdb / transact / TerminationPointUpdateCommandTest.java
1 /*
2  * Copyright (c) 2015 Inocybe Technologies and others.  All rights reserved.
3  *
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
7  */
8 package org.opendaylight.ovsdb.southbound.ovsdb.transact;
9
10 import static org.mockito.ArgumentMatchers.any;
11 import static org.mockito.ArgumentMatchers.anyString;
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;
18
19 import com.google.common.base.Optional;
20 import java.util.HashMap;
21 import java.util.Map;
22 import org.junit.Before;
23 import org.junit.Test;
24 import org.junit.runner.RunWith;
25 import org.mockito.Mockito;
26 import org.opendaylight.ovsdb.lib.notation.Column;
27 import org.opendaylight.ovsdb.lib.notation.Condition;
28 import org.opendaylight.ovsdb.lib.operations.Operation;
29 import org.opendaylight.ovsdb.lib.operations.Operations;
30 import org.opendaylight.ovsdb.lib.operations.TransactionBuilder;
31 import org.opendaylight.ovsdb.lib.operations.Update;
32 import org.opendaylight.ovsdb.lib.operations.Where;
33 import org.opendaylight.ovsdb.lib.schema.ColumnSchema;
34 import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
35 import org.opendaylight.ovsdb.schema.openvswitch.Interface;
36 import org.opendaylight.ovsdb.schema.openvswitch.Port;
37 import org.opendaylight.ovsdb.southbound.InstanceIdentifierCodec;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeAugmentation;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbPortInterfaceAttributes.VlanMode;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbTerminationPointAugmentation;
41 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
42 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
43 import org.powermock.api.mockito.PowerMockito;
44 import org.powermock.api.support.membermodification.MemberMatcher;
45 import org.powermock.api.support.membermodification.MemberModifier;
46 import org.powermock.core.classloader.annotations.PrepareForTest;
47 import org.powermock.modules.junit4.PowerMockRunner;
48
49 @RunWith(PowerMockRunner.class)
50 @PrepareForTest({ TerminationPointUpdateCommand.class, TransactUtils.class, VlanMode.class,
51         TerminationPointCreateCommand.class, InstanceIdentifier.class, Operations.class })
52 public class TerminationPointUpdateCommandTest {
53
54     private static final String TERMINATION_POINT_NAME = "termination point name";
55     private TerminationPointUpdateCommand terminationPointUpdateCommand;
56
57     @Before
58     public void setUp() {
59         terminationPointUpdateCommand = mock(TerminationPointUpdateCommand.class, Mockito.CALLS_REAL_METHODS);
60     }
61
62     @SuppressWarnings("unchecked")
63     @Test
64     public void testExecute() {
65         Map<InstanceIdentifier<OvsdbTerminationPointAugmentation>, OvsdbTerminationPointAugmentation> created
66             = new HashMap<>();
67         created.put(mock(InstanceIdentifier.class), mock(OvsdbTerminationPointAugmentation.class));
68         PowerMockito.mockStatic(TransactUtils.class);
69         PowerMockito.when(TransactUtils.extractCreated(any(DataChangeEvent.class),
70                 eq(OvsdbTerminationPointAugmentation.class))).thenReturn(created);
71         MemberModifier.suppress(MemberMatcher.method(TerminationPointUpdateCommand.class, "updateTerminationPoint",
72                 TransactionBuilder.class, BridgeOperationalState.class,
73                 InstanceIdentifier.class, OvsdbTerminationPointAugmentation.class, InstanceIdentifierCodec.class));
74         doNothing().when(terminationPointUpdateCommand).updateTerminationPoint(any(TransactionBuilder.class),
75                 any(BridgeOperationalState.class), any(InstanceIdentifier.class),
76                 any(OvsdbTerminationPointAugmentation.class), any(InstanceIdentifierCodec.class));
77
78         Map<InstanceIdentifier<OvsdbTerminationPointAugmentation>, OvsdbTerminationPointAugmentation> updated
79             = new HashMap<>();
80         updated.put(mock(InstanceIdentifier.class), mock(OvsdbTerminationPointAugmentation.class));
81         PowerMockito.when(TransactUtils.extractUpdated(any(DataChangeEvent.class),
82                 eq(OvsdbTerminationPointAugmentation.class))).thenReturn(updated);
83
84         TransactionBuilder transactionBuilder = mock(TransactionBuilder.class);
85         terminationPointUpdateCommand.execute(transactionBuilder, mock(BridgeOperationalState.class),
86                 mock(DataChangeEvent.class), mock(InstanceIdentifierCodec.class));
87         // TODO Verify something useful
88     }
89
90     @SuppressWarnings({ "unchecked", "rawtypes" })
91     @Test
92     public void testUpdateTerminationPoint() throws Exception {
93         TransactionBuilder transaction = mock(TransactionBuilder.class);
94         BridgeOperationalState state = mock(BridgeOperationalState.class);
95         OvsdbTerminationPointAugmentation terminationPoint = mock(OvsdbTerminationPointAugmentation.class);
96         when(terminationPoint.getName()).thenReturn(TERMINATION_POINT_NAME);
97         Node node = mock(Node.class);
98         when(node.augmentation(OvsdbBridgeAugmentation.class)).thenReturn(mock(OvsdbBridgeAugmentation.class));
99         Optional<Node> optNode = Optional.of(node);
100         when(state.getBridgeNode(any(InstanceIdentifier.class))).thenReturn(optNode);
101
102         // Test updateInterface()
103         Interface ovsInterface = mock(Interface.class);
104         when(transaction.getTypedRowWrapper(eq(Interface.class))).thenReturn(ovsInterface);
105
106         Interface extraInterface = mock(Interface.class);
107         when(transaction.getTypedRowWrapper(eq(Interface.class))).thenReturn(extraInterface);
108         doNothing().when(extraInterface).setName(anyString());
109
110         Operations op = OvsdbNodeUpdateCommandTest.setOpField();
111         Update update = mock(Update.class);
112         when(op.update(any(Interface.class))).thenReturn(update);
113
114         Column<GenericTableSchema, String> column = mock(Column.class);
115         when(extraInterface.getNameColumn()).thenReturn(column);
116         ColumnSchema<GenericTableSchema, String> columnSchema = mock(ColumnSchema.class);
117         when(column.getSchema()).thenReturn(columnSchema);
118         when(columnSchema.opEqual(anyString())).thenReturn(mock(Condition.class));
119         Where where = mock(Where.class);
120         when(update.where(any(Condition.class))).thenReturn(where);
121         when(where.build()).thenReturn(mock(Operation.class));
122         when(transaction.add(any(Operation.class))).thenReturn(transaction);
123         PowerMockito.suppress(MemberMatcher.methodsDeclaredIn(TerminationPointCreateCommand.class));
124         PowerMockito.suppress(MemberMatcher.methodsDeclaredIn(InstanceIdentifier.class));
125
126         // Test updatePort()
127         Port port = mock(Port.class);
128         when(transaction.getTypedRowWrapper(eq(Port.class))).thenReturn(port);
129         Port extraPort = mock(Port.class);
130         when(transaction.getTypedRowWrapper(eq(Port.class))).thenReturn(extraPort);
131         doNothing().when(extraPort).setName(anyString());
132         when(op.update(any(Port.class))).thenReturn(update);
133         when(extraPort.getNameColumn()).thenReturn(column);
134
135         InstanceIdentifier<OvsdbTerminationPointAugmentation> iid = mock(InstanceIdentifier.class);
136         terminationPointUpdateCommand.updateTerminationPoint(transaction, state, iid, terminationPoint,
137                 mock(InstanceIdentifierCodec.class));
138         verify(transaction, times(1)).add(any(Operation.class));
139     }
140 }