package org.msgpack;

import static org.junit.Assert.assertEquals;

import static org.junit.Assert.assertArrayEquals;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

import org.msgpack.MessagePack;

import org.msgpack.packer.Packer;

import org.msgpack.packer.BufferPacker;

import org.msgpack.unpacker.Unpacker;

import org.msgpack.unpacker.BufferUnpacker;

import org.junit.Test;

public class TestSimplePackable {

// all files are REQUIRED

public static class Sample01 implements MessagePackable {

public String f0;

public int[] f1;

public List<String> f2;

public Sample01() { }

public void writeTo(Packer pk) throws IOException {

pk.writeArrayBegin(3);

pk.write(f0);

pk.writeArrayBegin(f1.length);

for(int e : f1) {

pk.write(e);

}

pk.writeArrayEnd();

pk.writeArrayBegin(f2.size());

for(String e : f2) {

pk.write(e);

}

pk.writeArrayEnd();

pk.writeArrayEnd();

}

public void readFrom(Unpacker u) throws IOException {

u.readArrayBegin();

f0 = u.readString();

int nf1 = u.readArrayBegin();

f1 = new int[nf1];

for(int i=0; i < nf1; i++) {

f1[i] = u.readInt();

}

u.readArrayEnd();

int nf2 = u.readArrayBegin();

f2 = new ArrayList<String>(nf2);

for(int i=0; i < nf2; i++) {

f2.add(u.readString());

}

u.readArrayEnd();

u.readArrayEnd();

}

}

@Test

public void testSample01() throws IOException {

MessagePack msgpack = new MessagePack();

Sample01 a = new Sample01();

a.f0 = "aaa";

a.f1 = new int[3];

a.f1[0] = 1010;

a.f1[1] = 2020;

a.f1[2] = 3030;

a.f2 = new ArrayList<String>();

a.f2.add("xx");

a.f2.add("yy");

BufferPacker pk = msgpack.createBufferPacker();

a.writeTo(pk);

byte[] raw = pk.toByteArray();

BufferUnpacker u = msgpack.createBufferUnpacker().wrap(raw);

Sample01 b = new Sample01();

b.readFrom(u);

assertEquals(a.f0, b.f0);

assertArrayEquals(a.f1, b.f1);

assertEquals(a.f2, b.f2);

}

// some files are OPTIONAL or NULLABLE

public static class Sample02 implements MessagePackable {

public String f0; // nullable

public long f1; // primitive

public Integer f2; // required

public String f3; // optional

public Sample02() { }

public void writeTo(Packer pk) throws IOException {

pk.writeArrayBegin(4);

pk.write(f0);

pk.write(f1);

if(f2 == null) {

throw new MessageTypeException("f2 is required but null");

}

pk.write(f2);

pk.write(f3);

pk.writeArrayEnd();

}

public void readFrom(Unpacker u) throws IOException {

u.readArrayBegin();

f0 = u.read(String.class);

f1 = u.readLong();

if(u.trySkipNil()) {

f2 = null;

} else {

f2 = u.read(Integer.class);

}

if(u.trySkipNil()) {

f3 = null;

} else {

f3 = u.read(String.class);

}

u.readArrayEnd();

}

}

@Test

public void testSample02() throws IOException {

MessagePack msgpack = new MessagePack();

Sample02 a = new Sample02();

a.f0 = "aaa";

a.f1 = 1;

a.f2 = 22;

a.f3 = null;

BufferPacker pk = msgpack.createBufferPacker();

a.writeTo(pk);

byte[] raw = pk.toByteArray();

BufferUnpacker u = msgpack.createBufferUnpacker().wrap(raw);

Sample02 b = new Sample02();

b.readFrom(u);

assertEquals(a.f0, b.f0);

assertEquals(a.f1, b.f1);

assertEquals(a.f2, b.f2);

assertEquals(a.f3, b.f3);

}

}