34f1c67c9330d606027203d9f85a22c1f4fd573f
[bgpcep.git] / pcep / impl / src / test / java / org / opendaylight / protocol / pcep / impl / LabelSubobjectParserTest.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.impl;
9
10 import static org.junit.Assert.assertArrayEquals;
11 import static org.junit.Assert.assertEquals;
12 import static org.junit.Assert.fail;
13
14 import io.netty.buffer.ByteBuf;
15 import io.netty.buffer.Unpooled;
16 import org.junit.Test;
17 import org.opendaylight.protocol.pcep.parser.subobject.GeneralizedLabelParser;
18 import org.opendaylight.protocol.pcep.parser.subobject.Type1LabelParser;
19 import org.opendaylight.protocol.pcep.parser.subobject.WavebandSwitchingLabelParser;
20 import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
21 import org.opendaylight.protocol.util.ByteArray;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.label.subobject.label.type.GeneralizedLabelCaseBuilder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.label.subobject.label.type.Type1LabelCaseBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.label.subobject.label.type.WavebandSwitchingLabelCaseBuilder;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.label.subobject.label.type.generalized.label._case.GeneralizedLabelBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.label.subobject.label.type.type1.label._case.Type1LabelBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.label.subobject.label.type.waveband.switching.label._case.WavebandSwitchingLabelBuilder;
28
29 public class LabelSubobjectParserTest {
30
31     private static final byte[] GENERALIZED_LABEL_BYTES = {
32         (byte) 0x80, 0x02, 0x00, 0x04, 0x12, 0x00, 0x25, (byte) 0xFF
33     };
34
35     private static final byte[] TYPE_ONE_LABEL_BYTES = { (byte) 0x81, 0x01, 0x12, 0x00, 0x25, (byte) 0xFF };
36
37     private static final byte[] WAVEBAND_LABEL_BYTES = {
38         0x01, 0x03, 0x00, 0x00, 0x12, 0x34, 0x00, 0x00, (byte) 0x99, (byte) 0x99, 0x00, 0x00, 0x11, 0x11
39     };
40
41     @Test
42     public void testGeneralizedLabel() throws PCEPDeserializerException {
43         final GeneralizedLabelParser parser = new GeneralizedLabelParser();
44         final GeneralizedLabelBuilder iBuilder = new GeneralizedLabelBuilder();
45         iBuilder.setGeneralizedLabel(ByteArray.cutBytes(GENERALIZED_LABEL_BYTES, 2));
46         final GeneralizedLabelCaseBuilder builder =
47             new GeneralizedLabelCaseBuilder().setGeneralizedLabel(iBuilder.build());
48         assertEquals(
49             builder.build(), parser.parseLabel(Unpooled.wrappedBuffer(ByteArray.cutBytes(GENERALIZED_LABEL_BYTES, 2))));
50         final ByteBuf buff = Unpooled.buffer();
51         parser.serializeLabel(true, false, builder.build(), buff);
52         assertArrayEquals(GENERALIZED_LABEL_BYTES, ByteArray.getAllBytes(buff));
53
54         try {
55             parser.parseLabel(null);
56             fail();
57         } catch (final IllegalArgumentException e) {
58             assertEquals("Array of bytes is mandatory. Can't be null or empty.", e.getMessage());
59         }
60
61         try {
62             parser.parseLabel(Unpooled.EMPTY_BUFFER);
63             fail();
64         } catch (final IllegalArgumentException e) {
65             assertEquals("Array of bytes is mandatory. Can't be null or empty.", e.getMessage());
66         }
67     }
68
69     @Test
70     public void testWavebandLabel() throws PCEPDeserializerException {
71         final WavebandSwitchingLabelParser parser = new WavebandSwitchingLabelParser();
72         final WavebandSwitchingLabelBuilder iBuilder = new WavebandSwitchingLabelBuilder();
73         iBuilder.setWavebandId(0x1234L);
74         iBuilder.setStartLabel(0x9999L);
75         iBuilder.setEndLabel(0x1111L);
76         final WavebandSwitchingLabelCaseBuilder builder =
77             new WavebandSwitchingLabelCaseBuilder().setWavebandSwitchingLabel(iBuilder.build());
78         assertEquals(
79             builder.build(), parser.parseLabel(Unpooled.wrappedBuffer(ByteArray.cutBytes(WAVEBAND_LABEL_BYTES, 2))));
80         final ByteBuf buff = Unpooled.buffer();
81         parser.serializeLabel(false, true, builder.build(), buff);
82         assertArrayEquals(WAVEBAND_LABEL_BYTES, ByteArray.getAllBytes(buff));
83
84         try {
85             parser.parseLabel(null);
86             fail();
87         } catch (final IllegalArgumentException e) {
88             assertEquals("Array of bytes is mandatory. Cannot be null or empty.", e.getMessage());
89         }
90         try {
91             parser.parseLabel(Unpooled.EMPTY_BUFFER);
92             fail();
93         } catch (final IllegalArgumentException e) {
94             assertEquals("Array of bytes is mandatory. Cannot be null or empty.", e.getMessage());
95         }
96     }
97
98     @Test
99     public void testTypeOneLabel() throws PCEPDeserializerException {
100         final Type1LabelParser parser = new Type1LabelParser();
101         final Type1LabelBuilder iBuilder = new Type1LabelBuilder();
102         iBuilder.setType1Label(0x120025ffL);
103         final Type1LabelCaseBuilder builder = new Type1LabelCaseBuilder().setType1Label(iBuilder.build());
104         assertEquals(
105             builder.build(), parser.parseLabel(Unpooled.wrappedBuffer(ByteArray.cutBytes(TYPE_ONE_LABEL_BYTES, 2))));
106         final ByteBuf buff = Unpooled.buffer();
107         parser.serializeLabel(true, true,  builder.build(), buff);
108         assertArrayEquals(TYPE_ONE_LABEL_BYTES, ByteArray.getAllBytes(buff));
109
110         try {
111             parser.parseLabel(null);
112             fail();
113         } catch (final IllegalArgumentException e) {
114             assertEquals("Array of bytes is mandatory. Cannot be null or empty.", e.getMessage());
115         }
116         try {
117             parser.parseLabel(Unpooled.EMPTY_BUFFER);
118             fail();
119         } catch (final IllegalArgumentException e) {
120             assertEquals("Array of bytes is mandatory. Cannot be null or empty.", e.getMessage());
121         }
122     }
123 }