Java tutorial
/* * Copyright (C) 2009 Swedish Institute of Computer Science (SICS) Copyright (C) * 2009 Royal Institute of Technology (KTH) * * KompicsToolbox is free software; you can redistribute it and/or * modify it under the terms of the GNU General Public License * as published by the Free Software Foundation; either version 2 * of the License, or (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */ package se.sics.ktoolbox.util.network.basic; import com.google.common.base.Optional; import io.netty.buffer.ByteBuf; import io.netty.buffer.Unpooled; import java.net.InetAddress; import java.net.UnknownHostException; import java.util.Objects; import org.junit.Assert; import org.junit.BeforeClass; import org.junit.Test; import se.sics.kompics.network.Transport; import se.sics.kompics.network.netty.serialization.Serializer; import se.sics.kompics.network.netty.serialization.Serializers; import se.sics.ktoolbox.util.identifiable.BasicIdentifiers; import se.sics.ktoolbox.util.identifiable.IdentifierFactory; import se.sics.ktoolbox.util.identifiable.IdentifierRegistry; import se.sics.ktoolbox.util.network.KAddress; import se.sics.ktoolbox.util.network.nat.NatAwareAddressImpl; import se.sics.ktoolbox.util.setup.BasicSerializerSetup; /** * @author Alex Ormenisan <aaor@kth.se> */ public class BasicHeaderSerializerTest { @BeforeClass public static void setup() { BasicIdentifiers.registerDefaults(1234l); int serializerId = 128; serializerId = BasicSerializerSetup.registerBasicSerializers(serializerId); } @Test public void testBasicAddress() throws UnknownHostException { Serializer serializer = Serializers.lookupSerializer(BasicHeader.class); BasicHeader<KAddress> original, copy; ByteBuf serializedOriginal, serializedCopy; InetAddress localHost; try { localHost = InetAddress.getByName("localhost"); } catch (UnknownHostException ex) { throw new RuntimeException(ex); } IdentifierFactory nodeIdFactory = IdentifierRegistry.lookup(BasicIdentifiers.Values.NODE.toString()); BasicAddress simpleAdr1 = new BasicAddress(localHost, 10000, nodeIdFactory.randomId()); BasicAddress simpleAdr2 = new BasicAddress(localHost, 10000, nodeIdFactory.randomId()); original = new BasicHeader(simpleAdr1, simpleAdr2, Transport.UDP); serializedOriginal = Unpooled.buffer(); serializer.toBinary(original, serializedOriginal); serializedCopy = Unpooled.buffer(); serializedOriginal.getBytes(0, serializedCopy, serializedOriginal.readableBytes()); copy = (BasicHeader) serializer.fromBinary(serializedCopy, Optional.absent()); Assert.assertEquals(original.getSource(), copy.getSource()); Assert.assertEquals(original.getDestination(), copy.getDestination()); Assert.assertEquals(original.getProtocol(), copy.getProtocol()); Assert.assertEquals(0, serializedCopy.readableBytes()); } @Test public void testNatAwareOpenAddress() throws UnknownHostException { Serializer serializer = Serializers.lookupSerializer(BasicHeader.class); BasicHeader<NatAwareAddressImpl> original, copy; ByteBuf serializedOriginal, serializedCopy; InetAddress localHost; try { localHost = InetAddress.getByName("localhost"); } catch (UnknownHostException ex) { throw new RuntimeException(ex); } IdentifierFactory nodeIdFactory = IdentifierRegistry.lookup(BasicIdentifiers.Values.NODE.toString()); NatAwareAddressImpl simpleAdr1 = NatAwareAddressImpl .open(new BasicAddress(localHost, 10000, nodeIdFactory.randomId())); NatAwareAddressImpl simpleAdr2 = NatAwareAddressImpl .open(new BasicAddress(localHost, 10000, nodeIdFactory.randomId())); original = new BasicHeader(simpleAdr1, simpleAdr2, Transport.UDP); serializedOriginal = Unpooled.buffer(); serializer.toBinary(original, serializedOriginal); serializedCopy = Unpooled.buffer(); serializedOriginal.getBytes(0, serializedCopy, serializedOriginal.readableBytes()); copy = (BasicHeader) serializer.fromBinary(serializedCopy, Optional.absent()); Assert.assertFalse(copy.getSource().getPrivateAdr().isPresent()); Assert.assertEquals(original.getSource().getPublicAdr(), copy.getSource().getPublicAdr()); Assert.assertEquals(original.getSource().getNatType(), copy.getSource().getNatType()); Assert.assertTrue(Objects.equals(original.getSource().getParents(), copy.getSource().getParents())); Assert.assertFalse(copy.getDestination().getPrivateAdr().isPresent()); Assert.assertEquals(original.getDestination().getPublicAdr(), copy.getDestination().getPublicAdr()); Assert.assertEquals(original.getDestination().getNatType(), copy.getDestination().getNatType()); Assert.assertTrue( Objects.equals(original.getDestination().getParents(), copy.getDestination().getParents())); Assert.assertEquals(original.getProtocol(), copy.getProtocol()); Assert.assertEquals(0, serializedCopy.readableBytes()); } }