Add LispMessageEncoder and LispMessageDecoder with unit tests
Change-Id: If73a41687a9c2400de23bbde6179a63ac7f75d15
diff --git a/protocols/lisp/ctl/src/test/java/org/onosproject/lisp/ctl/LispMessageDecoderTest.java b/protocols/lisp/ctl/src/test/java/org/onosproject/lisp/ctl/LispMessageDecoderTest.java
new file mode 100644
index 0000000..025181e
--- /dev/null
+++ b/protocols/lisp/ctl/src/test/java/org/onosproject/lisp/ctl/LispMessageDecoderTest.java
@@ -0,0 +1,145 @@
+/*
+ * Copyright 2016-present Open Networking Laboratory
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.onosproject.lisp.ctl;
+
+import com.google.common.collect.Lists;
+import io.netty.buffer.ByteBuf;
+import io.netty.buffer.Unpooled;
+import org.junit.Test;
+import org.onosproject.lisp.msg.protocols.LispMapNotify;
+import org.onosproject.lisp.msg.protocols.LispMapRegister;
+import org.onosproject.lisp.msg.protocols.LispMapReply;
+import org.onosproject.lisp.msg.protocols.LispMapRequest;
+
+import java.util.List;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.instanceOf;
+import static org.hamcrest.Matchers.is;
+
+/**
+ * Tests for LISP message decoder.
+ */
+public class LispMessageDecoderTest {
+
+ private static final int TYPE_SHIFT_BIT = 4;
+ private static final byte MAP_REQUEST = 1;
+ private static final byte MAP_REPLY = 2;
+ private static final byte MAP_REGISTER = 3;
+ private static final byte MAP_NOTIFY = 4;
+
+
+ private ByteBuf getLispMapRequestBuffer() {
+ ByteBuf buffer = Unpooled.buffer();
+
+ // specify message type
+ buffer.writeByte(MAP_REQUEST << TYPE_SHIFT_BIT);
+
+ // fill up message payload
+ // second byte denotes the number of RLOCs
+ byte[] messageData = {0x0, 0x1, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0};
+ byte[] eidData = {0x0, 0x1, 0x0, 0x0, 0x0, 0x0};
+ byte[] rlocData = {0x0, 0x1, 0x0, 0x0, 0x0, 0x0};
+ buffer.writeBytes(messageData);
+ buffer.writeBytes(eidData);
+ buffer.writeBytes(rlocData);
+ return buffer;
+ }
+
+ private ByteBuf getLispMapReplyBuffer() {
+ ByteBuf buffer = Unpooled.buffer();
+
+ // specify message type
+ buffer.writeByte(MAP_REPLY << TYPE_SHIFT_BIT);
+
+ // fill up message payload
+ byte[] messageData = {0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0};
+ buffer.writeBytes(messageData);
+ return buffer;
+ }
+
+ private ByteBuf getLispMapRegisterBuffer() {
+ ByteBuf buffer = Unpooled.buffer();
+
+ // specify message type
+ buffer.writeByte(MAP_REGISTER << TYPE_SHIFT_BIT);
+
+ // fill up message payload
+ byte[] messageData = {0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0};
+ byte[] keyId = {0x0, 0x1};
+
+ // assume that we have auth data which has 2 bytes size
+ byte[] authDataLength = {0x0, 0x2};
+ byte[] authData = {0x0, 0x0};
+
+ buffer.writeBytes(messageData);
+ buffer.writeBytes(keyId);
+ buffer.writeBytes(authDataLength);
+ buffer.writeBytes(authData);
+ return buffer;
+ }
+
+ private ByteBuf getLispMapNotifyBuffer() {
+ ByteBuf buffer = Unpooled.buffer();
+
+ // specify message type
+ buffer.writeByte(MAP_NOTIFY << TYPE_SHIFT_BIT);
+
+ // fill up message payload
+ byte[] messageData = {0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0};
+ byte[] keyId = {0x0, 0x1};
+
+ // assume that we have auth data which has 2 bytes size
+ byte[] authDataLength = {0x0, 0x2};
+ byte[] authData = {0x0, 0x0};
+
+ buffer.writeBytes(messageData);
+ buffer.writeBytes(keyId);
+ buffer.writeBytes(authDataLength);
+ buffer.writeBytes(authData);
+
+ return buffer;
+ }
+
+ @Test(expected = IllegalArgumentException.class)
+ public void testDecodeNoChannel() throws Exception {
+ LispMessageDecoder decoder = new LispMessageDecoder();
+
+ List<Object> list = Lists.newArrayList();
+ decoder.decode(new ChannelHandlerContextAdapter(), Unpooled.buffer(), list);
+ }
+
+ @Test
+ public void testDecode() throws Exception {
+ LispMessageDecoder decoder = new LispMessageDecoder();
+ ByteBuf requestBuff = getLispMapRequestBuffer();
+ ByteBuf replyBuff = getLispMapReplyBuffer();
+ ByteBuf registerBuff = getLispMapRegisterBuffer();
+ ByteBuf notifyBuff = getLispMapNotifyBuffer();
+
+ List<Object> list = Lists.newArrayList();
+ decoder.decode(new ChannelHandlerContextAdapter(), requestBuff, list);
+ decoder.decode(new ChannelHandlerContextAdapter(), replyBuff, list);
+ decoder.decode(new ChannelHandlerContextAdapter(), registerBuff, list);
+ decoder.decode(new ChannelHandlerContextAdapter(), notifyBuff, list);
+
+ assertThat(list.size(), is(4));
+ assertThat(list.get(0), is(instanceOf(LispMapRequest.class)));
+ assertThat(list.get(1), is(instanceOf(LispMapReply.class)));
+ assertThat(list.get(2), is(instanceOf(LispMapRegister.class)));
+ assertThat(list.get(3), is(instanceOf(LispMapNotify.class)));
+ }
+}