Refactor LinkstateNlriParser

[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 3cae9787a530ece29feb9245afdecb8e2eee5993..9b34e7f6e73d84cda7619285cdb3c7784a4c5804 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,46 +10,54 @@ package org.opendaylight.protocol.bgp.rib.mock;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
-
-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;
-
+import org.opendaylight.protocol.bgp.rib.spi.BGPSession;
+import org.opendaylight.protocol.bgp.rib.spi.BGPSessionListener;
+import org.opendaylight.protocol.bgp.rib.spi.BGPTerminationReason;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
+import org.opendaylight.yangtools.yang.binding.Notification;
 
 /**
- * Mock implementation of {@link BGPListener} for testing purposes.
+ * Mock implementation of {@link BGPSessionListener} for testing purposes.
  */
-final class BGPListenerMock implements BGPSessionListener {
-       private final List<BGPMessage> buffer = Collections.synchronizedList(new ArrayList<BGPMessage>());
-       private boolean connected = false;
-
-       protected List<BGPMessage> getBuffer() {
-               return this.buffer;
-       }
-
-       protected boolean isConnected() {
-               return this.connected;
-       }
-
-       @Override
-       public void onMessage(final BGPSession session, final BGPMessage message) {
-               this.buffer.add(message);
-       }
-
-       @Override
-       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 BGPSession session, final BGPTerminationReason reason) {
-               this.connected = false;
-       }
+public final class BGPListenerMock implements BGPSessionListener {
+    private final List<Notification> buffer = Collections.synchronizedList(new ArrayList<Notification>());
+    private boolean connected = false;
+
+    protected List<Notification> getBuffer() {
+        return this.buffer;
+    }
+
+    protected boolean isConnected() {
+        return this.connected;
+    }
+
+    @Override
+    public void onMessage(final BGPSession session, final Notification message) {
+        this.buffer.add(message);
+    }
+
+    @Override
+    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 BGPSession session, final BGPTerminationReason reason) {
+        this.connected = false;
+    }
+
+    @Override
+    public boolean isSessionActive() {
+        return true;
+    }
+
+    @Override
+    public void markUptodate(final TablesKey tablesKey) {
+    }
 }