Initial framework migration to netty.

[bgpcep.git] / bgp / rib-mock / src / main / java / org / opendaylight / protocol / bgp / rib / mock / BGPMock.java

diff --git a/bgp/rib-mock/src/main/java/org/opendaylight/protocol/bgp/rib/mock/BGPMock.java b/bgp/rib-mock/src/main/java/org/opendaylight/protocol/bgp/rib/mock/BGPMock.java
index 7b2aaeaf5ba0cb918ea6183de47c679414c8b694..886b4cbdc440dd8dd8c1c9dc4fc7266cf1a293dd 100644 (file)
--- a/bgp/rib-mock/src/main/java/org/opendaylight/protocol/bgp/rib/mock/BGPMock.java
+++ b/bgp/rib-mock/src/main/java/org/opendaylight/protocol/bgp/rib/mock/BGPMock.java
@@ -8,7 +8,6 @@
 package org.opendaylight.protocol.bgp.rib.mock;
 
 import java.io.Closeable;
-import java.io.IOException;
 import java.util.Arrays;
 import java.util.Iterator;
 import java.util.List;
@@ -18,17 +17,16 @@ import javax.annotation.concurrent.ThreadSafe;
 
 import org.opendaylight.protocol.bgp.parser.BGPError;
 import org.opendaylight.protocol.bgp.parser.BGPMessage;
-import org.opendaylight.protocol.bgp.parser.BGPMessageHeader;
-import org.opendaylight.protocol.bgp.parser.BGPMessageParser;
 import org.opendaylight.protocol.bgp.parser.BGPSessionListener;
 import org.opendaylight.protocol.bgp.parser.impl.BGPMessageFactory;
 import org.opendaylight.protocol.bgp.parser.message.BGPNotificationMessage;
 import org.opendaylight.protocol.bgp.rib.impl.BGP;
-import org.opendaylight.protocol.util.ByteArray;
-
 import org.opendaylight.protocol.concepts.ListenerRegistration;
 import org.opendaylight.protocol.framework.DeserializerException;
 import org.opendaylight.protocol.framework.DocumentedException;
+import org.opendaylight.protocol.framework.ProtocolMessageFactory;
+import org.opendaylight.protocol.util.ByteArray;
+
 import com.google.common.collect.Lists;
 import com.google.common.eventbus.EventBus;
 
@@ -56,19 +54,14 @@ public final class BGPMock implements BGP, Closeable {
 
        private List<BGPMessage> parsePrevious(final List<byte[]> msgs) {
                final List<BGPMessage> messages = Lists.newArrayList();
-               final BGPMessageParser parser = new BGPMessageFactory();
+               final ProtocolMessageFactory parser = new BGPMessageFactory();
                try {
                        for (final byte[] b : msgs) {
-                               final BGPMessageHeader header = new BGPMessageHeader();
-                               header.fromBytes(ByteArray.subByte(b, 0, BGPMessageHeader.COMMON_HEADER_LENGTH));
 
-                               final byte[] body = ByteArray.cutBytes(b, BGPMessageHeader.COMMON_HEADER_LENGTH);
+                               final byte[] body = ByteArray.cutBytes(b, 1);
 
-                               messages.add(parser.parse(body, header));
+                               messages.add((BGPMessage) parser.parse(body));
                        }
-                       parser.close();
-               } catch (final IOException e) {
-                       e.printStackTrace();
                } catch (final DeserializerException e) {
                        e.printStackTrace();
                } catch (final DocumentedException e) {