bgp/extensions/evpn/src/test/java/org/opendaylight/protocol/bgp/evpn/spi/pojo/SimpleEvpnNlriRegistryTest.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.evpn.spi.pojo;
   9
  10 import static org.junit.Assert.assertArrayEquals;
  11 import static org.junit.Assert.assertEquals;
  12 import static org.junit.Assert.assertNull;
  13 import static org.opendaylight.protocol.bgp.evpn.impl.EvpnTestUtil.VALUE_SIZE;
  14 import static org.opendaylight.protocol.bgp.evpn.impl.EvpnTestUtil.createContBuilder;
  15 import static org.opendaylight.protocol.bgp.evpn.impl.nlri.EthADRParserTest.ETHERNET_AD_ROUTE_CASE;
  16 import static org.opendaylight.protocol.bgp.evpn.impl.nlri.EthADRParserTest.ROUDE_DISTIN;
  17 import static org.opendaylight.protocol.bgp.evpn.impl.nlri.EthADRParserTest.createEthADRModel;
  18
  19 import io.netty.buffer.ByteBuf;
  20 import io.netty.buffer.Unpooled;
  21 import java.util.ArrayList;
  22 import org.junit.Before;
  23 import org.junit.Test;
  24 import org.opendaylight.protocol.bgp.evpn.impl.esi.types.ESIActivator;
  25 import org.opendaylight.protocol.bgp.evpn.impl.nlri.EthADRParserTest;
  26 import org.opendaylight.protocol.bgp.evpn.impl.nlri.NlriActivator;
  27 import org.opendaylight.protocol.util.ByteArray;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.Evpn;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.NlriType;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.evpn.EvpnChoice;
  31 import org.opendaylight.yangtools.concepts.Registration;
  32 import org.opendaylight.yangtools.yang.binding.DataContainer;
  33 import org.opendaylight.yangtools.yang.common.QName;
  34 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  35 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
  36 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
  37 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeBuilder;
  38
  39 public final class SimpleEvpnNlriRegistryTest {
  40     public static final NodeIdentifier EVPN_NID = new NodeIdentifier(EvpnChoice.QNAME);
  41
  42     private class NotRegistered implements EvpnChoice {
  43         @Override
  44         public Class<? extends DataContainer> getImplementedInterface() {
  45             return NotRegistered.class;
  46         }
  47     }
  48
  49     @Before
  50     public void setUp() {
  51         final ArrayList<Registration> reg = new ArrayList<>();
  52         ESIActivator.registerEsiTypeParsers(reg);
  53         NlriActivator.registerNlriParsers(reg);
  54     }
  55
  56     @Test
  57     public void registryTest() {
  58         final ByteBuf buff = SimpleEvpnNlriRegistry.getInstance().serializeEvpn(ETHERNET_AD_ROUTE_CASE,
  59                 Unpooled.wrappedBuffer(ROUDE_DISTIN));
  60         assertArrayEquals(EthADRParserTest.RESULT, ByteArray.getAllBytes(buff));
  61         final EvpnChoice resultModel = SimpleEvpnNlriRegistry.getInstance().serializeEvpnModel(createEthADRModel());
  62         assertEquals(ETHERNET_AD_ROUTE_CASE, resultModel);
  63         final NlriType type = NlriType.forValue(buff.readUnsignedByte());
  64         buff.skipBytes(VALUE_SIZE); // length + RD
  65         assertEquals(ETHERNET_AD_ROUTE_CASE, SimpleEvpnNlriRegistry.getInstance().parseEvpn(type, buff));
  66     }
  67
  68     @Test(expected = IllegalArgumentException.class)
  69     public void registryParseTest() {
  70         SimpleEvpnNlriRegistry.getInstance().parseEvpn(NlriType.EthADDisc, null);
  71     }
  72
  73     @Test
  74     public void registryNullTest() {
  75         final ByteBuf body = Unpooled.buffer();
  76         SimpleEvpnNlriRegistry.getInstance().serializeEvpn(new NotRegistered(), body);
  77         assertEquals(0, body.readableBytes());
  78     }
  79
  80     @Test
  81     public void registryNullModelTest() {
  82         final DataContainerNodeBuilder<NodeIdentifier, ChoiceNode> choice = Builders.choiceBuilder()
  83                 .withNodeIdentifier(EVPN_NID);
  84         choice.addChild(createContBuilder(new NodeIdentifier(QName.create(Evpn.QNAME, "test").intern()))
  85                 .build()).build();
  86         assertNull(SimpleEvpnNlriRegistry.getInstance().serializeEvpnModel(choice.build()));
  87     }
  88 }