openflowjava/openflow-protocol-impl/src/test/java/org/opendaylight/openflowjava/protocol/impl/serialization/factories/VendorInputMessageFactoryTest.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 io.netty.buffer.PooledByteBufAllocator;
  13 import org.junit.Test;
  14 import org.junit.runner.RunWith;
  15 import org.mockito.ArgumentMatchers;
  16 import org.mockito.Mock;
  17 import org.mockito.Mockito;
  18 import org.mockito.runners.MockitoJUnitRunner;
  19 import org.opendaylight.openflowjava.protocol.api.extensibility.OFSerializer;
  20 import org.opendaylight.openflowjava.protocol.api.extensibility.SerializerRegistry;
  21 import org.opendaylight.openflowjava.protocol.api.keys.ExperimenterIdSerializerKey;
  22 import org.opendaylight.openflowjava.protocol.api.keys.MessageTypeKey;
  23 import org.opendaylight.openflowjava.protocol.api.util.EncodeConstants;
  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.experimenter.core.ExperimenterDataOfChoice;
  28
  29 /**
  30  * Unit tests for VendorInputMessageFactory.
  31  *
  32  * @author michal.polkorab
  33  */
  34 @RunWith(MockitoJUnitRunner.class)
  35 public class VendorInputMessageFactoryTest {
  36
  37     @Mock SerializerRegistry registry;
  38     @Mock OFSerializer<ExperimenterDataOfChoice> foundSerializer;
  39     @Mock ExperimenterDataOfChoice vendorData;
  40     VendorInputMessageFactory serializer;
  41
  42     /**
  43      * Tests {@link VendorInputMessageFactory#serialize(org.opendaylight.yang.gen.v1.urn.opendaylight.openflow
  44      * .protocol.rev130731.ExperimenterOfMessage, ByteBuf)}.
  45      */
  46     @Test
  47     public void test() {
  48         Mockito.when(registry.getSerializer(ArgumentMatchers.<MessageTypeKey<?>>any()))
  49             .thenReturn(serializer);
  50         VendorInputMessageFactory factory = new VendorInputMessageFactory();
  51         factory.injectSerializerRegistry(registry);
  52         final ByteBuf buffer = PooledByteBufAllocator.DEFAULT.buffer();
  53         ExperimenterInputBuilder builder = new ExperimenterInputBuilder();
  54         builder.setVersion((short) EncodeConstants.OF10_VERSION_ID);
  55         builder.setXid(12345L);
  56         builder.setExperimenter(new ExperimenterId(42L));
  57         builder.setExpType(84L);
  58         builder.setExperimenterDataOfChoice(vendorData);
  59         ExperimenterInput experimenterInput = builder.build();
  60
  61         Mockito.when(registry.getSerializer(
  62                 ArgumentMatchers.<ExperimenterIdSerializerKey<ExperimenterDataOfChoice>>any()))
  63                 .thenReturn(foundSerializer);
  64         factory.serialize(experimenterInput, buffer);
  65         Mockito.verify(foundSerializer, Mockito.times(1))
  66             .serialize(experimenterInput.getExperimenterDataOfChoice(), buffer);
  67     }
  68 }