2 * Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.protocol.integration.bgp;
10 import static junit.framework.Assert.assertFalse;
11 import static junit.framework.Assert.assertTrue;
13 import com.google.common.base.Function;
14 import com.google.common.base.Optional;
15 import com.google.common.collect.Collections2;
16 import com.google.common.collect.ImmutableList;
17 import com.google.common.collect.Lists;
18 import com.google.common.eventbus.EventBus;
19 import io.netty.util.concurrent.GlobalEventExecutor;
20 import java.net.InetSocketAddress;
21 import java.util.Collection;
22 import java.util.List;
23 import java.util.concurrent.ExecutionException;
24 import javax.annotation.Nullable;
25 import org.junit.After;
26 import org.junit.Before;
27 import org.junit.Test;
28 import org.mockito.Mock;
29 import org.mockito.Mockito;
30 import org.mockito.MockitoAnnotations;
31 import org.opendaylight.controller.md.sal.binding.test.AbstractDataBrokerTest;
32 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
33 import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
34 import org.opendaylight.protocol.bgp.parser.spi.pojo.ServiceLoaderBGPExtensionProviderContext;
35 import org.opendaylight.protocol.bgp.rib.impl.BGPPeer;
36 import org.opendaylight.protocol.bgp.rib.impl.RIBActivator;
37 import org.opendaylight.protocol.bgp.rib.impl.RIBImpl;
38 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPDispatcher;
39 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPPeerRegistry;
40 import org.opendaylight.protocol.bgp.rib.mock.BGPMock;
41 import org.opendaylight.protocol.bgp.rib.spi.AbstractRIBExtensionProviderActivator;
42 import org.opendaylight.protocol.bgp.rib.spi.RIBExtensionProviderContext;
43 import org.opendaylight.protocol.bgp.rib.spi.SimpleRIBExtensionProviderContext;
44 import org.opendaylight.protocol.bgp.util.HexDumpBGPFileParser;
45 import org.opendaylight.protocol.framework.ReconnectStrategyFactory;
46 import org.opendaylight.tcpmd5.api.KeyMapping;
47 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.AsNumber;
48 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev131125.LinkstateAddressFamily;
50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev131125.LinkstateSubsequentAddressFamily;
51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType;
52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.BgpRib;
53 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.RibId;
54 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.Rib;
55 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.RibKey;
56 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.rib.LocRib;
57 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.Tables;
58 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
59 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
60 import org.opendaylight.yangtools.concepts.ListenerRegistration;
61 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
63 public class ParserToSalTest extends AbstractDataBrokerTest {
65 private static final String TEST_RIB_ID = "testRib";
67 private final String hex_messages = "/bgp_hex.txt";
70 private AbstractRIBExtensionProviderActivator baseact, lsact;
71 private RIBExtensionProviderContext ext;
74 BGPDispatcher dispatcher;
77 ReconnectStrategyFactory tcpStrategyFactory;
80 ReconnectStrategyFactory sessionStrategy;
83 public void setUp() throws Exception {
84 MockitoAnnotations.initMocks(this);
85 final List<byte[]> bgpMessages = HexDumpBGPFileParser.parseMessages(ParserToSalTest.class.getResourceAsStream(this.hex_messages));
86 this.mock = new BGPMock(new EventBus("test"), ServiceLoaderBGPExtensionProviderContext.getSingletonInstance().getMessageRegistry(), Lists.newArrayList(fixMessages(bgpMessages)));
88 Mockito.doReturn(GlobalEventExecutor.INSTANCE.newSucceededFuture(null)).when(this.dispatcher).createReconnectingClient(
89 Mockito.any(InetSocketAddress.class), Mockito.any(AsNumber.class),
90 Mockito.any(BGPPeerRegistry.class), Mockito.eq(this.tcpStrategyFactory), Mockito.eq(this.sessionStrategy),
91 Mockito.any(KeyMapping.class));
93 this.ext = new SimpleRIBExtensionProviderContext();
94 this.baseact = new RIBActivator();
95 this.lsact = new org.opendaylight.protocol.bgp.linkstate.RIBActivator();
97 this.baseact.startRIBExtensionProvider(this.ext);
98 this.lsact.startRIBExtensionProvider(this.ext);
102 public void tearDown() {
104 this.baseact.close();
107 private void runTestWithTables(final List<BgpTableType> tables) {
108 final RIBImpl rib = new RIBImpl(new RibId(TEST_RIB_ID), new AsNumber(72L), new Ipv4Address("127.0.0.1"), this.ext, this.dispatcher, this.tcpStrategyFactory, this.sessionStrategy, getDataBroker(), tables);
109 final BGPPeer peer = new BGPPeer("peer-" + this.mock.toString(), rib);
111 ListenerRegistration<?> reg = this.mock.registerUpdateListener(peer);
116 public void testWithLinkstate() throws InterruptedException, ExecutionException {
117 final List<BgpTableType> tables = ImmutableList.of(
118 (BgpTableType) new BgpTableTypeImpl(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class),
119 new BgpTableTypeImpl(LinkstateAddressFamily.class, LinkstateSubsequentAddressFamily.class));
120 runTestWithTables(tables);
121 assertTablesExists(tables);
125 public void testWithoutLinkstate() throws InterruptedException, ExecutionException {
126 final List<BgpTableType> tables = ImmutableList.of((BgpTableType) new BgpTableTypeImpl(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class));
127 runTestWithTables(tables);
128 assertTablesExists(tables);
131 private Collection<byte[]> fixMessages(final Collection<byte[]> bgpMessages) {
132 return Collections2.transform(bgpMessages, new Function<byte[], byte[]>() {
136 public byte[] apply(@Nullable final byte[] input) {
137 final byte[] ret = new byte[input.length + 1];
140 for (int i = 0; i < input.length; i++) {
141 ret[i + 1] = input[i];
148 private void assertTablesExists(final List<BgpTableType> expectedTables) throws InterruptedException, ExecutionException {
149 final Optional<LocRib> lockRib = getLocRibTable();
150 assertTrue(lockRib.isPresent());
151 final List<Tables> tables = lockRib.get().getTables();
152 assertFalse(tables.isEmpty());
153 for (final BgpTableType tableType : expectedTables) {
154 boolean found = false;
155 for (final Tables table : tables) {
156 if(table.getAfi().equals(tableType.getAfi()) && table.getSafi().equals(tableType.getSafi())) {
164 private Optional<LocRib> getLocRibTable() throws InterruptedException, ExecutionException {
165 return getDataBroker().newReadOnlyTransaction().read(LogicalDatastoreType.OPERATIONAL, InstanceIdentifier.builder(BgpRib.class).child(Rib.class, new RibKey(new RibId(TEST_RIB_ID))).child(LocRib.class).build()).get();