openflow-protocol-impl/src/test/java/org/opendaylight/openflowjava/protocol/impl/serialization/factories/VendorInputMessageFactoryTest.java

   1 /*\r
   2  * Copyright (c) 2014 Pantheon Technologies s.r.o. and others. All rights reserved.\r
   3  *\r
   4  * This program and the accompanying materials are made available under the\r
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,\r
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html\r
   7  */\r
   8 \r
   9 package org.opendaylight.openflowjava.protocol.impl.serialization.factories;\r
  10 \r
  11 import io.netty.buffer.ByteBuf;\r
  12 import io.netty.buffer.PooledByteBufAllocator;\r
  13 \r
  14 import org.junit.Test;\r
  15 import org.junit.runner.RunWith;\r
  16 import org.mockito.Matchers;\r
  17 import org.mockito.Mock;\r
  18 import org.mockito.Mockito;\r
  19 import org.mockito.runners.MockitoJUnitRunner;\r
  20 import org.opendaylight.openflowjava.protocol.api.extensibility.MessageTypeKey;\r
  21 import org.opendaylight.openflowjava.protocol.api.extensibility.SerializerRegistry;\r
  22 import org.opendaylight.openflowjava.protocol.api.util.EncodeConstants;\r
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.ExperimenterId;\r
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.ExperimenterInput;\r
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.ExperimenterInputBuilder;\r
  26 \r
  27 /**\r
  28  * @author michal.polkorab\r
  29  *\r
  30  */\r
  31 @RunWith(MockitoJUnitRunner.class)\r
  32 public class VendorInputMessageFactoryTest {\r
  33 \r
  34     @Mock SerializerRegistry registry;\r
  35     @Mock ExperimenterInputMessageFactory serializer;\r
  36 \r
  37     /**\r
  38      * Tests {@link VendorInputMessageFactory#serialize(ExperimenterInput, ByteBuf)}\r
  39      */\r
  40     @Test\r
  41     public void test() {\r
  42         Mockito.when(registry.getSerializer((MessageTypeKey<?>) Matchers.any(MessageTypeKey.class)))\r
  43             .thenReturn(serializer);\r
  44         VendorInputMessageFactory factory = new VendorInputMessageFactory();\r
  45         factory.injectSerializerRegistry(registry);\r
  46         ByteBuf buffer = PooledByteBufAllocator.DEFAULT.buffer();\r
  47         ExperimenterInputBuilder builder = new ExperimenterInputBuilder();\r
  48         builder.setVersion((short) EncodeConstants.OF10_VERSION_ID);\r
  49         builder.setXid(12345L);\r
  50         builder.setExperimenter(new ExperimenterId(42L));\r
  51         builder.setExpType(84L);\r
  52         ExperimenterInput experimenterInput = builder.build();\r
  53 \r
  54         factory.serialize(experimenterInput, buffer);\r
  55         Mockito.verify(serializer, Mockito.times(1)).serialize(experimenterInput, buffer);\r
  56     }\r
  57 }