Add LLGR restart utilities

[bgpcep.git] / bgp / rib-impl / src / test / java / org / opendaylight / protocol / bgp / rib / impl / SimpleSessionListener.java

diff --git a/bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/SimpleSessionListener.java b/bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/SimpleSessionListener.java
index 2cf19287d449b6d6790c0ce295f494c1a5bf499b..8826ddfdbadb57b4269464753fd21aca32da0221 100644 (file)
--- a/bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/SimpleSessionListener.java
+++ b/bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/SimpleSessionListener.java
@@ -7,68 +7,104 @@
  */
 package org.opendaylight.protocol.bgp.rib.impl;
 
-import java.util.List;
+import static org.junit.Assert.assertTrue;
 
-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 com.google.common.collect.Lists;
+import com.google.common.util.concurrent.Futures;
+import com.google.common.util.concurrent.ListenableFuture;
+import com.google.common.util.concurrent.Uninterruptibles;
+import java.util.List;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.TimeUnit;
+import javax.annotation.concurrent.GuardedBy;
+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.protocol.bgp.rib.spi.State;
+import org.opendaylight.protocol.util.CheckUtil.ListenerCheck;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
+import org.opendaylight.yangtools.yang.binding.Notification;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-import com.google.common.collect.Lists;
-
 /**
  * Listener for the client.
  */
-public class SimpleSessionListener implements BGPSessionListener {
-
-       private final List<BGPMessage> listMsg = Lists.newArrayList();
-
-       private BGPSessionImpl session = null;
-
-       public boolean up = false;
-
-       private static final Logger logger = LoggerFactory.getLogger(SimpleSessionListener.class);
-
-       public boolean down = false;
-
-       public SimpleSessionListener() {
-       }
-
-       public void sendMessage(final BGPMessage msg) {
-               this.session.handleMessage(msg);
-       }
-
-       public List<BGPMessage> getListMsg() {
-               return this.listMsg;
-       }
-
-       public void addSession(final BGPSessionImpl l) {
-               this.session = l;
-       }
-
-       @Override
-       public void onMessage(final BGPSession session, final BGPMessage message) {
-               this.listMsg.add(message);
-               logger.debug("Message received:" + message);
-       }
-
-       @Override
-       public void onSessionUp(final BGPSession session) {
-               logger.debug("Session Up");
-               this.up = true;
-               this.notifyAll();
-       }
-
-       @Override
-       public void onSessionDown(final BGPSession session, final Exception e) {
-               logger.debug("Session Down", e);
-               this.down = true;
-       }
-
-       @Override
-       public void onSessionTerminated(final BGPSession session, final BGPTerminationReason cause) {
-               logger.debug("Session terminated. Cause : " + cause.toString());
-       }
+public final class SimpleSessionListener implements BGPSessionListener, ListenerCheck {
+
+    private static final Logger LOG = LoggerFactory.getLogger(SimpleSessionListener.class);
+    @GuardedBy("this")
+    private final List<Notification> listMsg = Lists.newArrayList();
+    private BGPSession bgpSession;
+    private final CountDownLatch sessionLatch = new CountDownLatch(1);
+
+    public SimpleSessionListener() {
+    }
+
+    synchronized List<Notification> getListMsg() {
+        return this.listMsg;
+    }
+
+    @Override
+    public void markUptodate(final TablesKey tablesKey) {
+        LOG.debug("Table marked as up-to-date {}", tablesKey);
+    }
+
+    @Override
+    public void onSessionUp(final BGPSession session) {
+        LOG.info("Session Up");
+        this.bgpSession = session;
+        this.sessionLatch.countDown();
+    }
+
+    @Override
+    public void onSessionDown(final BGPSession session, final Exception exception) {
+        LOG.debug("Session Down", exception);
+    }
+
+    @Override
+    public void onSessionTerminated(final BGPSession session, final BGPTerminationReason cause) {
+        LOG.debug("Session terminated. Cause : {}", cause.toString());
+    }
+
+    @Override
+    public synchronized void onMessage(final BGPSession session, final Notification message) {
+        this.listMsg.add(message);
+        LOG.debug("Message received: {}", message);
+    }
+
+    @Override
+    @SuppressWarnings("checkstyle:IllegalCatch")
+    public ListenableFuture<Void> releaseConnection() {
+        LOG.debug("Releasing connection");
+        if (this.bgpSession != null) {
+            try {
+                this.bgpSession.close();
+            } catch (final Exception e) {
+                LOG.warn("Error closing session", e);
+            }
+        }
+        return Futures.immediateFuture(null);
+    }
+
+    @Override
+    public ListenableFuture<?> restartGracefully(final long selectionDeferralTimerSeconds) {
+        return Futures.immediateFailedFuture(
+                new UnsupportedOperationException("SimpleSessionListener doesn't support graceful restart"));
+    }
+
+    public State getState() {
+        return getSession().getState();
+    }
+
+    BGPSessionImpl getSession() {
+        assertTrue("Session up",
+                Uninterruptibles.awaitUninterruptibly(this.sessionLatch, 10, TimeUnit.SECONDS));
+        return (BGPSessionImpl) this.bgpSession;
+    }
+
+    @Override
+    public int getListMessageSize() {
+        return this.listMsg.size();
+    }
 }