summaryrefslogtreecommitdiffstats
path: root/bgp/bgp-prefix-sid/src/test
diff options
context:
space:
mode:
Diffstat (limited to 'bgp/bgp-prefix-sid/src/test')
-rw-r--r--bgp/bgp-prefix-sid/src/test/java/io/fd/hc2vpp/bgp/prefix/sid/BgpPrefixSidMplsWriterTest.java165
1 files changed, 165 insertions, 0 deletions
diff --git a/bgp/bgp-prefix-sid/src/test/java/io/fd/hc2vpp/bgp/prefix/sid/BgpPrefixSidMplsWriterTest.java b/bgp/bgp-prefix-sid/src/test/java/io/fd/hc2vpp/bgp/prefix/sid/BgpPrefixSidMplsWriterTest.java
new file mode 100644
index 000000000..36df877c1
--- /dev/null
+++ b/bgp/bgp-prefix-sid/src/test/java/io/fd/hc2vpp/bgp/prefix/sid/BgpPrefixSidMplsWriterTest.java
@@ -0,0 +1,165 @@
+/*
+ * Copyright (c) 2017 Cisco and/or its affiliates.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.fd.hc2vpp.bgp.prefix.sid;
+
+import static io.fd.hc2vpp.bgp.prefix.sid.BgpPrefixSidMplsWriter.MPLS_LABEL_INVALID;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+import static org.mockito.MockitoAnnotations.initMocks;
+
+import com.google.common.collect.Lists;
+import io.fd.hc2vpp.common.test.util.FutureProducer;
+import io.fd.hc2vpp.common.translate.util.ByteDataTranslator;
+import io.fd.honeycomb.translate.write.WriteFailedException;
+import io.fd.vpp.jvpp.core.dto.MplsRouteAddDel;
+import io.fd.vpp.jvpp.core.dto.MplsRouteAddDelReply;
+import io.fd.vpp.jvpp.core.future.FutureJVppCore;
+import java.util.Collections;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4AddressNoZone;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev150525.LabeledUnicastRoutes;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev150525.Srgb;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev150525.labeled.unicast.LabelStackBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev150525.labeled.unicast.routes.list.LabeledUnicastRoute;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev150525.labeled.unicast.routes.list.LabeledUnicastRouteBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev150525.labeled.unicast.routes.list.LabeledUnicastRouteKey;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev150525.labeled.unicast.routes.list.labeled.unicast.route.attributes.bgp.prefix.sid.bgp.prefix.sid.tlvs.bgp.prefix.sid.tlv.LuOriginatorSrgbTlvBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev150525.originator.srgb.tlv.SrgbValueBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.labeled.unicast.rev150525.update.attributes.bgp.prefix.sid.bgp.prefix.sid.tlvs.bgp.prefix.sid.tlv.LuLabelIndexTlvBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.PathId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.AttributesBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.attributes.BgpPrefixSid;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.attributes.BgpPrefixSidBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.attributes.bgp.prefix.sid.BgpPrefixSidTlvs;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.attributes.bgp.prefix.sid.BgpPrefixSidTlvsBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.BgpRib;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.RibId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.Rib;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.RibKey;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.rib.LocRib;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.Tables;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.c.next.hop.Ipv4NextHopCase;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.c.next.hop.Ipv4NextHopCaseBuilder;
+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;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.network.concepts.rev131125.MplsLabel;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+
+public class BgpPrefixSidMplsWriterTest implements FutureProducer, ByteDataTranslator {
+ private static final InstanceIdentifier<Tables> TABLE_ID = InstanceIdentifier.create(BgpRib.class)
+ .child(Rib.class, new RibKey(new RibId("test-rib"))).child(LocRib.class)
+ .child(Tables.class, new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class));
+
+ @Mock
+ private FutureJVppCore vppApi;
+ private BgpPrefixSidMplsWriter writer;
+
+ @SuppressWarnings("unchecked")
+ private static InstanceIdentifier<LabeledUnicastRoute> id(final PathId pathId, final String routeKey) {
+ return TABLE_ID.child((Class) LabeledUnicastRoutes.class)
+ .child(LabeledUnicastRoute.class, new LabeledUnicastRouteKey(pathId, routeKey));
+ }
+
+ private static LabeledUnicastRoute route(final String routeKey, final PathId pathId,
+ final Ipv4Address nextHopAddress,
+ final BgpPrefixSid bgpPrefixSid) {
+ final Ipv4NextHopCase nextHop =
+ new Ipv4NextHopCaseBuilder().setIpv4NextHop(new Ipv4NextHopBuilder().setGlobal(nextHopAddress).build())
+ .build();
+ return new LabeledUnicastRouteBuilder()
+ .setKey(new LabeledUnicastRouteKey(pathId, routeKey))
+ .setPathId(pathId)
+ .setAttributes(new AttributesBuilder()
+ .setCNextHop(nextHop)
+ .setBgpPrefixSid(bgpPrefixSid)
+ .build())
+ .setLabelStack(
+ Collections.singletonList(new LabelStackBuilder().setLabelValue(new MplsLabel(16101L)).build()))
+ .build();
+ }
+
+ private static BgpPrefixSidTlvs labelIndexTlv(final long label) {
+ return new BgpPrefixSidTlvsBuilder()
+ .setBgpPrefixSidTlv(new LuLabelIndexTlvBuilder()
+ .setLabelIndexTlv(label)
+ .build())
+ .build();
+ }
+
+ @Before
+ public void setUp() {
+ initMocks(this);
+ writer = new BgpPrefixSidMplsWriter(vppApi);
+ when(vppApi.mplsRouteAddDel(any())).thenReturn(future(new MplsRouteAddDelReply()));
+ }
+
+ private BgpPrefixSidTlvs originatorSrgbTlv(final long base, final long range) {
+ return new BgpPrefixSidTlvsBuilder()
+ .setBgpPrefixSidTlv(new LuOriginatorSrgbTlvBuilder()
+ .setSrgbValue(Collections.singletonList(new SrgbValueBuilder()
+ .setBase(new Srgb(base))
+ .setRange(new Srgb(range))
+ .build()))
+ .build())
+ .build();
+ }
+
+ @Test
+ public void testCreate() throws WriteFailedException.CreateFailedException {
+ final String routeKey = "route-key";
+ final PathId pathId = new PathId(123L);
+ final Ipv4Address nextHopAddress = new Ipv4AddressNoZone("5.6.7.8");
+
+ final BgpPrefixSid bgpPrefixSid = new BgpPrefixSidBuilder()
+ .setBgpPrefixSidTlvs(
+ Lists.newArrayList(
+ labelIndexTlv(102L),
+ originatorSrgbTlv(16000, 800)
+ ))
+ .build();
+ writer.create(
+ id(pathId, routeKey),
+ route(routeKey, pathId, nextHopAddress, bgpPrefixSid)
+ );
+ verifyRequest(true);
+ }
+
+ private void verifyRequest(boolean isAdd) {
+ final MplsRouteAddDel request = new MplsRouteAddDel();
+ request.mrIsAdd = booleanToByte(isAdd);
+ request.mrClassifyTableIndex = -1;
+ request.mrNextHopWeight = 1;
+ request.mrNextHopViaLabel = MPLS_LABEL_INVALID;
+
+ request.mrNextHop = new byte[] {5, 6, 7, 8};
+ request.mrNextHopSwIfIndex = -1;
+
+ request.mrLabel = 16102;
+
+ request.mrNextHopOutLabelStack = new int[] {16101};
+ request.mrNextHopNOutLabels = 1;
+
+ request.mrEos = 1;
+ verify(vppApi).mplsRouteAddDel(request);
+ }
+}