Extensibility support (serialization part)

[openflowjava.git] / openflow-protocol-impl / src / test / java / org / opendaylight / openflowjava / protocol / impl / serialization / factories / GetConfigInputMessageFactoryTest.java

diff --git a/openflow-protocol-impl/src/test/java/org/opendaylight/openflowjava/protocol/impl/serialization/factories/GetConfigInputMessageFactoryTest.java b/openflow-protocol-impl/src/test/java/org/opendaylight/openflowjava/protocol/impl/serialization/factories/GetConfigInputMessageFactoryTest.java
index 0c3c5a62667f547afab0b6482689fc9779c9ff8f..31d5247f30fbc24b4ed89507029a9b99f5bb0d54 100644 (file)
--- a/openflow-protocol-impl/src/test/java/org/opendaylight/openflowjava/protocol/impl/serialization/factories/GetConfigInputMessageFactoryTest.java
+++ b/openflow-protocol-impl/src/test/java/org/opendaylight/openflowjava/protocol/impl/serialization/factories/GetConfigInputMessageFactoryTest.java
@@ -11,7 +11,12 @@ package org.opendaylight.openflowjava.protocol.impl.serialization.factories;
 import io.netty.buffer.ByteBuf;
 import io.netty.buffer.UnpooledByteBufAllocator;
 
+import org.junit.Before;
 import org.junit.Test;
+import org.opendaylight.openflowjava.protocol.api.extensibility.MessageTypeKey;
+import org.opendaylight.openflowjava.protocol.api.extensibility.OFSerializer;
+import org.opendaylight.openflowjava.protocol.api.extensibility.SerializerRegistry;
+import org.opendaylight.openflowjava.protocol.impl.serialization.SerializerRegistryImpl;
 import org.opendaylight.openflowjava.protocol.impl.util.BufferHelper;
 import org.opendaylight.openflowjava.protocol.impl.util.EncodeConstants;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetConfigInput;
@@ -23,8 +28,21 @@ import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731
  */
 public class GetConfigInputMessageFactoryTest {
 
-    private static final byte GET_CONFIG_REQUEST_MESSAGE_CODE_TYPE = GetConfigInputMessageFactory.MESSAGE_TYPE;
-    
+    private static final byte GET_CONFIG_REQUEST_MESSAGE_CODE_TYPE = 7;
+    private SerializerRegistry registry;
+    private OFSerializer<GetConfigInput> getConfigFactory;
+
+    /**
+     * Initializes serializer registry and stores correct factory in field
+     */
+    @Before
+    public void startUp() {
+        registry = new SerializerRegistryImpl();
+        registry.init();
+        getConfigFactory = registry.getSerializer(
+                new MessageTypeKey<>(EncodeConstants.OF13_VERSION_ID, GetConfigInput.class));
+    }
+
     /**
      * Testing of {@link GetConfigInputMessageFactory} for correct translation from POJO
      * @throws Exception 
@@ -36,9 +54,8 @@ public class GetConfigInputMessageFactoryTest {
         GetConfigInput gci = gcib.build();
         
         ByteBuf out = UnpooledByteBufAllocator.DEFAULT.buffer();
-        GetConfigInputMessageFactory gcimf = GetConfigInputMessageFactory.getInstance();
-        gcimf.messageToBuffer(EncodeConstants.OF13_VERSION_ID, out, gci);
-        
+        getConfigFactory.serialize(gci, out);
+
         BufferHelper.checkHeaderV13(out, GET_CONFIG_REQUEST_MESSAGE_CODE_TYPE, 8);
     }
     
@@ -53,8 +70,7 @@ public class GetConfigInputMessageFactoryTest {
         GetConfigInput gci = gcib.build();
         
         ByteBuf out = UnpooledByteBufAllocator.DEFAULT.buffer();
-        GetConfigInputMessageFactory gcimf = GetConfigInputMessageFactory.getInstance();
-        gcimf.messageToBuffer(EncodeConstants.OF10_VERSION_ID, out, gci);
+        getConfigFactory.serialize(gci, out);
         
         BufferHelper.checkHeaderV10(out, GET_CONFIG_REQUEST_MESSAGE_CODE_TYPE, 8);
     }