openflow-protocol-impl/src/test/java/org/opendaylight/openflowjava/protocol/impl/serialization/factories/ExperimenterInputMessageFactoryTest.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
   9 package org.opendaylight.openflowjava.protocol.impl.serialization.factories;
  10
  11 import io.netty.buffer.ByteBuf;
  12 import org.junit.Test;
  13 import org.mockito.Matchers;
  14 import org.mockito.Mock;
  15 import org.mockito.Mockito;
  16 import org.mockito.MockitoAnnotations;
  17 import org.opendaylight.openflowjava.protocol.api.extensibility.OFSerializer;
  18 import org.opendaylight.openflowjava.protocol.api.extensibility.SerializerRegistry;
  19 import org.opendaylight.openflowjava.protocol.api.extensibility.SerializerRegistryInjector;
  20 import org.opendaylight.openflowjava.protocol.api.keys.ExperimenterIdSerializerKey;
  21 import org.opendaylight.openflowjava.protocol.api.util.EncodeConstants;
  22 import org.opendaylight.openflowjava.protocol.impl.serialization.SerializerRegistryImpl;
  23 import org.opendaylight.openflowjava.protocol.impl.util.BufferHelper;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.ExperimenterId;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.ExperimenterInput;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.ExperimenterInputBuilder;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.ExperimenterOfMessage;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.experimenter.core.ExperimenterDataOfChoice;
  29
  30 /**
  31  * @author michal.polkorab
  32  *
  33  */
  34 public class ExperimenterInputMessageFactoryTest {
  35
  36     @Mock SerializerRegistry registry;
  37     @Mock
  38     private OFSerializer<ExperimenterDataOfChoice> serializer;
  39     private OFSerializer<ExperimenterOfMessage> expFactory;
  40     @Mock
  41     private ExperimenterDataOfChoice vendorData;
  42     @Mock
  43     private ByteBuf out;
  44
  45     /**
  46      * Sets up ExperimenterInputMessageFactory
  47      * @param real true if setup should use real registry, false when mock is desired
  48      */
  49     public void startUp(boolean real) {
  50         MockitoAnnotations.initMocks(this);
  51         expFactory = new ExperimenterInputMessageFactory();
  52         if (real) {
  53             SerializerRegistry realRegistry = new SerializerRegistryImpl();
  54             realRegistry.init();
  55             ((SerializerRegistryInjector) expFactory).injectSerializerRegistry(realRegistry);
  56         } else {
  57             ((SerializerRegistryInjector) expFactory).injectSerializerRegistry(registry);
  58         }
  59     }
  60
  61     /**
  62      * Testing of {@link ExperimenterInputMessageFactory} for correct serializer
  63      * lookup and serialization
  64      * @throws Exception
  65      */
  66     @Test(expected=IllegalStateException.class)
  67     public void testV10Real() throws Exception {
  68         startUp(true);
  69         ExperimenterInputBuilder builder = new ExperimenterInputBuilder();
  70         BufferHelper.setupHeader(builder, EncodeConstants.OF10_VERSION_ID);
  71         builder.setExperimenter(new ExperimenterId(42L));
  72         builder.setExpType(21L);
  73         builder.setExperimenterDataOfChoice(vendorData);
  74         ExperimenterInput input = builder.build();
  75
  76         expFactory.serialize(input, out);
  77     }
  78
  79     /**
  80      * Testing of {@link ExperimenterInputMessageFactory} for correct serializer
  81      * lookup and serialization
  82      * @throws Exception
  83      */
  84     @Test(expected=IllegalStateException.class)
  85     public void testV13Real() throws Exception {
  86         startUp(true);
  87         ExperimenterInputBuilder builder = new ExperimenterInputBuilder();
  88         BufferHelper.setupHeader(builder, EncodeConstants.OF13_VERSION_ID);
  89         builder.setExperimenter(new ExperimenterId(42L));
  90         builder.setExpType(22L);
  91         builder.setExperimenterDataOfChoice(vendorData);
  92         ExperimenterInput input = builder.build();
  93
  94         expFactory.serialize(input, out);
  95     }
  96
  97     /**
  98      * Testing of {@link ExperimenterInputMessageFactory} for correct serializer
  99      * lookup and serialization
 100      * @throws Exception
 101      */
 102     @Test
 103     public void testV10() throws Exception {
 104         startUp(false);
 105         ExperimenterInputBuilder builder = new ExperimenterInputBuilder();
 106         BufferHelper.setupHeader(builder, EncodeConstants.OF10_VERSION_ID);
 107         builder.setExperimenter(new ExperimenterId(42L));
 108         builder.setExpType(21L);
 109         builder.setExperimenterDataOfChoice(vendorData);
 110         ExperimenterInput input = builder.build();
 111
 112         Mockito.when(registry.getSerializer(
 113                 (ExperimenterIdSerializerKey<?>) Matchers.any())).thenReturn(serializer);
 114
 115         expFactory.serialize(input, out);
 116         Mockito.verify(serializer, Mockito.times(1)).serialize(input.getExperimenterDataOfChoice(), out);
 117     }
 118
 119     /**
 120      * Testing of {@link ExperimenterInputMessageFactory} for correct serializer
 121      * lookup and serialization
 122      * @throws Exception
 123      */
 124     @Test
 125     public void testV13() throws Exception {
 126         startUp(false);
 127         ExperimenterInputBuilder builder = new ExperimenterInputBuilder();
 128         BufferHelper.setupHeader(builder, EncodeConstants.OF13_VERSION_ID);
 129         builder.setExperimenter(new ExperimenterId(42L));
 130         builder.setExpType(21L);
 131         builder.setExperimenterDataOfChoice(vendorData);
 132         ExperimenterInput input = builder.build();
 133
 134         Mockito.when(registry.getSerializer(
 135                 (ExperimenterIdSerializerKey<?>) Matchers.any())).thenReturn(serializer);
 136
 137         expFactory.serialize(input, out);
 138         Mockito.verify(serializer, Mockito.times(1)).serialize(input.getExperimenterDataOfChoice(), out);
 139     }
 140 }