move bmp to bmp aggregator

[bgpcep.git] / bmp / bmp-spi / src / test / java / org / opendaylight / protocol / bmp / spi / parser / AbstractBmpMessageWithTlvParserTest.java

diff --git a/bmp/bmp-spi/src/test/java/org/opendaylight/protocol/bmp/spi/parser/AbstractBmpMessageWithTlvParserTest.java b/bmp/bmp-spi/src/test/java/org/opendaylight/protocol/bmp/spi/parser/AbstractBmpMessageWithTlvParserTest.java
new file mode 100644 (file)
index 0000000..1e16c24
--- /dev/null
+++ b/bmp/bmp-spi/src/test/java/org/opendaylight/protocol/bmp/spi/parser/AbstractBmpMessageWithTlvParserTest.java
@@ -0,0 +1,102 @@
+/*
+ * Copyright (c) 2015 Pantheon Technologies s.r.o. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.opendaylight.protocol.bmp.spi.parser;
+
+import static org.junit.Assert.assertArrayEquals;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNull;
+import com.google.common.base.Preconditions;
+import io.netty.buffer.ByteBuf;
+import io.netty.buffer.Unpooled;
+import java.nio.charset.StandardCharsets;
+import org.junit.Before;
+import org.junit.Test;
+import org.opendaylight.protocol.bmp.spi.registry.SimpleBmpTlvRegistry;
+import org.opendaylight.protocol.util.ByteArray;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev150512.Tlv;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev150512.description.tlv.DescriptionTlv;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev150512.description.tlv.DescriptionTlvBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev150512.initiation.TlvsBuilder;
+import org.opendaylight.yangtools.yang.binding.Notification;
+
+public class AbstractBmpMessageWithTlvParserTest {
+
+    private final SimpleBmpTlvRegistry registry = new SimpleBmpTlvRegistry();
+    private final SimpleHandler parser = new SimpleHandler(this.registry);
+    private static final byte[] DATA = { 0, 1, 0, 4, 't', 'e', 's', 't' };
+    private static final int TYPE = 1;
+
+    public static final BmpTlvSerializer DESCRIPTION_TLV_SERIALIZER = (tlv, output) -> {
+        Preconditions.checkArgument(tlv instanceof DescriptionTlv, "DescriptionTlv is mandatory.");
+        TlvUtil.formatTlvAscii(TYPE, ((DescriptionTlv) tlv).getDescription(), output);
+    };
+
+    public static final BmpTlvParser DESCRIPTION_TLV_PARSER = buffer -> {
+        if (buffer == null) {
+            return null;
+        }
+        return new DescriptionTlvBuilder().setDescription(buffer.toString(StandardCharsets.US_ASCII)).build();
+    };
+
+    @Before
+    public void setUp() {
+        this.registry.registerBmpTlvParser(TYPE, DESCRIPTION_TLV_PARSER);
+        this.registry.registerBmpTlvSerializer(DescriptionTlv.class, DESCRIPTION_TLV_SERIALIZER);
+    }
+
+    @Test
+    public void testParseTlvs() throws BmpDeserializationException {
+        final ByteBuf buffer = Unpooled.EMPTY_BUFFER;
+        final TlvsBuilder builder = new TlvsBuilder();
+        this.parser.parseTlvs(builder, buffer);
+        assertNull(builder.getDescriptionTlv());
+
+        this.parser.parseTlvs(builder, Unpooled.wrappedBuffer(DATA));
+        assertNotNull(builder.getDescriptionTlv());
+        assertEquals("test", builder.getDescriptionTlv().getDescription());
+    }
+
+    @Test
+    public void testSerializeTlv() {
+        final ByteBuf output = Unpooled.buffer();
+        final DescriptionTlvBuilder builder = new DescriptionTlvBuilder().setDescription("test");
+        this.parser.serializeTlv(builder.build(), output);
+        assertArrayEquals(DATA, ByteArray.getAllBytes(output));
+    }
+
+    @Test(expected=BmpDeserializationException.class)
+    public void testParseCorruptedTlv() throws BmpDeserializationException {
+        final byte[] wrongData = { 0, 1, 0, 10, 't', 'e', 's', 't' };
+        this.parser.parseTlvs(new TlvsBuilder(), Unpooled.wrappedBuffer(wrongData));
+    }
+
+    private static final class SimpleHandler extends AbstractBmpMessageWithTlvParser<TlvsBuilder> {
+        public SimpleHandler(final BmpTlvRegistry tlvRegistry) {
+            super(tlvRegistry);
+        }
+        @Override
+        public void serializeMessageBody(final Notification message, final ByteBuf buffer) {
+        }
+        @Override
+        public Notification parseMessageBody(final ByteBuf bytes) throws BmpDeserializationException {
+            return null;
+        }
+        @Override
+        public int getBmpMessageType() {
+            return 0;
+        }
+        @Override
+        protected void addTlv(final TlvsBuilder builder, final Tlv tlv) {
+            if(tlv != null && tlv instanceof DescriptionTlv) {
+                builder.setDescriptionTlv((DescriptionTlv)tlv);
+            }
+        }
+    }
+}