BUG-58: refactor to take advantage of netty

[bgpcep.git] / bgp / rib-mock / src / test / java / org / opendaylight / protocol / bgp / rib / mock / BGPListenerMock.java

diff --git a/bgp/rib-mock/src/test/java/org/opendaylight/protocol/bgp/rib/mock/BGPListenerMock.java b/bgp/rib-mock/src/test/java/org/opendaylight/protocol/bgp/rib/mock/BGPListenerMock.java
index 86331cf1245090817bdff10e17e7d0aa652ea476..3cae9787a530ece29feb9245afdecb8e2eee5993 100644 (file)
--- a/bgp/rib-mock/src/test/java/org/opendaylight/protocol/bgp/rib/mock/BGPListenerMock.java
+++ b/bgp/rib-mock/src/test/java/org/opendaylight/protocol/bgp/rib/mock/BGPListenerMock.java
@@ -10,19 +10,17 @@ package org.opendaylight.protocol.bgp.rib.mock;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
-import java.util.Set;
 
-import org.opendaylight.protocol.bgp.concepts.BGPTableType;
-import org.opendaylight.protocol.bgp.parser.BGPError;
 import org.opendaylight.protocol.bgp.parser.BGPMessage;
 import org.opendaylight.protocol.bgp.parser.BGPSession;
 import org.opendaylight.protocol.bgp.parser.BGPSessionListener;
+import org.opendaylight.protocol.bgp.parser.BGPTerminationReason;
 
 
 /**
  * Mock implementation of {@link BGPListener} for testing purposes.
  */
-final class BGPListenerMock extends BGPSessionListener {
+final class BGPListenerMock implements BGPSessionListener {
        private final List<BGPMessage> buffer = Collections.synchronizedList(new ArrayList<BGPMessage>());
        private boolean connected = false;
 
@@ -35,22 +33,23 @@ final class BGPListenerMock extends BGPSessionListener {
        }
 
        @Override
-       public void onMessage(final BGPMessage message) {
+       public void onMessage(final BGPSession session, final BGPMessage message) {
                this.buffer.add(message);
        }
 
        @Override
-       public void onSessionUp(final Set<BGPTableType> remoteParams) {
+       public void onSessionUp(final BGPSession session) {
                this.connected = true;
        }
 
        @Override
        public void onSessionDown(final BGPSession session, final Exception e) {
                this.connected = false;
+
        }
 
        @Override
-       public void onSessionTerminated(final BGPError cause) {
+       public void onSessionTerminated(final BGPSession session, final BGPTerminationReason reason) {
                this.connected = false;
        }
 }