bgp/rib-mock/src/main/java/org/opendaylight/protocol/bgp/rib/mock/BGPMock.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.protocol.bgp.rib.mock;
   9
  10 import java.io.Closeable;
  11 import java.io.IOException;
  12 import java.util.Arrays;
  13 import java.util.Iterator;
  14 import java.util.List;
  15
  16 import javax.annotation.concurrent.GuardedBy;
  17 import javax.annotation.concurrent.ThreadSafe;
  18
  19 import org.opendaylight.protocol.bgp.parser.BGPError;
  20 import org.opendaylight.protocol.bgp.parser.BGPMessage;
  21 import org.opendaylight.protocol.bgp.parser.BGPMessageHeader;
  22 import org.opendaylight.protocol.bgp.parser.BGPMessageParser;
  23 import org.opendaylight.protocol.bgp.parser.BGPSessionListener;
  24 import org.opendaylight.protocol.bgp.parser.impl.BGPMessageFactory;
  25 import org.opendaylight.protocol.bgp.parser.message.BGPNotificationMessage;
  26 import org.opendaylight.protocol.bgp.rib.impl.BGP;
  27 import org.opendaylight.protocol.util.ByteArray;
  28
  29 import org.opendaylight.protocol.concepts.ListenerRegistration;
  30 import org.opendaylight.protocol.framework.DeserializerException;
  31 import org.opendaylight.protocol.framework.DocumentedException;
  32 import com.google.common.collect.Lists;
  33 import com.google.common.eventbus.EventBus;
  34
  35 /**
  36  *
  37  * Mock implementation of {@link BGP}.
  38  *
  39  */
  40 @ThreadSafe
  41 public final class BGPMock implements BGP, Closeable {
  42         static final BGPMessage connectionLostMagicMessage = new BGPNotificationMessage(BGPError.CEASE);
  43
  44         @GuardedBy("this")
  45         private final List<byte[]> allPreviousByteMessages;
  46         private final List<BGPMessage> allPreviousBGPMessages;
  47         private final EventBus eventBus;
  48         @GuardedBy("this")
  49         private final List<EventBusRegistration> openRegistrations = Lists.newLinkedList();
  50
  51         public BGPMock(final EventBus eventBus, final List<byte[]> bgpMessages) {
  52                 this.allPreviousByteMessages = Lists.newLinkedList(bgpMessages);
  53                 this.eventBus = eventBus;
  54                 this.allPreviousBGPMessages = this.parsePrevious(this.allPreviousByteMessages);
  55         }
  56
  57         private List<BGPMessage> parsePrevious(final List<byte[]> msgs) {
  58                 final List<BGPMessage> messages = Lists.newArrayList();
  59                 final BGPMessageParser parser = new BGPMessageFactory();
  60                 try {
  61                         for (final byte[] b : msgs) {
  62                                 final BGPMessageHeader header = new BGPMessageHeader();
  63                                 header.fromBytes(ByteArray.subByte(b, 0, BGPMessageHeader.COMMON_HEADER_LENGTH));
  64
  65                                 final byte[] body = ByteArray.cutBytes(b, BGPMessageHeader.COMMON_HEADER_LENGTH);
  66
  67                                 messages.add(parser.parse(body, header));
  68                         }
  69                         parser.close();
  70                 } catch (final IOException e) {
  71                         e.printStackTrace();
  72                 } catch (final DeserializerException e) {
  73                         e.printStackTrace();
  74                 } catch (final DocumentedException e) {
  75                         e.printStackTrace();
  76                 }
  77                 return messages;
  78         }
  79
  80         /**
  81          * @param listener BGPListener
  82          * @return ListenerRegistration
  83          */
  84         @Override
  85         public synchronized ListenerRegistration<BGPSessionListener> registerUpdateListener(final BGPSessionListener listener) {
  86                 return EventBusRegistration.createAndRegister(this.eventBus, listener, this.allPreviousBGPMessages);
  87         }
  88
  89         public synchronized void insertConnectionLostEvent() {
  90                 this.insertMessage(connectionLostMagicMessage);
  91         }
  92
  93         public synchronized void insertMessages(final List<BGPMessage> messages) {
  94                 for (final BGPMessage message : messages)
  95                         this.insertMessage(message);
  96         }
  97
  98         private synchronized void insertMessage(final BGPMessage message) {
  99                 this.allPreviousBGPMessages.add(message);
 100                 this.eventBus.post(message);
 101         }
 102
 103         @Override
 104         public synchronized void close() {
 105                 // unregister all EventBusRegistration instances
 106                 for (final EventBusRegistration registration : this.openRegistrations) {
 107                         registration.close();
 108                 }
 109                 this.openRegistrations.clear();
 110         }
 111
 112         public boolean isMessageListSame(final List<byte[]> newMessages) {
 113                 if (this.allPreviousBGPMessages.size() != newMessages.size())
 114                         return false;
 115                 final Iterator<byte[]> i1 = this.allPreviousByteMessages.iterator();
 116                 final Iterator<byte[]> i2 = this.allPreviousByteMessages.iterator();
 117                 for (int i = 0; i < this.allPreviousBGPMessages.size(); i++) {
 118                         if (!Arrays.equals(i1.next(), i2.next()))
 119                                 return false;
 120                 }
 121                 return true;
 122         }
 123
 124         public EventBus getEventBus() {
 125                 return this.eventBus;
 126         }
 127 }