pcep/ietf-stateful02/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
  12 import java.io.IOException;
  13
  14 import org.junit.Before;
  15 import org.junit.Test;
  16 import org.opendaylight.protocol.pcep.ietf.stateful02.Stateful02OpenObjectParser;
  17 import org.opendaylight.protocol.pcep.ietf.stateful02.Stateful02LspaObjectParser;
  18 import org.opendaylight.protocol.pcep.spi.ObjectHeaderImpl;
  19 import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
  20 import org.opendaylight.protocol.pcep.spi.TlvHandlerRegistry;
  21 import org.opendaylight.protocol.pcep.spi.pojo.ServiceLoaderPCEPExtensionProviderContext;
  22 import org.opendaylight.protocol.util.ByteArray;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.Stateful1;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.Stateful1Builder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.Tlvs1;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.Tlvs1Builder;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.lsp.cleanup.tlv.LspCleanupBuilder;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.Tlvs2;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.Tlvs2Builder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.stateful.capability.tlv.Stateful;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.stateful.capability.tlv.StatefulBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.symbolic.path.name.tlv.SymbolicPathName;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.symbolic.path.name.tlv.SymbolicPathNameBuilder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.ProtocolVersion;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lspa.object.LspaBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.OpenBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.AttributeFilter;
  38
  39 public class PCEPObjectParserTest {
  40
  41         private TlvHandlerRegistry tlvRegistry;
  42
  43         @Before
  44         public void setUp() throws Exception {
  45                 this.tlvRegistry = ServiceLoaderPCEPExtensionProviderContext.create().getTlvHandlerRegistry();
  46         }
  47
  48         @Test
  49         public void testOpenObjectWithTLV() throws PCEPDeserializerException, IOException {
  50                 final Stateful02OpenObjectParser parser = new Stateful02OpenObjectParser(this.tlvRegistry);
  51                 final byte[] result = ByteArray.fileToBytes("src/test/resources/PCEPOpenObject1.bin");
  52
  53                 final OpenBuilder builder = new OpenBuilder();
  54                 builder.setProcessingRule(false);
  55                 builder.setIgnore(false);
  56                 builder.setVersion(new ProtocolVersion((short) 1));
  57                 builder.setKeepalive((short) 30);
  58                 builder.setDeadTimer((short) 120);
  59                 builder.setSessionId((short) 1);
  60
  61                 final Stateful tlv1 = new StatefulBuilder().setLspUpdateCapability(Boolean.TRUE).setIncludeDbVersion(Boolean.FALSE).addAugmentation(
  62                                 Stateful1.class, new Stateful1Builder().setInitiation(true).build()).build();
  63
  64                 final Tlvs2Builder statBuilder = new Tlvs2Builder();
  65                 statBuilder.setStateful(tlv1);
  66
  67                 final Tlvs1Builder cleanupBuilder = new Tlvs1Builder();
  68                 cleanupBuilder.setLspCleanup(new LspCleanupBuilder().setTimeout(180L).build());
  69
  70                 builder.setTlvs(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.open.TlvsBuilder().addAugmentation(
  71                                 Tlvs2.class, statBuilder.build()).addAugmentation(Tlvs1.class, cleanupBuilder.build()).build());
  72
  73                 // FIXME: enable once the registry is rewritten
  74                 // assertEquals(builder.build(), parser.parseObject(new ObjectHeaderImpl(false, false),
  75                 // ByteArray.cutBytes(result, 4)));
  76                 // assertArrayEquals(result, parser.serializeObject(builder.build()));
  77         }
  78
  79         @Test
  80         public void testLspaObjectWithTlv() throws IOException, PCEPDeserializerException {
  81                 final Stateful02LspaObjectParser parser = new Stateful02LspaObjectParser(this.tlvRegistry);
  82                 final byte[] result = ByteArray.fileToBytes("src/test/resources/PCEPLspaObject1LowerBounds.bin");
  83
  84                 final LspaBuilder builder = new LspaBuilder();
  85                 builder.setIgnore(false);
  86                 builder.setProcessingRule(false);
  87                 builder.setIncludeAny(new AttributeFilter(0l));
  88                 builder.setExcludeAny(new AttributeFilter(0l));
  89                 builder.setIncludeAll(new AttributeFilter(0l));
  90                 builder.setSetupPriority((short) 0);
  91                 builder.setHoldPriority((short) 0);
  92                 builder.setLocalProtectionDesired(false);
  93
  94                 final SymbolicPathName tlv = new SymbolicPathNameBuilder().setPathName(
  95                                 new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.SymbolicPathName(new byte[] {
  96                                                 (byte) 0x4d, (byte) 0x65, (byte) 0x64, (byte) 0x20, (byte) 0x74, (byte) 0x65, (byte) 0x73, (byte) 0x74,
  97                                                 (byte) 0x20, (byte) 0x6f, (byte) 0x66, (byte) 0x20, (byte) 0x73, (byte) 0x79, (byte) 0x6d, (byte) 0x62,
  98                                                 (byte) 0x6f, (byte) 0x6c, (byte) 0x69, (byte) 0x63, (byte) 0x20, (byte) 0x6e, (byte) 0x61, (byte) 0x6d, (byte) 0x65 })).build();
  99
 100                 builder.setTlvs(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lspa.object.lspa.TlvsBuilder().addAugmentation(
 101                                 org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.Tlvs2.class,
 102                                 new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.Tlvs2Builder().setSymbolicPathName(
 103                                                 tlv).build()).build());
 104                 // Tlvs container does not contain toString
 105                 // assertEquals(builder.build(), parser.parseObject(new ObjectHeaderImpl(false, false),
 106                 // ByteArray.cutBytes(result, 4)));
 107                 assertArrayEquals(result, parser.serializeObject(builder.build()));
 108         }
 109
 110         @Test
 111         public void testLspObjectWithTLV() throws IOException, PCEPDeserializerException {
 112                 // final PCEPLspObjectParser parser = new PCEPLspObjectParser(this.tlvRegistry);
 113                 // final byte[] result = ByteArray.fileToBytes("src/test/resources/PCEPLspObject1WithTLV.bin");
 114                 //
 115                 // final LspBuilder builder = new LspBuilder();
 116                 // builder.setProcessingRule(true);
 117                 // builder.setIgnore(true);
 118                 // builder.setDelegate(false);
 119                 // builder.setRemove(true);
 120                 // builder.setSync(false);
 121                 //
 122                 // final LspErrorCode tlv1 = new LspErrorCodeBuilder().setErrorCode(627610883L).build();
 123                 // final SymbolicPathName tlv2 = new SymbolicPathNameBuilder().setPathName(
 124                 // new
 125                 // org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.SymbolicPathName("Med".getBytes())).build();
 126                 // builder.setTlvs(new
 127                 // org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.lsp.TlvsBuilder().setLspErrorCode(
 128                 // tlv1).setSymbolicPathName(tlv2).build());
 129                 // assertEquals(builder.build(), parser.parseObject(new ObjectHeaderImpl(true, true), ByteArray.cutBytes(result,
 130                 // 4)));
 131                 // assertArrayEquals(result, parser.serializeObject(builder.build()));
 132         }
 133 }