/* * Copyright (c) 2015-2016, Christoph Engelbert (aka noctarius) and * contributors. All rights reserved. * * 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.noctarius.tengi.core.serialization.impl; import com.noctarius.tengi.core.model.Packet; import com.noctarius.tengi.core.serialization.debugger.SerializationDebugger; import com.noctarius.tengi.spi.buffer.MemoryBuffer; import com.noctarius.tengi.spi.buffer.impl.MemoryBufferFactory; import com.noctarius.tengi.spi.serialization.Protocol; import com.noctarius.tengi.spi.serialization.Serializer; import com.noctarius.tengi.spi.serialization.codec.Codec; import com.noctarius.tengi.spi.serialization.codec.impl.DefaultCodec; import com.noctarius.tengi.spi.serialization.impl.DefaultProtocol; import com.noctarius.tengi.spi.serialization.impl.SerializationClasses; import io.netty.buffer.ByteBuf; import io.netty.buffer.Unpooled; import org.junit.Test; import java.util.Collections; import static org.junit.Assert.assertEquals; public class PacketSerializationTestCase { @Test public void testPacket() throws Exception { Protocol protocol = new DefaultProtocol(Collections.emptyList()); Packet packet = new Packet("Test"); ByteBuf buffer = Unpooled.buffer(); MemoryBuffer memoryBuffer = MemoryBufferFactory.create(buffer); Codec codec = new DefaultCodec(protocol, memoryBuffer); protocol.writeObject("packet", packet, codec); Packet read = protocol.readObject("object", codec); assertEquals(packet, read); } @Test public void testSubclassWithDefaultConstructor() throws Exception { Protocol protocol = new DefaultProtocol(Collections.emptyList()); Packet packet = new SerializationClasses.SubPacketWithDefaultConstructor(); ByteBuf buffer = Unpooled.buffer(); MemoryBuffer memoryBuffer = MemoryBufferFactory.create(buffer); Codec codec = new DefaultCodec(protocol, memoryBuffer); protocol.writeObject("packet", packet, codec); Packet read = protocol.readObject("object", codec); assertEquals(packet, read); } @Test public void testSubclassWithoutDefaultConstructor() throws Exception { Protocol protocol = new DefaultProtocol(Collections.emptyList()); Packet packet = new SerializationClasses.SubPacketWithoutDefaultConstructor("Test"); ByteBuf buffer = Unpooled.buffer(); MemoryBuffer memoryBuffer = MemoryBufferFactory.create(buffer); Codec codec = new DefaultCodec(protocol, memoryBuffer); protocol.writeObject("packet", packet, codec); Packet read = protocol.readObject("object", codec); assertEquals(packet, read); } @Test(expected = NullPointerException.class) public void testSubclassMarshallException() throws Exception { SerializationDebugger.Debugger.ENABLED = true; try { Protocol protocol = new DefaultProtocol(Collections.emptyList()); Serializer serializer = Serializer.create(protocol); Packet packet = new SerializationClasses.SubPacketMarshallException("Test"); ByteBuf buffer = Unpooled.buffer(); MemoryBuffer memoryBuffer = MemoryBufferFactory.create(buffer); Codec codec = new DefaultCodec(protocol, memoryBuffer); serializer.writeObject("packet", packet, codec); } finally { SerializationDebugger.Debugger.ENABLED = false; } } @Test(expected = NullPointerException.class) public void testSubclassUnmarshallException() throws Exception { SerializationDebugger.Debugger.ENABLED = true; try { Protocol protocol = new DefaultProtocol(Collections.emptyList()); Serializer serializer = Serializer.create(protocol); Packet packet = new SerializationClasses.SubPacketUnmarshallException("Test"); ByteBuf buffer = Unpooled.buffer(); MemoryBuffer memoryBuffer = MemoryBufferFactory.create(buffer); Codec codec = new DefaultCodec(protocol, memoryBuffer); serializer.writeObject("packet", packet, codec); Packet read = protocol.readObject("object", codec); assertEquals(packet, read); } finally { SerializationDebugger.Debugger.ENABLED = false; } } }