1 package org.opendaylight.vpnservice.interfacemgr.test;
3 import com.google.common.base.Optional;
4 import com.google.common.util.concurrent.Futures;
6 import org.junit.Before;
8 import org.junit.runner.RunWith;
9 import org.mockito.Mock;
10 import org.mockito.runners.MockitoJUnitRunner;
11 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
12 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
13 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
14 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
15 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker.DataChangeScope;
16 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
17 import org.opendaylight.vpnservice.interfacemgr.IfmUtil;
18 import org.opendaylight.vpnservice.interfacemgr.servicebindings.flowbased.confighelpers.FlowBasedServicesConfigBindHelper;
19 import org.opendaylight.vpnservice.interfacemgr.servicebindings.flowbased.confighelpers.FlowBasedServicesConfigUnbindHelper;
20 import org.opendaylight.vpnservice.mdsalutil.*;
21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.iana._if.type.rev140508.L2vlan;
22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.Interface;
23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface.OperStatus;
24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.InterfaceBuilder;
25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.PhysAddress;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowId;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.table.Flow;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.table.FlowKey;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.instruction.list.Instruction;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.instruction.list.InstructionKey;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorId;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.servicebinding.rev151015.ServiceBindings;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.servicebinding.rev151015.StypeOpenflow;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.servicebinding.rev151015.service.bindings.ServicesInfo;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.servicebinding.rev151015.service.bindings.ServicesInfoKey;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.servicebinding.rev151015.service.bindings.services.info.BoundServices;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.servicebinding.rev151015.service.bindings.services.info.BoundServicesKey;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.rev150331.TunnelTypeGre;
39 import org.opendaylight.yangtools.concepts.ListenerRegistration;
40 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
42 import java.math.BigInteger;
43 import java.util.ArrayList;
44 import java.util.List;
46 import static org.mockito.Matchers.any;
47 import static org.mockito.Mockito.*;
49 @RunWith(MockitoJUnitRunner.class)
50 public class FlowBasedServicesConfigurationTest {
52 BigInteger dpId = BigInteger.valueOf(1);
54 int instructionKeyval = 2;
56 Interface interfaceEnabled = null;
57 String serviceName = "VPN";
58 InstanceIdentifier<BoundServices> boundServicesIid = null;
59 org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface stateInterface;
60 InstanceIdentifier<org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface> interfaceStateIdentifier = null;
61 BoundServices boundServiceNew = null;
62 NodeConnectorId nodeConnectorId = null;
63 ServicesInfo servicesInfo = null;
64 ServicesInfo servicesInfoUnbind = null;
65 StypeOpenflow stypeOpenflow = null;
66 InstanceIdentifier<Interface> interfaceInstanceIdentifier = null;
67 InstanceIdentifier<Flow> flowInstanceId = null;
68 Flow ingressFlow = null;
69 Instruction instruction = null;
70 InstructionKey instructionKey = null;
71 List<Instruction>instructions = new ArrayList<>();
75 @Mock DataBroker dataBroker;
76 @Mock ListenerRegistration<DataChangeListener> dataChangeListenerRegistration;
77 @Mock ReadOnlyTransaction mockReadTx;
78 @Mock WriteTransaction mockWriteTx;
80 FlowBasedServicesConfigBindHelper bindHelper;
81 FlowBasedServicesConfigUnbindHelper unbindHelper;
84 public void setUp() throws Exception {
85 when(dataBroker.registerDataChangeListener(
86 any(LogicalDatastoreType.class),
87 any(InstanceIdentifier.class),
88 any(DataChangeListener.class),
89 any(DataChangeScope.class)))
90 .thenReturn(dataChangeListenerRegistration);
94 private void setupMocks(){
96 interfaceEnabled = InterfaceManagerTestUtil.buildInterface(InterfaceManagerTestUtil.interfaceName, "Test Vlan Interface1",true,L2vlan.class,dpId);
97 nodeConnectorId = InterfaceManagerTestUtil.buildNodeConnectorId(dpId, portNum);
98 interfaceInstanceIdentifier = IfmUtil.buildId(InterfaceManagerTestUtil.interfaceName);
100 InterfaceBuilder ifaceBuilder = new InterfaceBuilder();
101 List<String> lowerLayerIfList = new ArrayList<>();
102 lowerLayerIfList.add(nodeConnectorId.getValue());
103 ifaceBuilder.setOperStatus(OperStatus.Up).setAdminStatus(org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface.AdminStatus.Up)
104 .setPhysAddress(PhysAddress.getDefaultInstance("AA:AA:AA:AA:AA:AA"))
105 .setIfIndex(ifIndexval)
106 .setLowerLayerIf(lowerLayerIfList)
107 .setKey(IfmUtil.getStateInterfaceKeyFromName(InterfaceManagerTestUtil.interfaceName))
108 .setName(InterfaceManagerTestUtil.interfaceName);
110 stypeOpenflow = InterfaceManagerTestUtil.buildStypeOpenflow(dpId, flowPriority,NwConstants.LPORT_DISPATCHER_TABLE,instructions);
111 boundServiceNew = InterfaceManagerTestUtil.buildBoundServices(serviceName, key, new BoundServicesKey(key), stypeOpenflow);
112 instructionKey = new InstructionKey(instructionKeyval);
113 BigInteger[] metadataValues = IfmUtil.mergeOpenflowMetadataWriteInstructions(instructions);
114 short sIndex = boundServiceNew.getServicePriority();
115 BigInteger metadata = MetaDataUtil.getMetaDataForLPortDispatcher(ifaceBuilder.getIfIndex(),
116 ++sIndex, metadataValues[0]);
117 BigInteger mask = MetaDataUtil.getMetaDataMaskForLPortDispatcher(
118 MetaDataUtil.METADATA_MASK_SERVICE_INDEX,
119 MetaDataUtil.METADATA_MASK_LPORT_TAG, metadataValues[1]);
121 instruction = InterfaceManagerTestUtil.buildInstruction(InterfaceManagerTestUtil.buildWriteMetaDataCase(InterfaceManagerTestUtil.buildWriteMetaData(metadata, mask)),
122 new InstructionKey(instructionKey));
123 instructions.add(instruction);
124 ServicesInfoKey servicesInfoKey = new ServicesInfoKey(InterfaceManagerTestUtil.interfaceName);
125 boundServicesIid = InstanceIdentifier.builder(ServiceBindings.class).child(ServicesInfo.class, servicesInfoKey).
126 child(BoundServices.class, new BoundServicesKey(key)).build();
128 interfaceStateIdentifier = IfmUtil.buildStateInterfaceId(interfaceEnabled.getName());
129 stateInterface = ifaceBuilder.build();
130 List<BoundServices> boundServiceslist = new ArrayList<>();
131 boundServiceslist.add(boundServiceNew);
132 servicesInfo = InterfaceManagerTestUtil.buildServicesInfo(InterfaceManagerTestUtil.interfaceName, servicesInfoKey, boundServiceslist);
133 servicesInfoUnbind = InterfaceManagerTestUtil.buildServicesInfo(InterfaceManagerTestUtil.interfaceName,servicesInfoKey,new ArrayList<>());
135 String flowRef = InterfaceManagerTestUtil.buildflowRef(dpId,InterfaceManagerTestUtil.interfaceName,boundServiceNew.getServiceName(),boundServiceNew.getServicePriority());
136 List<Instruction> instructionList = boundServiceNew.getAugmentation(StypeOpenflow.class).getInstruction();
137 String serviceRef = boundServiceNew.getServiceName();
138 List<MatchInfo> matches = new ArrayList<>();
139 matches.add(new MatchInfo(MatchFieldType.metadata, new BigInteger[] {
140 MetaDataUtil.getMetaDataForLPortDispatcher(ifaceBuilder.getIfIndex(), boundServiceNew.getServicePriority()),
141 MetaDataUtil.getMetaDataMaskForLPortDispatcher() }));
142 ingressFlow = MDSALUtil.buildFlowNew(NwConstants.LPORT_DISPATCHER_TABLE, flowRef, boundServiceNew.getServicePriority(), serviceRef, 0, 0,
143 stypeOpenflow.getFlowCookie(), matches, instructionList);
144 FlowKey flowKey = new FlowKey(new FlowId(ingressFlow.getId()));
145 flowInstanceId = InterfaceManagerTestUtil.getFlowInstanceIdentifier(dpId,ingressFlow.getTableId(),flowKey);
147 when(dataBroker.newReadOnlyTransaction()).thenReturn(mockReadTx);
148 when(dataBroker.newWriteOnlyTransaction()).thenReturn(mockWriteTx);
152 public void testConfigBindSingleService(){
154 Optional<org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface> expectedStateInterface = Optional.of(stateInterface);
155 Optional<Interface> expectedInterface = Optional.of(interfaceEnabled);
156 Optional<ServicesInfo>expectedservicesInfo = Optional.of(servicesInfo);
158 ServicesInfoKey servicesInfoKey = new ServicesInfoKey(InterfaceManagerTestUtil.interfaceName);
159 InstanceIdentifier.InstanceIdentifierBuilder<ServicesInfo> servicesInfoIdentifierBuilder =
160 InstanceIdentifier.builder(ServiceBindings.class).child(ServicesInfo.class, servicesInfoKey);
163 doReturn(Futures.immediateCheckedFuture(expectedStateInterface)).when(mockReadTx).read(
164 LogicalDatastoreType.OPERATIONAL,interfaceStateIdentifier);
165 doReturn(Futures.immediateCheckedFuture(expectedInterface)).when(mockReadTx).read(
166 LogicalDatastoreType.CONFIGURATION, interfaceInstanceIdentifier);
167 doReturn(Futures.immediateCheckedFuture(expectedservicesInfo)).when(mockReadTx).read(
168 LogicalDatastoreType.CONFIGURATION, servicesInfoIdentifierBuilder.build());
170 bindHelper.bindService(boundServicesIid,boundServiceNew,dataBroker);
172 verify(mockWriteTx).put(LogicalDatastoreType.CONFIGURATION,flowInstanceId,ingressFlow, true);
176 public void testConfigUnbindSingleService(){
178 Optional<org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface> expectedStateInterface = Optional.of(stateInterface);
179 Optional<Interface> expectedInterface = Optional.of(interfaceEnabled);
180 Optional<ServicesInfo>expectedservicesInfo = Optional.of(servicesInfoUnbind);
182 ServicesInfoKey servicesInfoKey = new ServicesInfoKey(InterfaceManagerTestUtil.interfaceName);
183 InstanceIdentifier.InstanceIdentifierBuilder<ServicesInfo> servicesInfoIdentifierBuilder =
184 InstanceIdentifier.builder(ServiceBindings.class).child(ServicesInfo.class, servicesInfoKey);
186 doReturn(Futures.immediateCheckedFuture(expectedStateInterface)).when(mockReadTx).read(
187 LogicalDatastoreType.OPERATIONAL,interfaceStateIdentifier);
188 doReturn(Futures.immediateCheckedFuture(expectedInterface)).when(mockReadTx).read(
189 LogicalDatastoreType.CONFIGURATION, interfaceInstanceIdentifier);
190 doReturn(Futures.immediateCheckedFuture(expectedservicesInfo)).when(mockReadTx).read(
191 LogicalDatastoreType.CONFIGURATION,servicesInfoIdentifierBuilder.build());
193 unbindHelper.unbindService(boundServicesIid,boundServiceNew,dataBroker);
195 verify(mockWriteTx).delete(LogicalDatastoreType.CONFIGURATION,flowInstanceId);