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