bgp/linkstate/src/test/java/org/opendaylight/protocol/bgp/linkstate/LinkstateRIBSupportTest.java

   1 /*
   2  * Copyright (c) 2015 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.linkstate;
   9
  10 import static org.junit.Assert.assertEquals;
  11
  12 import java.math.BigInteger;
  13 import java.util.ArrayList;
  14 import java.util.List;
  15 import org.junit.Assert;
  16 import org.junit.Before;
  17 import org.junit.Test;
  18 import org.mockito.Mock;
  19 import org.mockito.Mockito;
  20 import org.mockito.MockitoAnnotations;
  21 import org.mockito.invocation.InvocationOnMock;
  22 import org.mockito.stubbing.Answer;
  23 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  24 import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
  25 import org.opendaylight.protocol.bgp.linkstate.nlri.LinkstateNlriParser;
  26 import org.opendaylight.protocol.bgp.linkstate.nlri.NodeNlriParser;
  27 import org.opendaylight.protocol.bgp.parser.BGPParsingException;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.LinkstateAddressFamily;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.LinkstateSubsequentAddressFamily;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.linkstate.destination.CLinkstateDestination;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.node.identifier.CRouterIdentifier;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.Attributes;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.update.attributes.MpReachNlri;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.update.attributes.MpUnreachNlri;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.tables.Routes;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.CNextHop;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.c.next.hop.Ipv4NextHopCaseBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.c.next.hop.ipv4.next.hop._case.Ipv4NextHopBuilder;
  40 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  41 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  42 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
  43 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  44 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  45 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  46 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListEntryNode;
  47 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListNode;
  48 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
  49 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeAttrBuilder;
  50 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeBuilder;
  51 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableContainerNodeBuilder;
  52 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableLeafNodeBuilder;
  53 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableUnkeyedListEntryNodeBuilder;
  54 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableUnkeyedListNodeBuilder;
  55
  56 public class LinkstateRIBSupportTest {
  57
  58     private static final Ipv4Address ipv4 = new Ipv4Address("42.42.42.42");
  59     private static final NodeIdentifier ROUTES_NODE_ID = new NodeIdentifier(Routes.QNAME);
  60     final LinkstateRIBSupport link = LinkstateRIBSupport.getInstance();
  61     final List<MapEntryNode> linkList = new ArrayList<>();
  62     private List<YangInstanceIdentifier> routes;
  63
  64     @Mock
  65     private DOMDataWriteTransaction tx;
  66
  67     @Before
  68     public void setUp() {
  69         MockitoAnnotations.initMocks(this);
  70         routes = new ArrayList<>();
  71         Mockito.doAnswer(new Answer<Object>() {
  72             @Override
  73             public Object answer(final InvocationOnMock invocation) throws Throwable {
  74                 final Object[] args = invocation.getArguments();
  75                 LinkstateRIBSupportTest.this.routes.add((YangInstanceIdentifier) args[1]);
  76                 return args[1];
  77             }
  78         }).when(this.tx).put(Mockito.any(LogicalDatastoreType.class), Mockito.any(YangInstanceIdentifier.class), Mockito.any(NormalizedNode.class));
  79
  80         Mockito.doAnswer(new Answer<Object>() {
  81             @Override
  82             public Object answer(final InvocationOnMock invocation) throws Throwable {
  83                 final Object[] args = invocation.getArguments();
  84                 LinkstateRIBSupportTest.this.routes.remove(args[1]);
  85                 return args[1];
  86             }
  87         }).when(this.tx).delete(Mockito.any(LogicalDatastoreType.class), Mockito.any(YangInstanceIdentifier.class));
  88     }
  89
  90     @Test
  91     public void testbuildReach() throws BGPParsingException {
  92         final CNextHop hop = new Ipv4NextHopCaseBuilder().setIpv4NextHop(new Ipv4NextHopBuilder().setGlobal(ipv4).build()).build();
  93         final MpReachNlri result = link.buildReach(linkList, hop);
  94         assertEquals(LinkstateAddressFamily.class, result.getAfi());
  95         assertEquals(LinkstateSubsequentAddressFamily.class, result.getSafi());
  96         assertEquals(new Ipv4NextHopCaseBuilder().setIpv4NextHop(new Ipv4NextHopBuilder().setGlobal(new Ipv4Address("42.42.42.42")).build()).build(), result.getCNextHop());
  97     }
  98
  99     @Test
 100     public void testBuildUnreach() {
 101         final MpUnreachNlri result = link.buildUnreach(linkList);
 102         assertEquals(LinkstateAddressFamily.class, result.getAfi());
 103         assertEquals(LinkstateSubsequentAddressFamily.class, result.getSafi());
 104     }
 105
 106     @Test
 107     public void testDestinationRoutes() {
 108         final YangInstanceIdentifier yangIdentifier = YangInstanceIdentifier.of(Routes.QNAME);
 109
 110         final DataContainerNodeAttrBuilder<NodeIdentifier, UnkeyedListEntryNode> linkstateBI = ImmutableUnkeyedListEntryNodeBuilder.create();
 111         linkstateBI.withNodeIdentifier(new NodeIdentifier(CLinkstateDestination.QNAME));
 112
 113         final ImmutableLeafNodeBuilder<String> protocolId = new ImmutableLeafNodeBuilder<>();
 114         protocolId.withNodeIdentifier(LinkstateNlriParser.PROTOCOL_ID_NID);
 115         protocolId.withValue("isis-level2");
 116         linkstateBI.addChild(protocolId.build());
 117
 118         final ImmutableLeafNodeBuilder<BigInteger> identifier = new ImmutableLeafNodeBuilder<>();
 119         identifier.withNodeIdentifier(LinkstateNlriParser.IDENTIFIER_NID);
 120         identifier.withValue(BigInteger.ONE);
 121         linkstateBI.addChild(identifier.build());
 122
 123         final DataContainerNodeBuilder<NodeIdentifier, ChoiceNode> objectType = Builders.choiceBuilder();
 124         objectType.withNodeIdentifier(LinkstateNlriParser.OBJECT_TYPE_NID);
 125
 126         final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> nodeDescriptors = Builders.containerBuilder();
 127         nodeDescriptors.withNodeIdentifier(LinkstateNlriParser.NODE_DESCRIPTORS_NID);
 128
 129         final ImmutableLeafNodeBuilder<Long> asNumber = new ImmutableLeafNodeBuilder<>();
 130         asNumber.withNodeIdentifier(NodeNlriParser.AS_NUMBER_NID);
 131         asNumber.withValue(72L);
 132         nodeDescriptors.addChild(asNumber.build());
 133
 134         final ImmutableLeafNodeBuilder<Long> areaID = new ImmutableLeafNodeBuilder<>();
 135         areaID.withNodeIdentifier(NodeNlriParser.AREA_NID);
 136         areaID.withValue(2697513L);
 137         nodeDescriptors.addChild(areaID.build());
 138
 139         final ImmutableLeafNodeBuilder<Long> domainID = new ImmutableLeafNodeBuilder<>();
 140         domainID.withNodeIdentifier(NodeNlriParser.DOMAIN_NID);
 141         domainID.withValue(0x28282828L);
 142         nodeDescriptors.addChild(domainID.build());
 143
 144         final DataContainerNodeBuilder<NodeIdentifier, ChoiceNode> crouterId = Builders.choiceBuilder();
 145         crouterId.withNodeIdentifier(new NodeIdentifier(CRouterIdentifier.QNAME));
 146
 147         final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> isisNode = Builders.containerBuilder();
 148         isisNode.withNodeIdentifier(NodeNlriParser.ISIS_PSEUDONODE_NID);
 149
 150         final ImmutableLeafNodeBuilder<byte[]> isoSystemID = new ImmutableLeafNodeBuilder<>();
 151         isoSystemID.withNodeIdentifier(NodeNlriParser.ISO_SYSTEM_NID);
 152         isoSystemID.withValue(new byte[]{0, 0, 0, 0, 0, (byte) 0x39});
 153
 154         final DataContainerNodeAttrBuilder<NodeIdentifier, ContainerNode> isisRouter = Builders.containerBuilder();
 155         isisRouter.withNodeIdentifier(NodeNlriParser.ISIS_ROUTER_NID);
 156         isisRouter.addChild(isoSystemID.build());
 157
 158         isisNode.addChild(isisRouter.build());
 159         isisNode.addChild(Builders.leafBuilder().withNodeIdentifier(NodeNlriParser.PSN_NID).withValue((short) 5).build());
 160         crouterId.addChild(isisNode.build());
 161
 162         nodeDescriptors.addChild(crouterId.build());
 163         objectType.addChild(nodeDescriptors.build());
 164         linkstateBI.addChild(objectType.build());
 165
 166         final UnkeyedListNode routes = ImmutableUnkeyedListNodeBuilder.create()
 167             .withNodeIdentifier(new NodeIdentifier(CLinkstateDestination.QNAME))
 168             .addChild(linkstateBI.build())
 169             .build();
 170
 171         final ContainerNode destination = ImmutableContainerNodeBuilder.create().
 172             addChild(routes)
 173             .withNodeIdentifier(new NodeIdentifier(CLinkstateDestination.QNAME))
 174             .build();
 175
 176         final ContainerNode attributes = ImmutableContainerNodeBuilder.create()
 177             .withNodeIdentifier(new NodeIdentifier(Attributes.QNAME))
 178             .build();
 179
 180         link.putDestinationRoutes(tx, yangIdentifier, destination, attributes, ROUTES_NODE_ID);
 181
 182         Assert.assertEquals(1, this.routes.size());
 183
 184         link.deleteDestinationRoutes(tx, yangIdentifier, destination, ROUTES_NODE_ID);
 185
 186         Assert.assertEquals(0, this.routes.size());
 187     }
 188
 189 }