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