1 package org.opendaylight.openflowplugin.impl.translator;
3 import com.google.common.collect.Lists;
4 import java.math.BigInteger;
5 import org.junit.Assert;
6 import org.junit.Before;
8 import org.junit.runner.RunWith;
9 import org.mockito.Mock;
10 import org.mockito.Mockito;
11 import org.mockito.runners.MockitoJUnitRunner;
12 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
13 import org.opendaylight.openflowplugin.api.OFConstants;
14 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
15 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
16 import org.opendaylight.openflowplugin.api.openflow.device.DeviceState;
17 import org.opendaylight.openflowplugin.openflow.md.util.OpenflowPortsUtil;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnector;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorKey;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.PacketInReason;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.PortNumber;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.TableId;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.InPort;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.MatchField;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.OpenflowBasicClass;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.match.entries.grouping.MatchEntryBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.match.entry.value.grouping.match.entry.value.InPortCaseBuilder;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.match.entry.value.grouping.match.entry.value.in.port._case.InPortBuilder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.match.grouping.MatchBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.FeaturesReply;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PacketInMessage;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PacketInMessageBuilder;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.packet.service.rev130709.PacketReceived;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.packet.service.rev130709.packet.received.Match;
39 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
42 * Created by tkubas on 4/1/15.
45 @RunWith(MockitoJUnitRunner.class)
46 public class PacketReceivedTranslatorTest {
49 ConnectionContext connectionContext;
51 FeaturesReply featuresReply;
53 DeviceState deviceState;
55 DataBroker dataBroker;
57 DeviceContext deviceContext;
59 final String data = "Test_Data";
61 public PacketReceivedTranslatorTest() {
62 OpenflowPortsUtil.init();
66 public void setUp() throws Exception {
67 Mockito.when(deviceContext.getPrimaryConnectionContext()).thenReturn(connectionContext);
68 Mockito.when(connectionContext.getFeatures()).thenReturn(featuresReply);
69 Mockito.when(featuresReply.getDatapathId()).thenReturn(BigInteger.TEN);
70 Mockito.when(deviceContext.getDeviceState()).thenReturn(deviceState);
74 public void testTranslate() throws Exception {
75 final KeyedInstanceIdentifier<Node, NodeKey> nodePath = KeyedInstanceIdentifier
77 .child(Node.class, new NodeKey(new NodeId("openflow:10")));
78 final PacketReceivedTranslator packetReceivedTranslator = new PacketReceivedTranslator();
79 final PacketInMessage packetInMessage = createPacketInMessage(data.getBytes(), 5L);
80 Mockito.when(deviceState.getNodeInstanceIdentifier()).thenReturn(nodePath);
82 final PacketReceived packetReceived = packetReceivedTranslator.translate(packetInMessage, deviceContext, null);
84 Assert.assertArrayEquals(packetInMessage.getData(), packetReceived.getPayload());
85 Assert.assertEquals("org.opendaylight.yang.gen.v1.urn.opendaylight.packet.service.rev130709.SendToController",
86 packetReceived.getPacketInReason().getName());
87 Assert.assertEquals("openflow:10:5",
88 packetReceived.getIngress().getValue().firstKeyOf(NodeConnector.class, NodeConnectorKey.class)
90 Assert.assertEquals(0L, packetReceived.getFlowCookie().getValue().longValue());
91 Assert.assertEquals(42L, packetReceived.getTableId().getValue().longValue());
94 private static PacketInMessage createPacketInMessage(final byte[] data,
96 final PacketInReason reason = PacketInReason.OFPRACTION;
98 MatchEntryBuilder matchEntryBuilder = assembleMatchEntryBld(port);
99 MatchBuilder packetInMatchBld = new MatchBuilder()
100 .setMatchEntry(Lists.newArrayList(matchEntryBuilder.build()));
102 return new PacketInMessageBuilder()
103 .setVersion(OFConstants.OFP_VERSION_1_0)
104 .setData(data).setReason(reason)
105 .setMatch(packetInMatchBld.build())
106 .setVersion(OFConstants.OFP_VERSION_1_3)
107 .setCookie(BigInteger.ZERO)
108 .setTableId(new TableId(42L))
114 public void testGetPacketInMatch() throws Exception {
115 final long portNumValue = 11L;
117 MatchEntryBuilder matchEntryBuilder = assembleMatchEntryBld(portNumValue);
118 MatchBuilder packetInMatchBld = new MatchBuilder()
119 .setMatchEntry(Lists.newArrayList(matchEntryBuilder.build()));
120 PacketInMessageBuilder inputBld = new PacketInMessageBuilder()
121 .setMatch(packetInMatchBld.build())
122 .setVersion(OFConstants.OFP_VERSION_1_3);
123 BigInteger dpid = BigInteger.TEN;
125 final Match packetInMatch = PacketReceivedTranslator.getPacketInMatch(inputBld.build(), dpid);
127 Assert.assertNotNull(packetInMatch.getInPort());
128 Assert.assertEquals("openflow:10:11", packetInMatch.getInPort().getValue());
131 private static MatchEntryBuilder assembleMatchEntryBld(long portNumValue) {
132 MatchEntryBuilder matchEntryBuilder = prepareHeader(InPort.class, false);
133 InPortBuilder inPortBld = new InPortBuilder().setPortNumber(new PortNumber(portNumValue));
134 InPortCaseBuilder caseBuilder = new InPortCaseBuilder();
135 caseBuilder.setInPort(inPortBld.build());
136 matchEntryBuilder.setMatchEntryValue(caseBuilder.build());
137 return matchEntryBuilder;
140 private static MatchEntryBuilder prepareHeader(Class<? extends MatchField> oxmMatchField, boolean hasMask) {
141 MatchEntryBuilder builder = new MatchEntryBuilder();
142 builder.setOxmClass(OpenflowBasicClass.class);
143 builder.setOxmMatchField(oxmMatchField);
144 builder.setHasMask(hasMask);
150 public void testGetPortNumberFromMatch() throws Exception {
151 final Long portNumber = PacketReceivedTranslator.getPortNumberFromMatch(Lists.newArrayList(assembleMatchEntryBld(11L).build()));
152 Assert.assertEquals(11L, portNumber.longValue());