openflow-protocol-impl/src/test/java/org/opendaylight/openflowjava/protocol/impl/serialization/action/OF13SetFieldActionSerializerTest.java

   1 /*
   2  * Copyright (c) 2014 Pantheon Technologies s.r.o. 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.openflowjava.protocol.impl.serialization.action;
   9 import io.netty.buffer.ByteBuf;
  10 import io.netty.buffer.UnpooledByteBufAllocator;
  11
  12 import java.util.ArrayList;
  13 import java.util.List;
  14
  15 import org.junit.Assert;
  16 import org.junit.Before;
  17 import org.junit.Test;
  18 import org.mockito.Mock;
  19 import org.mockito.Mockito;
  20 import org.mockito.MockitoAnnotations;
  21 import org.opendaylight.openflowjava.protocol.api.extensibility.OFSerializer;
  22 import org.opendaylight.openflowjava.protocol.api.extensibility.SerializerRegistry;
  23 import org.opendaylight.openflowjava.protocol.api.keys.MatchEntrySerializerKey;
  24 import org.opendaylight.openflowjava.protocol.api.util.EncodeConstants;
  25 import org.opendaylight.openflowjava.protocol.impl.serialization.SerializerRegistryImpl;
  26 import org.opendaylight.openflowjava.protocol.impl.util.ActionConstants;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.augments.rev150225.OxmFieldsAction;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.augments.rev150225.OxmFieldsActionBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.augments.rev150225.oxm.container.match.entry.value.ExperimenterIdCaseBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.augments.rev150225.oxm.container.match.entry.value.experimenter.id._case.ExperimenterBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.action.rev130731.SetField;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.action.rev130731.actions.grouping.ActionBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.ExperimenterId;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.ExperimenterClass;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.MatchField;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.match.entries.grouping.MatchEntry;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.match.entries.grouping.MatchEntryBuilder;
  38
  39 /**
  40  * @author madamjak
  41  *
  42  */
  43 public class OF13SetFieldActionSerializerTest {
  44
  45     private SerializerRegistry registry;
  46     @Mock OFSerializer<MatchEntry> serializerMock;
  47
  48     /**
  49      * Initialize registry and mock
  50      */
  51     @Before
  52     public void startUp() {
  53         MockitoAnnotations.initMocks(this);
  54         registry = new SerializerRegistryImpl();
  55         registry.init();
  56     }
  57
  58     /**
  59      * Test identify ExperimenterClass serializer
  60      */
  61     @Test
  62     public void test(){
  63         OF13SetFieldActionSerializer ser = new OF13SetFieldActionSerializer();
  64         ser.injectSerializerRegistry(registry);
  65         ActionBuilder actionBuilder = new ActionBuilder();
  66         long experimenterId = 12L;
  67         ExperimenterIdCaseBuilder expCaseBuilder = new ExperimenterIdCaseBuilder();
  68         ExperimenterBuilder expBuilder = new ExperimenterBuilder();
  69         expBuilder.setExperimenter(new ExperimenterId(experimenterId));
  70         expCaseBuilder.setExperimenter(expBuilder.build());
  71         MatchEntryBuilder meb = new MatchEntryBuilder();
  72         meb.setOxmClass(ExperimenterClass.class);
  73         meb.setOxmMatchField(OxmMatchFieldClass.class);
  74         meb.setMatchEntryValue(expCaseBuilder.build());
  75         List<MatchEntry> MatchEntry = new ArrayList<>();
  76         MatchEntry me = meb.build();
  77         MatchEntry.add(me);
  78         OxmFieldsActionBuilder oxmActBuilder = new OxmFieldsActionBuilder();
  79         oxmActBuilder.setMatchEntry(MatchEntry);
  80         actionBuilder.addAugmentation(OxmFieldsAction.class, oxmActBuilder.build());
  81         actionBuilder.setType(SetField.class);
  82         MatchEntrySerializerKey<?, ?> key = new MatchEntrySerializerKey<>(
  83                 EncodeConstants.OF13_VERSION_ID, ExperimenterClass.class, OxmMatchFieldClass.class);
  84         key.setExperimenterId(experimenterId);
  85         registry.registerSerializer(key, serializerMock);
  86         ByteBuf out = UnpooledByteBufAllocator.DEFAULT.buffer();
  87         ser.serialize(actionBuilder.build(), out);
  88         Mockito.verify(serializerMock, Mockito.times(1)).serialize((MatchEntry)Mockito.anyObject(), (ByteBuf)Mockito.anyObject());
  89         int lenght = out.readableBytes();
  90         Assert.assertEquals("Wrong - bad field code", ActionConstants.SET_FIELD_CODE, out.readUnsignedShort());
  91         Assert.assertEquals("Wrong - bad lenght", lenght, out.readUnsignedShort());
  92     }
  93
  94     private class OxmMatchFieldClass extends MatchField {
  95         // only for testing purposes
  96     }
  97 }