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

   1 /*
   2  * Copyright (c) 2016 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.ArrayList;
  16 import java.util.List;
  17 import org.junit.Before;
  18 import org.junit.Test;
  19 import org.mockito.Mock;
  20 import org.mockito.Mockito;
  21 import org.mockito.MockitoAnnotations;
  22 import org.opendaylight.protocol.bgp.parser.BGPDocumentedException;
  23 import org.opendaylight.protocol.bgp.parser.BGPParsingException;
  24 import org.opendaylight.protocol.bgp.parser.impl.message.open.AddPathCapabilityHandler;
  25 import org.opendaylight.protocol.bgp.parser.spi.AddressFamilyRegistry;
  26 import org.opendaylight.protocol.bgp.parser.spi.SubsequentAddressFamilyRegistry;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.open.message.bgp.parameters.optional.capabilities.CParameters;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.open.message.bgp.parameters.optional.capabilities.CParametersBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.CParameters1;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.CParameters1Builder;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.SendReceive;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.AddPathCapabilityBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.add.path.capability.AddressFamilies;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.add.path.capability.AddressFamiliesBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.Ipv6AddressFamily;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.UnicastSubsequentAddressFamily;
  37
  38 public class AddPathCapabilityHandlerTest {
  39     private static final Class<Ipv6AddressFamily> AFI = Ipv6AddressFamily.class;
  40     private static final Class<UnicastSubsequentAddressFamily> SAFI = UnicastSubsequentAddressFamily.class;
  41
  42     @Mock
  43     private AddressFamilyRegistry afiRegistry;
  44     @Mock
  45     private AddressFamilyRegistry afirExpection;
  46     @Mock
  47     private SubsequentAddressFamilyRegistry safiRegistry;
  48     @Mock
  49     private SubsequentAddressFamilyRegistry safirException;
  50     private final ByteBuf parseCorrectBytes = Unpooled.copiedBuffer(new byte[] {1, 4, 4, 1});
  51     private final ByteBuf parseWrongBytes = Unpooled.copiedBuffer(new byte[] {1, 4, 4, 4});
  52     private final byte[] serializedBytes = new byte[] {AddPathCapabilityHandler.CODE, 4, 1, 4, 4, 1};
  53
  54     @Before
  55     public void setUp() {
  56         MockitoAnnotations.initMocks(this);
  57         Mockito.doReturn(260).when(this.afiRegistry).numberForClass(AFI);
  58         Mockito.doReturn(AddPathCapabilityHandlerTest.AFI).when(this.afiRegistry).classForFamily(260);
  59
  60         Mockito.doReturn(null).when(this.afirExpection).numberForClass(AFI);
  61         Mockito.doReturn(null).when(this.afirExpection).classForFamily(260);
  62
  63         Mockito.doReturn(4).when(this.safiRegistry).numberForClass(SAFI);
  64         Mockito.doReturn(AddPathCapabilityHandlerTest.SAFI).when(this.safiRegistry).classForFamily(4);
  65
  66         Mockito.doReturn(null).when(this.safirException).numberForClass(SAFI);
  67         Mockito.doReturn(null).when(this.safirException).classForFamily(4);
  68     }
  69
  70     @Test
  71     public void testCapabilityHandler() throws BGPDocumentedException, BGPParsingException {
  72         final List<AddressFamilies> family = new ArrayList<>();
  73         family.add(new AddressFamiliesBuilder().setAfi(AddPathCapabilityHandlerTest.AFI)
  74             .setSafi(AddPathCapabilityHandlerTest.SAFI).setSendReceive(SendReceive.forValue(1)).build());
  75
  76         final CParameters capabilityToSerialize = new CParametersBuilder()
  77                 .addAugmentation(new CParameters1Builder()
  78                     .setAddPathCapability(new AddPathCapabilityBuilder().setAddressFamilies(family).build())
  79                     .build())
  80                 .build();
  81
  82         final ByteBuf bytes = Unpooled.buffer(6);
  83         final AddPathCapabilityHandler handler = new AddPathCapabilityHandler(this.afiRegistry, this.safiRegistry);
  84         handler.serializeCapability(capabilityToSerialize, bytes);
  85         assertArrayEquals(this.serializedBytes, bytes.array());
  86
  87         final CParameters newCaps = handler.parseCapability(this.parseCorrectBytes);
  88         assertEquals(capabilityToSerialize.hashCode(), newCaps.hashCode());
  89     }
  90
  91     @Test(expected = BGPParsingException.class)
  92     public void testAfiException() throws BGPDocumentedException, BGPParsingException {
  93         final ByteBuf bytes = this.parseWrongBytes.copy();
  94         final AddPathCapabilityHandler handler = new AddPathCapabilityHandler(this.afirExpection, this.safiRegistry);
  95         handler.parseCapability(bytes);
  96     }
  97
  98     @Test(expected = BGPParsingException.class)
  99     public void testSafiException() throws BGPDocumentedException, BGPParsingException {
 100         final ByteBuf bytes = this.parseWrongBytes.copy();
 101         final AddPathCapabilityHandler handler = new AddPathCapabilityHandler(this.afiRegistry, this.safirException);
 102         handler.parseCapability(bytes);
 103     }
 104
 105     @Test
 106     public void testSendReceiveIgnored() throws BGPDocumentedException, BGPParsingException {
 107         final CParameters capabilityToSerialize = new CParametersBuilder()
 108                 .addAugmentation(new CParameters1Builder()
 109                     .setAddPathCapability(new AddPathCapabilityBuilder().setAddressFamilies(new ArrayList<>()).build())
 110                     .build())
 111                 .build();
 112
 113         final ByteBuf bytes = this.parseWrongBytes.copy();
 114         final AddPathCapabilityHandler handler = new AddPathCapabilityHandler(this.afiRegistry, this.safiRegistry);
 115         final CParameters newCaps = handler.parseCapability(bytes);
 116         assertEquals(capabilityToSerialize.hashCode(), newCaps.hashCode());
 117     }
 118
 119     @Test(expected = IllegalArgumentException.class)
 120     public void testUnhandledAfi() {
 121         final List<AddressFamilies> family = new ArrayList<>();
 122         family.add(new AddressFamiliesBuilder().setAfi(AddPathCapabilityHandlerTest.AFI)
 123             .setSafi(AddPathCapabilityHandlerTest.SAFI).setSendReceive(SendReceive.forValue(2)).build());
 124
 125         final CParameters capabilityToSerialize = new CParametersBuilder()
 126                 .addAugmentation(new CParameters1Builder()
 127                     .setAddPathCapability(new AddPathCapabilityBuilder().setAddressFamilies(family).build())
 128                     .build())
 129                 .build();
 130
 131         final ByteBuf bytes = Unpooled.buffer();
 132         final AddPathCapabilityHandler handler = new AddPathCapabilityHandler(this.afirExpection, this.safiRegistry);
 133         handler.serializeCapability(capabilityToSerialize, bytes);
 134     }
 135
 136     @Test(expected = IllegalArgumentException.class)
 137     public void testUnhandledSafi() {
 138         final List<AddressFamilies> family = new ArrayList<>();
 139         family.add(new AddressFamiliesBuilder().setAfi(AddPathCapabilityHandlerTest.AFI)
 140             .setSafi(AddPathCapabilityHandlerTest.SAFI).setSendReceive(SendReceive.forValue(3)).build());
 141
 142         final CParameters capabilityToSerialize = new CParametersBuilder()
 143                 .addAugmentation(new CParameters1Builder()
 144                     .setAddPathCapability(new AddPathCapabilityBuilder().setAddressFamilies(family).build())
 145                     .build())
 146                 .build();
 147
 148         final ByteBuf bytes = Unpooled.buffer();
 149         final AddPathCapabilityHandler handler = new AddPathCapabilityHandler(this.afiRegistry, this.safirException);
 150         handler.serializeCapability(capabilityToSerialize, bytes);
 151     }
 152
 153     @Test(expected = IllegalArgumentException.class)
 154     public void testUnhandledSendReceive() {
 155         final List<AddressFamilies> family = new ArrayList<>();
 156         family.add(new AddressFamiliesBuilder().setAfi(AddPathCapabilityHandlerTest.AFI)
 157             .setSafi(AddPathCapabilityHandlerTest.SAFI).setSendReceive(SendReceive.forValue(4)).build());
 158
 159         final CParameters capabilityToSerialize = new CParametersBuilder()
 160                 .addAugmentation(new CParameters1Builder()
 161                     .setAddPathCapability(new AddPathCapabilityBuilder().setAddressFamilies(family).build())
 162                     .build())
 163                 .build();
 164
 165         final ByteBuf bytes = Unpooled.buffer();
 166         final AddPathCapabilityHandler handler = new AddPathCapabilityHandler(this.afiRegistry, this.safiRegistry);
 167         handler.serializeCapability(capabilityToSerialize, bytes);
 168     }
 169
 170     @Test
 171     public void noSerializationTest() {
 172         final CParameters capabilityNoAugmentation = new CParametersBuilder().removeAugmentation(CParameters1.class)
 173                 .build();
 174         final CParameters capabilityNoMP = new CParametersBuilder().addAugmentation(new CParameters1Builder().build())
 175                 .build();
 176
 177         final ByteBuf bytes = Unpooled.buffer();
 178         final AddPathCapabilityHandler handler = new AddPathCapabilityHandler(this.afiRegistry, this.safirException);
 179         handler.serializeCapability(capabilityNoAugmentation, bytes);
 180         assertEquals(0, bytes.readableBytes());
 181         handler.serializeCapability(capabilityNoMP, bytes);
 182         assertEquals(0, bytes.readableBytes());
 183     }
 184 }