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
  14 import org.junit.Test;
  15 import org.junit.runner.RunWith;
  16 import org.mockito.Matchers;
  17 import org.mockito.Mock;
  18 import org.mockito.Mockito;
  19 import org.mockito.runners.MockitoJUnitRunner;
  20 import org.opendaylight.openflowjava.protocol.api.extensibility.MessageTypeKey;
  21 import org.opendaylight.openflowjava.protocol.api.extensibility.SerializerRegistry;
  22 import org.opendaylight.openflowjava.protocol.api.util.EncodeConstants;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.ExperimenterId;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.ExperimenterInput;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.ExperimenterInputBuilder;
  26
  27 /**
  28  * @author michal.polkorab
  29  *
  30  */
  31 @RunWith(MockitoJUnitRunner.class)
  32 public class VendorInputMessageFactoryTest {
  33
  34     @Mock SerializerRegistry registry;
  35     @Mock ExperimenterInputMessageFactory serializer;
  36
  37     /**
  38      * Tests {@link VendorInputMessageFactory#serialize(ExperimenterInput, ByteBuf)}
  39      */
  40     @Test
  41     public void test() {
  42         Mockito.when(registry.getSerializer((MessageTypeKey<?>) Matchers.any(MessageTypeKey.class)))
  43             .thenReturn(serializer);
  44         VendorInputMessageFactory factory = new VendorInputMessageFactory();
  45         factory.injectSerializerRegistry(registry);
  46         ByteBuf buffer = PooledByteBufAllocator.DEFAULT.buffer();
  47         ExperimenterInputBuilder builder = new ExperimenterInputBuilder();
  48         builder.setVersion((short) EncodeConstants.OF10_VERSION_ID);
  49         builder.setXid(12345L);
  50         builder.setExperimenter(new ExperimenterId(42L));
  51         builder.setExpType(84L);
  52         ExperimenterInput experimenterInput = builder.build();
  53
  54         factory.serialize(experimenterInput, buffer);
  55         Mockito.verify(serializer, Mockito.times(1)).serialize(experimenterInput, buffer);
  56     }
  57 }