package org.msgpack;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertArrayEquals;

import java.io.IOException;
import java.io.ByteArrayOutputStream;
import java.io.FileInputStream;
import java.util.Iterator;

import org.msgpack.MessagePack;
import org.msgpack.type.Value;
import org.msgpack.packer.BufferPacker;
import org.msgpack.unpacker.BufferUnpacker;

import org.junit.Test;


public class TestCrossLang {
    private byte[] readData(String path) throws IOException {
        ByteArrayOutputStream bo = new ByteArrayOutputStream();
        FileInputStream input = new FileInputStream(path);
        byte[] buffer = new byte[32*1024];
        while(true) {
            int count = input.read(buffer);
            if(count < 0) {
                break;
            }
            bo.write(buffer, 0, count);
        }
        return bo.toByteArray();
    }

    private byte[] readCompactTestData() throws IOException {
        return readData("src/test/resources/cases_compact.mpac");
    }

    private byte[] readTestData() throws IOException {
        return readData("src/test/resources/cases.mpac");
    }

    @Test
    public void testReadValue() throws IOException {
        MessagePack msgpack = new MessagePack();
        byte[] a = readTestData();
        byte[] b = readCompactTestData();

        BufferUnpacker au = msgpack.createBufferUnpacker().wrap(a);
        BufferUnpacker bu = msgpack.createBufferUnpacker().wrap(b);

        Iterator<Value> at = au.iterator();
        Iterator<Value> bt = bu.iterator();

        while(at.hasNext()) {
            assertTrue(bt.hasNext());
            Value av = at.next();
            Value bv = bt.next();
            assertEquals(av, bv);
        }
        assertFalse(bt.hasNext());
    }

    @Test
    public void testCompactSerialize() throws IOException {
        MessagePack msgpack = new MessagePack();
        byte[] a = readTestData();
        byte[] b = readCompactTestData();

        BufferPacker pk = msgpack.createBufferPacker();

        BufferUnpacker au = msgpack.createBufferUnpacker().wrap(a);
        for(Value av : au) {
            pk.write(av);
        }

        byte[] c = pk.toByteArray();

        assertEquals(b.length, c.length);
        assertArrayEquals(b, c);
    }
}