/*
* Copyright 2013 Basho Technologies Inc.
*
* 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 com.basho.riak.client.core.netty;
import com.basho.riak.client.core.RiakMessage;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.channel.ChannelHandlerContext;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import org.junit.Before;
import org.junit.Test;
import static org.mockito.Mockito.*;
import org.powermock.reflect.Whitebox;
/**
*
* @author Brian Roach <roach at basho dot com>
* @since 2.0
*/
public class RiakMessageCodecTest
{
final private int SIZE_DATA = 30;
private byte[] data;
final private int SIZE_LENGTH = 4;
final private int SIZE_CODE = 1;
private byte code = 1;
private ByteBuf buffer;
private ChannelHandlerContext mockContext;
@Before
public void setUp() throws Exception
{
code = 1;
data = new byte[SIZE_DATA];
Arrays.fill(data, (byte)1);
RiakMessage pbMessage = new RiakMessage(code, data);
RiakMessageCodec codec = new RiakMessageCodec();
mockContext = mock(ChannelHandlerContext.class);
buffer = Unpooled.buffer();
Whitebox.invokeMethod(codec, "encode", mockContext, pbMessage, buffer);
}
@Test
public void encode() throws Exception
{
assertTrue(buffer.isReadable());
assertEquals(buffer.readableBytes(), SIZE_DATA + SIZE_LENGTH + SIZE_CODE);
int encodedLength = buffer.readInt();
byte encodedCode = buffer.readByte();
byte[] encodedData = new byte[SIZE_DATA];
buffer.readBytes(encodedData);
assertEquals(encodedCode, code);
assertEquals(encodedLength, SIZE_DATA + SIZE_CODE);
assertArrayEquals(data, encodedData);
}
@Test
public void decode() throws Exception
{
RiakMessageCodec codec = new RiakMessageCodec();
List<Object> outList = new ArrayList<Object>();
Whitebox.invokeMethod(codec, "decode", mockContext, buffer, outList);
RiakMessage message = (RiakMessage) outList.get(0);
assertEquals(code, message.getCode());
assertArrayEquals(data, message.getData());
}
}