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
   9 package org.opendaylight.protocol.bgp.parser.impl;
  10
  11 import com.google.common.collect.Lists;
  12 import io.netty.buffer.ByteBuf;
  13 import io.netty.buffer.Unpooled;
  14 import java.util.Collections;
  15 import org.junit.Assert;
  16 import org.junit.Before;
  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.BGPExtensionProviderContext;
  22 import org.opendaylight.protocol.bgp.parser.spi.pojo.ServiceLoaderBGPExtensionProviderContext;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.open.bgp.parameters.optional.capabilities.c.parameters.GracefulRestartCaseBuilder;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.open.bgp.parameters.optional.capabilities.c.parameters.graceful.restart._case.GracefulRestartCapability.RestartFlags;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.open.bgp.parameters.optional.capabilities.c.parameters.graceful.restart._case.GracefulRestartCapabilityBuilder;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.open.bgp.parameters.optional.capabilities.c.parameters.graceful.restart._case.graceful.restart.capability.Tables;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.open.bgp.parameters.optional.capabilities.c.parameters.graceful.restart._case.graceful.restart.capability.Tables.AfiFlags;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.open.bgp.parameters.optional.capabilities.c.parameters.graceful.restart._case.graceful.restart.capability.TablesBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.AddressFamily;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.SubsequentAddressFamily;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
  33
  34 public class GracefulCapabilityHandlerTest {
  35
  36     private BGPExtensionProviderContext ctx;
  37
  38     @Before
  39     public void setUp() {
  40         this.ctx = ServiceLoaderBGPExtensionProviderContext.getSingletonInstance();
  41     }
  42
  43     @Test
  44     public void testGracefulCapabilityHandler() throws BGPDocumentedException, BGPParsingException {
  45         final GracefulCapabilityHandler handler = new GracefulCapabilityHandler(this.ctx.getAddressFamilyRegistry(),
  46                 this.ctx.getSubsequentAddressFamilyRegistry());
  47
  48         final byte[] capaBytes = {
  49             (byte) 0x40, (byte) 0x06, (byte) 0x80, (byte) 0x32,
  50             (byte) 0x00, (byte) 0x01, (byte) 0x01, (byte) 0x80
  51         };
  52
  53         final GracefulRestartCapabilityBuilder capaBuilder = new GracefulRestartCapabilityBuilder();
  54         capaBuilder.setRestartFlags(new RestartFlags(true));
  55         capaBuilder.setRestartTime(50);
  56         final TablesBuilder tablesBuilder = new TablesBuilder();
  57         tablesBuilder.setAfiFlags(new AfiFlags(true));
  58         tablesBuilder.setAfi(Ipv4AddressFamily.class);
  59         tablesBuilder.setSafi(UnicastSubsequentAddressFamily.class);
  60         capaBuilder.setTables(Lists.newArrayList(tablesBuilder.build()));
  61
  62         final ByteBuf buffer = Unpooled.buffer(capaBytes.length);
  63         handler.serializeCapability(new GracefulRestartCaseBuilder().setGracefulRestartCapability(capaBuilder.build()).build(), buffer);
  64         Assert.assertArrayEquals(capaBytes, buffer.array());
  65
  66         Assert.assertEquals(new GracefulRestartCaseBuilder().setGracefulRestartCapability(capaBuilder.build()).build(),
  67                 handler.parseCapability(Unpooled.wrappedBuffer(capaBytes).slice(2, capaBytes.length - 2)));
  68
  69         final byte[] capaBytes2 = {
  70             (byte) 0x40, (byte) 0x06, (byte) 0x00, (byte) 0x00,
  71             (byte) 0x00, (byte) 0x01, (byte) 0x01, (byte) 0x00
  72         };
  73         capaBuilder.setRestartFlags(new RestartFlags(false));
  74         capaBuilder.setRestartTime(0);
  75         tablesBuilder.setAfiFlags(new AfiFlags(false));
  76         capaBuilder.setTables(Lists.newArrayList(tablesBuilder.build()));
  77         buffer.clear();
  78         handler.serializeCapability(new GracefulRestartCaseBuilder().setGracefulRestartCapability(capaBuilder.build()).build(), buffer);
  79         Assert.assertArrayEquals(capaBytes2, buffer.array());
  80         Assert.assertEquals(new GracefulRestartCaseBuilder().setGracefulRestartCapability(capaBuilder.build()).build(),
  81                 handler.parseCapability(Unpooled.wrappedBuffer(capaBytes2).slice(2, capaBytes2.length - 2)));
  82
  83         capaBuilder.setRestartFlags(null);
  84         tablesBuilder.setAfiFlags(null);
  85         capaBuilder.setRestartTime(null);
  86         capaBuilder.setTables(Lists.newArrayList(tablesBuilder.build()));
  87         buffer.clear();
  88         handler.serializeCapability(new GracefulRestartCaseBuilder().setGracefulRestartCapability(capaBuilder.build()).build(), buffer);
  89         Assert.assertArrayEquals(capaBytes2, buffer.array());
  90
  91         final byte[] capaBytes3 = {
  92             (byte) 0x40, (byte) 0x06, (byte) 0x00, (byte) 0x00,
  93             (byte) 0x00, (byte) 0x00, (byte) 0x00, (byte) 0x00
  94         };
  95         capaBuilder.setRestartFlags(new RestartFlags(false));
  96         capaBuilder.setRestartTime(0);
  97         capaBuilder.setTables(Collections.<Tables>emptyList());
  98         Assert.assertEquals(new GracefulRestartCaseBuilder().setGracefulRestartCapability(capaBuilder.build()).build(),
  99                 handler.parseCapability(Unpooled.wrappedBuffer(capaBytes3).slice(2, capaBytes3.length - 2)));
 100
 101         final byte[] capaBytes4 = {
 102             (byte) 0x40, (byte) 0x06, (byte) 0x00, (byte) 0x00,
 103             (byte) 0x00, (byte) 0x01, (byte) 0x00, (byte) 0x00
 104         };
 105         Assert.assertEquals(new GracefulRestartCaseBuilder().setGracefulRestartCapability(capaBuilder.build()).build(),
 106                 handler.parseCapability(Unpooled.wrappedBuffer(capaBytes4).slice(2, capaBytes4.length - 2)));
 107     }
 108
 109     @Test(expected=IllegalArgumentException.class)
 110     public void testUnhandledAfi() {
 111         final GracefulCapabilityHandler handler = new GracefulCapabilityHandler(this.ctx.getAddressFamilyRegistry(),
 112                 this.ctx.getSubsequentAddressFamilyRegistry());
 113
 114         final GracefulRestartCapabilityBuilder capaBuilder = new GracefulRestartCapabilityBuilder();
 115         capaBuilder.setRestartFlags(new RestartFlags(true));
 116         capaBuilder.setRestartTime(50);
 117         final TablesBuilder tablesBuilder = new TablesBuilder();
 118         tablesBuilder.setAfiFlags(new AfiFlags(true));
 119         tablesBuilder.setAfi(AddressFamily.class);
 120         tablesBuilder.setSafi(UnicastSubsequentAddressFamily.class);
 121         capaBuilder.setTables(Lists.newArrayList(tablesBuilder.build()));
 122
 123         final ByteBuf buffer = Unpooled.buffer();
 124         handler.serializeCapability(new GracefulRestartCaseBuilder().setGracefulRestartCapability(capaBuilder.build()).build(), buffer);
 125     }
 126
 127     @Test(expected=IllegalArgumentException.class)
 128     public void testUnhandledSafi() {
 129         final GracefulCapabilityHandler handler = new GracefulCapabilityHandler(this.ctx.getAddressFamilyRegistry(),
 130                 this.ctx.getSubsequentAddressFamilyRegistry());
 131
 132         final GracefulRestartCapabilityBuilder capaBuilder = new GracefulRestartCapabilityBuilder();
 133         capaBuilder.setRestartFlags(new RestartFlags(true));
 134         capaBuilder.setRestartTime(50);
 135         final TablesBuilder tablesBuilder = new TablesBuilder();
 136         tablesBuilder.setAfiFlags(new AfiFlags(true));
 137         tablesBuilder.setAfi(Ipv4AddressFamily.class);
 138         tablesBuilder.setSafi(SubsequentAddressFamily.class);
 139         capaBuilder.setTables(Lists.newArrayList(tablesBuilder.build()));
 140
 141         final ByteBuf buffer = Unpooled.buffer();
 142         handler.serializeCapability(new GracefulRestartCaseBuilder().setGracefulRestartCapability(capaBuilder.build()).build(), buffer);
 143     }
 144
 145     @Test(expected=IllegalArgumentException.class)
 146     public void testRestartTimeMinValue() {
 147         final GracefulCapabilityHandler handler = new GracefulCapabilityHandler(this.ctx.getAddressFamilyRegistry(),
 148                 this.ctx.getSubsequentAddressFamilyRegistry());
 149
 150         final GracefulRestartCapabilityBuilder capaBuilder = new GracefulRestartCapabilityBuilder();
 151         capaBuilder.setRestartFlags(new RestartFlags(true));
 152         capaBuilder.setRestartTime(-1);
 153         final TablesBuilder tablesBuilder = new TablesBuilder();
 154         tablesBuilder.setAfiFlags(new AfiFlags(true));
 155         tablesBuilder.setAfi(Ipv4AddressFamily.class);
 156         tablesBuilder.setSafi(UnicastSubsequentAddressFamily.class);
 157         capaBuilder.setTables(Lists.newArrayList(tablesBuilder.build()));
 158
 159         final ByteBuf buffer = Unpooled.buffer();
 160         handler.serializeCapability(new GracefulRestartCaseBuilder().setGracefulRestartCapability(capaBuilder.build()).build(), buffer);
 161     }
 162
 163     @Test(expected=IllegalArgumentException.class)
 164     public void testRestartTimeMaxValue() {
 165         final GracefulCapabilityHandler handler = new GracefulCapabilityHandler(this.ctx.getAddressFamilyRegistry(),
 166                 this.ctx.getSubsequentAddressFamilyRegistry());
 167
 168         final GracefulRestartCapabilityBuilder capaBuilder = new GracefulRestartCapabilityBuilder();
 169         capaBuilder.setRestartFlags(new RestartFlags(true));
 170         capaBuilder.setRestartTime(50 * 1000);
 171         final TablesBuilder tablesBuilder = new TablesBuilder();
 172         tablesBuilder.setAfiFlags(new AfiFlags(true));
 173         tablesBuilder.setAfi(Ipv4AddressFamily.class);
 174         tablesBuilder.setSafi(UnicastSubsequentAddressFamily.class);
 175         capaBuilder.setTables(Lists.newArrayList(tablesBuilder.build()));
 176
 177         final ByteBuf buffer = Unpooled.buffer();
 178         handler.serializeCapability(new GracefulRestartCaseBuilder().setGracefulRestartCapability(capaBuilder.build()).build(), buffer);
 179     }
 180
 181 }