pcep/ietf-stateful07/src/test/java/org/opendaylight/protocol/pcep/ietf/PCEPObjectParserTest.java

   1 /*
   2  * Copyright (c) 2013 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.ietf;
   9
  10 import static org.junit.Assert.assertArrayEquals;
  11 import static org.junit.Assert.assertEquals;
  12 import io.netty.buffer.ByteBuf;
  13 import io.netty.buffer.Unpooled;
  14
  15 import java.io.IOException;
  16
  17 import org.junit.Before;
  18 import org.junit.Test;
  19 import org.opendaylight.protocol.pcep.ietf.initiated00.CInitiated00LspObjectParser;
  20 import org.opendaylight.protocol.pcep.ietf.initiated00.CInitiated00SrpObjectParser;
  21 import org.opendaylight.protocol.pcep.ietf.stateful07.Stateful07LspaObjectParser;
  22 import org.opendaylight.protocol.pcep.ietf.stateful07.Stateful07OpenObjectParser;
  23 import org.opendaylight.protocol.pcep.spi.ObjectHeaderImpl;
  24 import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
  25 import org.opendaylight.protocol.pcep.spi.TlvRegistry;
  26 import org.opendaylight.protocol.pcep.spi.pojo.ServiceLoaderPCEPExtensionProviderContext;
  27 import org.opendaylight.protocol.util.ByteArray;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.Lsp1;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.Lsp1Builder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.Srp1;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.Srp1Builder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.OperationalStatus;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.PlspId;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.SrpIdNumber;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Tlvs1;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Tlvs1Builder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Tlvs2;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Tlvs2Builder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.error.code.tlv.LspErrorCode;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.error.code.tlv.LspErrorCodeBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.LspBuilder;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.SrpBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.stateful.capability.tlv.Stateful;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.stateful.capability.tlv.StatefulBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.symbolic.path.name.tlv.SymbolicPathName;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.symbolic.path.name.tlv.SymbolicPathNameBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.ProtocolVersion;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lspa.object.Lspa;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lspa.object.LspaBuilder;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.OpenBuilder;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.AttributeFilter;
  52
  53 public class PCEPObjectParserTest {
  54
  55     private TlvRegistry tlvRegistry;
  56
  57     @Before
  58     public void setUp() throws Exception {
  59         this.tlvRegistry = ServiceLoaderPCEPExtensionProviderContext.create().getTlvHandlerRegistry();
  60     }
  61
  62     @Test
  63     public void testOpenObjectWithTLV() throws PCEPDeserializerException, IOException {
  64         final Stateful07OpenObjectParser parser = new Stateful07OpenObjectParser(this.tlvRegistry);
  65         final ByteBuf result = Unpooled.wrappedBuffer(ByteArray.fileToBytes("src/test/resources/PCEPOpenObject1.bin"));
  66
  67         final OpenBuilder builder = new OpenBuilder();
  68         builder.setProcessingRule(false);
  69         builder.setIgnore(false);
  70         builder.setVersion(new ProtocolVersion((short) 1));
  71         builder.setKeepalive((short) 30);
  72         builder.setDeadTimer((short) 120);
  73         builder.setSessionId((short) 1);
  74
  75         final Stateful tlv1 = new StatefulBuilder().setLspUpdateCapability(Boolean.TRUE).build();
  76
  77         final Tlvs1Builder statBuilder = new Tlvs1Builder();
  78         statBuilder.setStateful(tlv1);
  79
  80         builder.setTlvs(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.open.TlvsBuilder().addAugmentation(
  81                 Tlvs1.class, statBuilder.build()).build());
  82
  83         assertEquals(builder.build(), parser.parseObject(new ObjectHeaderImpl(false, false), result.slice(4, result.readableBytes() - 4)));
  84         assertArrayEquals(result.array(), parser.serializeObject(builder.build()));
  85     }
  86
  87     @Test
  88     public void testLspObjectWithTLV() throws IOException, PCEPDeserializerException {
  89         final CInitiated00LspObjectParser parser = new CInitiated00LspObjectParser(this.tlvRegistry);
  90         final ByteBuf result = Unpooled.wrappedBuffer(ByteArray.fileToBytes("src/test/resources/PCEPLspObject1WithTLV.bin"));
  91
  92         final LspBuilder builder = new LspBuilder();
  93         builder.setProcessingRule(true);
  94         builder.setIgnore(true);
  95         builder.setAdministrative(true);
  96         builder.setDelegate(false);
  97         builder.setRemove(true);
  98         builder.setSync(false);
  99         builder.addAugmentation(Lsp1.class, new Lsp1Builder().setCreate(false).build());
 100         builder.setOperational(OperationalStatus.GoingDown);
 101         builder.setPlspId(new PlspId(0x12345L));
 102
 103         final LspErrorCode tlv1 = new LspErrorCodeBuilder().setErrorCode(627610883L).build();
 104         final SymbolicPathName tlv2 = new SymbolicPathNameBuilder().setPathName(
 105                 new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.SymbolicPathName("Med".getBytes())).build();
 106         builder.setTlvs(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.lsp.TlvsBuilder().setLspErrorCode(
 107                 tlv1).setSymbolicPathName(tlv2).build());
 108         assertEquals(builder.build(), parser.parseObject(new ObjectHeaderImpl(true, true), result.slice(4, result.readableBytes() - 4)));
 109         assertArrayEquals(result.array(), parser.serializeObject(builder.build()));
 110     }
 111
 112     @Test
 113     public void testLspaObject() throws IOException, PCEPDeserializerException {
 114         final Stateful07LspaObjectParser parser = new Stateful07LspaObjectParser(this.tlvRegistry);
 115         final LspaBuilder builder = new LspaBuilder();
 116         final ByteBuf result = Unpooled.wrappedBuffer(ByteArray.fileToBytes("src/test/resources/PCEPLspaObject3RandVals.bin"));
 117
 118         final SymbolicPathName tlv = new SymbolicPathNameBuilder().setPathName(
 119                 new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.SymbolicPathName(new byte[] {
 120                     (byte) 0x4d, (byte) 0x65, (byte) 0x64, (byte) 0x20, (byte) 0x74, (byte) 0x65, (byte) 0x73, (byte) 0x74, (byte) 0x20,
 121                     (byte) 0x6f, (byte) 0x66, (byte) 0x20, (byte) 0x73, (byte) 0x79, (byte) 0x6d, (byte) 0x62, (byte) 0x6f, (byte) 0x6c,
 122                     (byte) 0x69, (byte) 0x63, (byte) 0x20, (byte) 0x6e, (byte) 0x61, (byte) 0x6d, (byte) 0x65 })).build();
 123
 124         builder.setIgnore(false);
 125         builder.setProcessingRule(false);
 126         builder.setExcludeAny(new AttributeFilter(0x20A1FEE3L));
 127         builder.setIncludeAny(new AttributeFilter(0x1A025CC7L));
 128         builder.setIncludeAll(new AttributeFilter(0x2BB66532L));
 129         builder.setHoldPriority((short) 0x02);
 130         builder.setSetupPriority((short) 0x03);
 131         builder.setLocalProtectionDesired(true);
 132         builder.setTlvs(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lspa.object.lspa.TlvsBuilder().addAugmentation(
 133                 Tlvs2.class, new Tlvs2Builder().setSymbolicPathName(tlv).build()).build());
 134
 135         // Tlvs container does not contain toString
 136         final Object o = parser.parseObject(new ObjectHeaderImpl(true, true), result.slice(4, result.readableBytes() - 4));
 137         assertEquals(tlv, ((Lspa) o).getTlvs().getAugmentation(Tlvs2.class).getSymbolicPathName());
 138         // assertEquals(builder.build(), parser.parseObject(new ObjectHeaderImpl(true, true), ByteArray.cutBytes(result,
 139         // 4)));
 140         assertArrayEquals(result.array(), parser.serializeObject(builder.build()));
 141     }
 142
 143     @Test
 144     public void testSrpObject() throws IOException, PCEPDeserializerException {
 145         final CInitiated00SrpObjectParser parser = new CInitiated00SrpObjectParser(this.tlvRegistry);
 146         final ByteBuf result = Unpooled.wrappedBuffer(new byte[] { (byte) 0x21, (byte) 0x10, (byte) 0x00, (byte) 0x0c, 0, 0, 0,
 147             (byte) 0x01, 0, 0, 0, (byte) 0x01 });
 148
 149         final SrpBuilder builder = new SrpBuilder();
 150         builder.setProcessingRule(false);
 151         builder.setIgnore(false);
 152         builder.setOperationId(new SrpIdNumber(1L));
 153         builder.addAugmentation(Srp1.class, new Srp1Builder().setRemove(true).build());
 154
 155         assertEquals(builder.build(), parser.parseObject(new ObjectHeaderImpl(false, false), result.slice(4, result.readableBytes() - 4)));
 156         assertArrayEquals(result.array(), parser.serializeObject(builder.build()));
 157     }
 158 }