bgp/parser-mock/src/test/java/org/opendaylight/protocol/bgp/parser/mock/BGPMessageParserMockTest.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.parser.mock;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertNotSame;
  12
  13 import com.google.common.collect.Lists;
  14 import com.google.common.collect.Maps;
  15 import com.google.common.collect.Sets;
  16 import io.netty.buffer.ByteBuf;
  17 import io.netty.buffer.Unpooled;
  18 import java.io.ByteArrayOutputStream;
  19 import java.io.IOException;
  20 import java.io.InputStream;
  21 import java.net.UnknownHostException;
  22 import java.util.List;
  23 import java.util.Map;
  24 import java.util.Set;
  25 import org.junit.Before;
  26 import org.junit.Test;
  27 import org.opendaylight.protocol.bgp.parser.BGPDocumentedException;
  28 import org.opendaylight.protocol.bgp.parser.BGPParsingException;
  29 import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
  30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.AsNumber;
  31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv6Address;
  32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv6Prefix;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev150305.ipv6.prefixes.DestinationIpv6Builder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev150305.ipv6.prefixes.destination.ipv6.Ipv6PrefixesBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev150305.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationIpv6CaseBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.Open;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.OpenBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.ProtocolVersion;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.Update;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.UpdateBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.BgpParameters;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.BgpParametersBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.OptionalCapabilities;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.OptionalCapabilitiesBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.optional.capabilities.CParameters;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.optional.capabilities.CParametersBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.AttributesBuilder;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.attributes.AsPathBuilder;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.attributes.OriginBuilder;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.attributes.as.path.Segments;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.attributes.as.path.SegmentsBuilder;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.Attributes1;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.Attributes1Builder;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.CParameters1;
  56 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.CParameters1Builder;
  57 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.mp.capabilities.MultiprotocolCapabilityBuilder;
  58 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.update.attributes.MpReachNlriBuilder;
  59 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.update.attributes.mp.reach.nlri.AdvertizedRoutesBuilder;
  60 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpOrigin;
  61 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
  62 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv6AddressFamily;
  63 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.MplsLabeledVpnSubsequentAddressFamily;
  64 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
  65 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.CNextHop;
  66 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.c.next.hop.Ipv6NextHopCaseBuilder;
  67 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.c.next.hop.ipv6.next.hop._case.Ipv6NextHopBuilder;
  68 import org.opendaylight.yangtools.yang.binding.Notification;
  69
  70 public class BGPMessageParserMockTest {
  71
  72     private final byte[][] inputBytes = new byte[11][];
  73     private final List<Update> messages = Lists.newArrayList();
  74
  75     @Before
  76     public void init() throws Exception {
  77         // Creating input bytes and update messages
  78         for (int i = 0; i < this.inputBytes.length; i++) {
  79             this.inputBytes[i] = this.fillInputBytes(i);
  80             this.messages.add(this.fillMessages(i));
  81         }
  82     }
  83
  84     /**
  85      * Test if mock implementation of parser returns correct message
  86      *
  87      * @throws BGPParsingException
  88      * @throws BGPDocumentedException
  89      * @throws IOException
  90      */
  91     @Test
  92     public void testGetUpdateMessage() throws BGPParsingException, BGPDocumentedException, IOException {
  93         final Map<ByteBuf, Notification> updateMap = Maps.newHashMap();
  94         for (int i = 0; i < this.inputBytes.length; i++) {
  95             updateMap.put(Unpooled.copiedBuffer(this.inputBytes[i]), this.messages.get(i));
  96         }
  97         final BGPMessageParserMock mockParser = new BGPMessageParserMock(updateMap);
  98
  99         for (int i = 0; i < this.inputBytes.length; i++) {
 100             assertEquals(this.messages.get(i), mockParser.parseMessage(Unpooled.copiedBuffer(this.inputBytes[i])));
 101         }
 102         assertNotSame(this.messages.get(3), mockParser.parseMessage(Unpooled.copiedBuffer(this.inputBytes[8])));
 103     }
 104
 105     /**
 106      * Test if method throws IllegalArgumentException after finding no BGPUpdateMessage associated with given byte[] key
 107      *
 108      * @throws BGPDocumentedException
 109      * @throws BGPParsingException
 110      * @throws IOException
 111      */
 112     @Test(expected = IllegalArgumentException.class)
 113     public void testGetUpdateMessageException() throws BGPParsingException, BGPDocumentedException, IOException {
 114         final Map<ByteBuf, Notification> updateMap = Maps.newHashMap();
 115         for (int i = 0; i < this.inputBytes.length; i++) {
 116             updateMap.put(Unpooled.copiedBuffer(this.inputBytes[i]), this.messages.get(i));
 117         }
 118         final BGPMessageParserMock mockParser = new BGPMessageParserMock(updateMap);
 119         mockParser.parseMessage(Unpooled.copiedBuffer(new byte[] { 7, 4, 6 }));
 120     }
 121
 122     /**
 123      * Helper method to fill inputBytes variable
 124      *
 125      * @param fileNumber parameter to distinguish between files from which bytes are read
 126      */
 127     private byte[] fillInputBytes(final int fileNumber) throws Exception {
 128         final ByteArrayOutputStream bis = new ByteArrayOutputStream();
 129         final byte[] data = new byte[60];
 130         int nRead = 0;
 131         try (final InputStream is = this.getClass().getResourceAsStream("/up" + fileNumber + ".bin")) {
 132             while ((nRead = is.read(data, 0, data.length)) != -1) {
 133                 bis.write(data, 0, nRead);
 134             }
 135             bis.flush();
 136             is.close();
 137         }
 138         return bis.toByteArray();
 139     }
 140
 141     /**
 142      * Helper method to fill messages variable
 143      *
 144      * @param asn this parameter is passed to ASNumber constructor
 145      */
 146     private Update fillMessages(final long asn) throws UnknownHostException {
 147
 148         final UpdateBuilder builder = new UpdateBuilder();
 149
 150         final List<Segments> asPath = Lists.newArrayList();
 151         asPath.add(new SegmentsBuilder().setAsSequence(Lists.newArrayList(new AsNumber(asn))).build());
 152         final CNextHop nextHop = new Ipv6NextHopCaseBuilder().setIpv6NextHop(
 153             new Ipv6NextHopBuilder().setGlobal(new Ipv6Address("2001:db8::1")).setLinkLocal(new Ipv6Address("fe80::c001:bff:fe7e:0")).build()).build();
 154
 155         final Ipv6Prefix pref1 = new Ipv6Prefix("2001:db8:1:2::/64");
 156         final Ipv6Prefix pref2 = new Ipv6Prefix("2001:db8:1:1::/64");
 157         final Ipv6Prefix pref3 = new Ipv6Prefix("2001:db8:1::/64");
 158
 159         final AttributesBuilder paBuilder = new AttributesBuilder();
 160         paBuilder.setOrigin(new OriginBuilder().setValue(BgpOrigin.Igp).build());
 161         paBuilder.setAsPath(new AsPathBuilder().setSegments(asPath).build());
 162
 163         final MpReachNlriBuilder mpReachBuilder = new MpReachNlriBuilder();
 164         mpReachBuilder.setAfi(Ipv6AddressFamily.class);
 165         mpReachBuilder.setSafi(UnicastSubsequentAddressFamily.class);
 166         mpReachBuilder.setCNextHop(nextHop);
 167         mpReachBuilder.setAdvertizedRoutes(new AdvertizedRoutesBuilder().setDestinationType(
 168             new DestinationIpv6CaseBuilder().setDestinationIpv6(
 169                 new DestinationIpv6Builder().setIpv6Prefixes(Lists.newArrayList(
 170                     new Ipv6PrefixesBuilder().setPrefix(pref1).build(),new Ipv6PrefixesBuilder().setPrefix(pref2).build(),
 171                     new Ipv6PrefixesBuilder().setPrefix(pref3).build())).build()).build()).build());
 172
 173         paBuilder.addAugmentation(Attributes1.class, new Attributes1Builder().setMpReachNlri(mpReachBuilder.build()).build());
 174
 175         builder.setAttributes(paBuilder.build());
 176
 177         return builder.build();
 178     }
 179
 180     @Test
 181     public void testGetOpenMessage() throws BGPParsingException, BGPDocumentedException, IOException {
 182         final Map<ByteBuf, Notification> openMap = Maps.newHashMap();
 183
 184         final Set<BgpTableType> type = Sets.newHashSet();
 185         type.add(new BgpTableTypeImpl(Ipv4AddressFamily.class, MplsLabeledVpnSubsequentAddressFamily.class));
 186
 187         final List<BgpParameters> params = Lists.newArrayList();
 188
 189         final CParameters par = new CParametersBuilder().addAugmentation(CParameters1.class, new CParameters1Builder()
 190             .setMultiprotocolCapability(new MultiprotocolCapabilityBuilder().setAfi(Ipv4AddressFamily.class)
 191                 .setSafi(MplsLabeledVpnSubsequentAddressFamily.class).build()).build()).build();
 192         params.add(new BgpParametersBuilder().setOptionalCapabilities(Lists.newArrayList(
 193             new OptionalCapabilitiesBuilder().setCParameters(par).build())).build());
 194
 195         final byte[] input = new byte[] { 5, 8, 13, 21 };
 196
 197         openMap.put(Unpooled.copiedBuffer(input), new OpenBuilder().setMyAsNumber(30).setHoldTimer(30).setBgpParameters(params).setVersion(
 198             new ProtocolVersion((short) 4)).build());
 199
 200         final BGPMessageParserMock mockParser = new BGPMessageParserMock(openMap);
 201
 202         final Set<BgpTableType> result = Sets.newHashSet();
 203         for (final BgpParameters p : ((Open) mockParser.parseMessage(Unpooled.copiedBuffer(input))).getBgpParameters()) {
 204             for (final OptionalCapabilities capa : p.getOptionalCapabilities()) {
 205                 final CParameters cp = capa.getCParameters();
 206                 if (cp.getAugmentation(CParameters1.class) != null && cp.getAugmentation(CParameters1.class).getMultiprotocolCapability() != null) {
 207                     final BgpTableType t = new BgpTableTypeImpl(cp.getAugmentation(CParameters1.class).getMultiprotocolCapability().getAfi(),
 208                             cp.getAugmentation(CParameters1.class).getMultiprotocolCapability().getSafi());
 209                     result.add(t);
 210                 }
 211             }
 212         }
 213         assertEquals(type, result);
 214     }
 215 }