pcep/auto-bandwidth-extension/src/test/java/org/opendaylight/protocol/pcep/auto/bandwidth/extension/PcRptMessageCodecTest.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.pcep.auto.bandwidth.extension;
   9
  10 import static org.junit.Assert.assertNotNull;
  11 import static org.junit.Assert.assertNull;
  12 import static org.junit.Assert.assertTrue;
  13
  14 import com.google.common.collect.Lists;
  15 import io.netty.buffer.ByteBuf;
  16 import io.netty.buffer.Unpooled;
  17 import java.util.ArrayList;
  18 import java.util.Collections;
  19 import java.util.List;
  20 import javax.xml.bind.DatatypeConverter;
  21 import org.junit.After;
  22 import org.junit.Assert;
  23 import org.junit.Before;
  24 import org.junit.Test;
  25 import org.opendaylight.protocol.pcep.ietf.stateful07.StatefulActivator;
  26 import org.opendaylight.protocol.pcep.parser.BaseParserExtensionActivator;
  27 import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
  28 import org.opendaylight.protocol.pcep.spi.pojo.SimplePCEPExtensionProviderContext;
  29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4AddressNoZone;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.auto.bandwidth.rev181109.Bandwidth1;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.auto.bandwidth.rev181109.Bandwidth1Builder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.auto.bandwidth.rev181109.bandwidth.usage.object.BandwidthUsage;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.auto.bandwidth.rev181109.bandwidth.usage.object.BandwidthUsageBuilder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.network.concepts.rev131125.Bandwidth;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.Pcrpt;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.PlspId;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.identifiers.tlv.LspIdentifiers;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.identifiers.tlv.LspIdentifiersBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.identifiers.tlv.lsp.identifiers.AddressFamily;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.identifiers.tlv.lsp.identifiers.address.family.Ipv4CaseBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.identifiers.tlv.lsp.identifiers.address.family.ipv4._case.Ipv4Builder;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.object.Lsp;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.object.LspBuilder;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.object.lsp.TlvsBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.pcrpt.message.pcrpt.message.Reports;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.Object;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.bandwidth.object.BandwidthBuilder;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.explicit.route.object.Ero;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.explicit.route.object.EroBuilder;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.Ipv4ExtendedTunnelId;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.LspId;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.TunnelId;
  53 import org.opendaylight.yangtools.yang.common.Uint16;
  54 import org.opendaylight.yangtools.yang.common.Uint32;
  55
  56 public class PcRptMessageCodecTest {
  57
  58     private static final List<Bandwidth> BW = Collections.singletonList(new Bandwidth(new byte[]{0, 0, 0, 1}));
  59     private SimplePCEPExtensionProviderContext ctx;
  60     private BaseParserExtensionActivator act;
  61     private StatefulActivator statefulAct;
  62     private org.opendaylight.protocol.pcep.auto.bandwidth.extension.Activator autoBwActivator;
  63
  64     @Before
  65     public void setUp() {
  66         this.ctx = new SimplePCEPExtensionProviderContext();
  67         this.act = new BaseParserExtensionActivator();
  68         this.act.start(this.ctx);
  69         this.statefulAct = new StatefulActivator();
  70         this.statefulAct.start(this.ctx);
  71         this.autoBwActivator = new org.opendaylight.protocol.pcep.auto.bandwidth.extension
  72                 .Activator(5);
  73         this.autoBwActivator.start(this.ctx);
  74     }
  75
  76     @After
  77     public void tearDown() {
  78         this.act.stop();
  79         this.statefulAct.stop();
  80         this.autoBwActivator.stop();
  81     }
  82
  83     @Test
  84     public void testGetValidReportsPositive() {
  85         final PcRptMessageCodec codec = new PcRptMessageCodec(this.ctx.getObjectHandlerRegistry());
  86         final BandwidthUsage bw = new BandwidthUsageBuilder().setBwSample(BW).build();
  87         final Ipv4Builder builder = new Ipv4Builder();
  88         builder.setIpv4TunnelSenderAddress(new Ipv4AddressNoZone("127.0.1.1"));
  89         builder.setIpv4ExtendedTunnelId(new Ipv4ExtendedTunnelId(new Ipv4AddressNoZone("127.0.1.2")));
  90         builder.setIpv4TunnelEndpointAddress(new Ipv4AddressNoZone("127.0.1.3"));
  91         final AddressFamily afiLsp = new Ipv4CaseBuilder().setIpv4(builder.build()).build();
  92         final LspId lspId = new LspId(Uint32.ONE);
  93         final TunnelId tunnelId = new TunnelId(Uint16.ONE);
  94         final LspIdentifiers identifier = new LspIdentifiersBuilder().setAddressFamily(afiLsp)
  95                 .setLspId(lspId).setTunnelId(tunnelId).build();
  96         final Lsp lsp = new LspBuilder().setPlspId(new PlspId(Uint32.ONE))
  97                 .setTlvs(new TlvsBuilder().setLspIdentifiers(identifier).build()).build();
  98         final Ero ero = new EroBuilder().build();
  99         final List<Object> objects = Lists.newArrayList(lsp, ero, bw);
 100         final Reports validReports = codec.getValidReports(objects, Collections.emptyList());
 101         assertNotNull(validReports.getPath().getBandwidth().augmentation(Bandwidth1.class));
 102         assertTrue(objects.isEmpty());
 103     }
 104
 105     @Test
 106     public void testGetValidReportsNegative() {
 107         final PcRptMessageCodec codec = new PcRptMessageCodec(this.ctx.getObjectHandlerRegistry());
 108         final BandwidthUsage bw = new BandwidthUsageBuilder().setBwSample(BW).build();
 109         final Ipv4Builder builder = new Ipv4Builder();
 110         builder.setIpv4TunnelSenderAddress(new Ipv4AddressNoZone("127.0.1.1"));
 111         builder.setIpv4ExtendedTunnelId(new Ipv4ExtendedTunnelId(new Ipv4AddressNoZone("127.0.1.2")));
 112         builder.setIpv4TunnelEndpointAddress(new Ipv4AddressNoZone("127.0.1.3"));
 113         final Lsp lsp = new LspBuilder().setPlspId(new PlspId(Uint32.ONE)).build();
 114         final Ero ero = new EroBuilder().build();
 115         final List<Object> objects = Lists.newArrayList(lsp, ero, bw);
 116         final Reports validReports = codec.getValidReports(objects, new ArrayList<>());
 117         assertNull(validReports);
 118     }
 119
 120     @Test
 121     public void testserializeObject() {
 122         final PcRptMessageCodec codec = new PcRptMessageCodec(this.ctx.getObjectHandlerRegistry());
 123         final BandwidthBuilder bwBuilder = new BandwidthBuilder();
 124         bwBuilder.addAugmentation(Bandwidth1.class, new Bandwidth1Builder().setBwSample(BW).build());
 125         final ByteBuf buffer = Unpooled.buffer();
 126         codec.serializeObject(bwBuilder.build(), buffer);
 127         Assert.assertTrue(buffer.readableBytes() > 0);
 128     }
 129
 130     @Test
 131     public void testReportMsgWithRro() throws PCEPDeserializerException {
 132         final byte[] parseHexBinary = DatatypeConverter
 133                 .parseHexBinary("2010003c0084a019001100106e79636e7932316372735f7432313231001200100a0"
 134                         + "000d2004008490a0000d40a0000d4001f0006000005dd700000000710001401080a000706200001080a0000d420"
 135                         + "000910001400000000000000000000000005050100051000084998968005500008513a43b70810002401080a000"
 136                         + "0d42020030801010000000001080a00070620000308010100000000");
 137         final Pcrpt msg = (Pcrpt) this.ctx.getMessageHandlerRegistry().parseMessage(10,
 138                 Unpooled.wrappedBuffer(parseHexBinary), Collections.emptyList());
 139         Assert.assertNotNull(msg.getPcrptMessage().getReports().get(0).getPath()
 140                 .getBandwidth().augmentation(Bandwidth1.class));
 141     }
 142
 143 }