bgp/parser-impl/src/test/java/org/opendaylight/protocol/bgp/parser/impl/GracefulCapabilityHandlerTest.java

   1 /*
   2  * Copyright (c) 2014 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.impl;
   9
  10 import static org.junit.Assert.assertArrayEquals;
  11 import static org.junit.Assert.assertEquals;
  12
  13 import io.netty.buffer.ByteBuf;
  14 import io.netty.buffer.Unpooled;
  15 import java.util.Map;
  16 import java.util.ServiceLoader;
  17 import org.junit.Test;
  18 import org.opendaylight.protocol.bgp.parser.BGPDocumentedException;
  19 import org.opendaylight.protocol.bgp.parser.BGPParsingException;
  20 import org.opendaylight.protocol.bgp.parser.impl.message.open.GracefulCapabilityHandler;
  21 import org.opendaylight.protocol.bgp.parser.spi.BGPExtensionConsumerContext;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.open.message.bgp.parameters.optional.capabilities.CParametersBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.CParameters1Builder;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.GracefulRestartCapability.RestartFlags;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.GracefulRestartCapabilityBuilder;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.graceful.restart.capability.Tables.AfiFlags;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.graceful.restart.capability.TablesBuilder;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.AddressFamily;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.Ipv4AddressFamily;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.SubsequentAddressFamily;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.UnicastSubsequentAddressFamily;
  32 import org.opendaylight.yangtools.yang.binding.util.BindingMap;
  33 import org.opendaylight.yangtools.yang.common.Uint16;
  34
  35 public class GracefulCapabilityHandlerTest {
  36     private final BGPExtensionConsumerContext ctx = ServiceLoader.load(BGPExtensionConsumerContext.class).findFirst()
  37         .orElseThrow();
  38
  39     @Test
  40     public void testGracefulCapabilityHandler() throws BGPDocumentedException, BGPParsingException {
  41         final GracefulCapabilityHandler handler = new GracefulCapabilityHandler(
  42             ctx.getAddressFamilyRegistry(),ctx.getSubsequentAddressFamilyRegistry());
  43
  44         final byte[] capaBytes = {
  45             (byte) 0x40, (byte) 0x06, (byte) 0x81, (byte) 0xf4,
  46             (byte) 0x00, (byte) 0x01, (byte) 0x01, (byte) 0x80
  47         };
  48
  49         final GracefulRestartCapabilityBuilder capaBuilder = new GracefulRestartCapabilityBuilder()
  50                 .setRestartFlags(new RestartFlags(true))
  51                 .setRestartTime(Uint16.valueOf(500));
  52         final TablesBuilder tablesBuilder = new TablesBuilder()
  53                 .setAfiFlags(new AfiFlags(true))
  54                 .setAfi(Ipv4AddressFamily.VALUE)
  55                 .setSafi(UnicastSubsequentAddressFamily.VALUE);
  56         capaBuilder.setTables(BindingMap.of(tablesBuilder.build()));
  57
  58         final ByteBuf buffer = Unpooled.buffer(capaBytes.length);
  59         handler.serializeCapability(new CParametersBuilder()
  60             .addAugmentation(new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build())
  61             .build(), buffer);
  62         assertArrayEquals(capaBytes, buffer.array());
  63
  64         assertEquals(new CParametersBuilder()
  65             .addAugmentation(new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build())
  66             .build(), handler.parseCapability(Unpooled.wrappedBuffer(capaBytes).slice(2, capaBytes.length - 2)));
  67
  68         final byte[] capaBytes2 = {
  69             (byte) 0x40, (byte) 0x06, (byte) 0x00, (byte) 0x00,
  70             (byte) 0x00, (byte) 0x01, (byte) 0x01, (byte) 0x00
  71         };
  72         capaBuilder.setRestartFlags(new RestartFlags(false));
  73         capaBuilder.setRestartTime(Uint16.ZERO);
  74         tablesBuilder.setAfiFlags(new AfiFlags(false));
  75         capaBuilder.setTables(BindingMap.of(tablesBuilder.build()));
  76         buffer.clear();
  77         handler.serializeCapability(new CParametersBuilder()
  78             .addAugmentation(new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build())
  79             .build(), buffer);
  80         assertArrayEquals(capaBytes2, buffer.array());
  81         assertEquals(new CParametersBuilder()
  82             .addAugmentation(new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build())
  83             .build(), handler.parseCapability(Unpooled.wrappedBuffer(capaBytes2).slice(2, capaBytes2.length - 2)));
  84
  85         capaBuilder.setRestartFlags(null);
  86         tablesBuilder.setAfiFlags(null);
  87         capaBuilder.setRestartTime((Uint16) null);
  88         capaBuilder.setTables(BindingMap.of(tablesBuilder.build()));
  89         buffer.clear();
  90         handler.serializeCapability(new CParametersBuilder()
  91             .addAugmentation(new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build())
  92             .build(), buffer);
  93         assertArrayEquals(capaBytes2, buffer.array());
  94
  95         final byte[] capaBytes3 = {
  96             (byte) 0x40, (byte) 0x06, (byte) 0x00, (byte) 0x00,
  97             (byte) 0x00, (byte) 0x00, (byte) 0x00, (byte) 0x00
  98         };
  99         capaBuilder.setRestartFlags(new RestartFlags(false));
 100         capaBuilder.setRestartTime(Uint16.ZERO);
 101         capaBuilder.setTables(Map.of());
 102         assertEquals(new CParametersBuilder()
 103             .addAugmentation(new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build())
 104             .build(), handler.parseCapability(Unpooled.wrappedBuffer(capaBytes3).slice(2, capaBytes3.length - 2)));
 105
 106         final byte[] capaBytes4 = {
 107             (byte) 0x40, (byte) 0x06, (byte) 0x00, (byte) 0x00,
 108             (byte) 0x00, (byte) 0x01, (byte) 0x00, (byte) 0x00
 109         };
 110         assertEquals(new CParametersBuilder()
 111             .addAugmentation(new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build())
 112             .build(), handler.parseCapability(Unpooled.wrappedBuffer(capaBytes4).slice(2, capaBytes4.length - 2)));
 113         final byte[] capaBytes5 = {
 114             (byte) 0x40, (byte) 0x02, (byte) 0x00, (byte) 0x00
 115         };
 116         final ByteBuf b = Unpooled.buffer(capaBytes5.length);
 117
 118         handler.serializeCapability(new CParametersBuilder()
 119             .addAugmentation(new CParameters1Builder()
 120                 .setGracefulRestartCapability(new GracefulRestartCapabilityBuilder().build())
 121                 .build())
 122             .build(), b);
 123         assertArrayEquals(capaBytes5, b.array());
 124         assertEquals(new CParametersBuilder()
 125             .addAugmentation(new CParameters1Builder()
 126                 .setGracefulRestartCapability(new GracefulRestartCapabilityBuilder()
 127                     .setRestartFlags(new RestartFlags(Boolean.FALSE))
 128                     .setRestartTime(Uint16.ZERO)
 129                     .build())
 130                 .build())
 131             .build(), handler.parseCapability(Unpooled.wrappedBuffer(capaBytes5).slice(2, capaBytes5.length - 2)));
 132     }
 133
 134     @Test(expected = IllegalArgumentException.class)
 135     public void testUnhandledAfi() {
 136         final GracefulCapabilityHandler handler = new GracefulCapabilityHandler(ctx.getAddressFamilyRegistry(),
 137             ctx.getSubsequentAddressFamilyRegistry());
 138
 139         final GracefulRestartCapabilityBuilder capaBuilder = new GracefulRestartCapabilityBuilder();
 140         capaBuilder.setRestartFlags(new RestartFlags(true));
 141         capaBuilder.setRestartTime(Uint16.valueOf(50));
 142         capaBuilder.setTables(BindingMap.of(new TablesBuilder()
 143             .setAfiFlags(new AfiFlags(true))
 144             .setAfi(AddressFamily.VALUE)
 145             .setSafi(UnicastSubsequentAddressFamily.VALUE)
 146             .build()));
 147
 148         final ByteBuf buffer = Unpooled.buffer();
 149         handler.serializeCapability(new CParametersBuilder()
 150             .addAugmentation(new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build())
 151             .build(), buffer);
 152     }
 153
 154     @Test(expected = IllegalArgumentException.class)
 155     public void testUnhandledSafi() {
 156         final GracefulCapabilityHandler handler = new GracefulCapabilityHandler(ctx.getAddressFamilyRegistry(),
 157             ctx.getSubsequentAddressFamilyRegistry());
 158
 159         final GracefulRestartCapabilityBuilder capaBuilder = new GracefulRestartCapabilityBuilder();
 160         capaBuilder.setRestartFlags(new RestartFlags(true));
 161         capaBuilder.setRestartTime(Uint16.valueOf(50));
 162         capaBuilder.setTables(BindingMap.of(new TablesBuilder()
 163             .setAfiFlags(new AfiFlags(true))
 164             .setAfi(Ipv4AddressFamily.VALUE)
 165             .setSafi(SubsequentAddressFamily.VALUE)
 166             .build()));
 167
 168         final ByteBuf buffer = Unpooled.buffer();
 169         handler.serializeCapability(new CParametersBuilder()
 170             .addAugmentation(new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build())
 171             .build(), buffer);
 172     }
 173
 174     @Test(expected = IllegalArgumentException.class)
 175     public void testRestartTimeMinValue() {
 176         final GracefulCapabilityHandler handler = new GracefulCapabilityHandler(ctx.getAddressFamilyRegistry(),
 177             ctx.getSubsequentAddressFamilyRegistry());
 178
 179         final GracefulRestartCapabilityBuilder capaBuilder = new GracefulRestartCapabilityBuilder()
 180                 .setRestartFlags(new RestartFlags(true))
 181                 // FIXME: this is throwing IAE, why is the rest of the test even here?
 182                 .setRestartTime(Uint16.MAX_VALUE)
 183                 .setTables(BindingMap.of(new TablesBuilder()
 184                     .setAfiFlags(new AfiFlags(true))
 185                     .setAfi(Ipv4AddressFamily.VALUE)
 186                     .setSafi(UnicastSubsequentAddressFamily.VALUE)
 187                     .build()));
 188
 189         final ByteBuf buffer = Unpooled.buffer();
 190         handler.serializeCapability(new CParametersBuilder()
 191             .addAugmentation(new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build())
 192             .build(), buffer);
 193     }
 194
 195     @Test(expected = IllegalArgumentException.class)
 196     public void testRestartTimeMaxValue() {
 197         final GracefulCapabilityHandler handler = new GracefulCapabilityHandler(ctx.getAddressFamilyRegistry(),
 198             ctx.getSubsequentAddressFamilyRegistry());
 199
 200         final GracefulRestartCapabilityBuilder capaBuilder = new GracefulRestartCapabilityBuilder()
 201                 .setRestartFlags(new RestartFlags(true))
 202                 .setRestartTime(Uint16.valueOf(50000));
 203
 204         capaBuilder.setTables(BindingMap.of(new TablesBuilder()
 205             .setAfiFlags(new AfiFlags(true))
 206             .setAfi(Ipv4AddressFamily.VALUE)
 207             .setSafi(UnicastSubsequentAddressFamily.VALUE)
 208             .build()));
 209
 210         final ByteBuf buffer = Unpooled.buffer();
 211         handler.serializeCapability(new CParametersBuilder()
 212             .addAugmentation(new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build())
 213             .build(), buffer);
 214     }
 215 }