#include "ByteBuffer.h"

#define BOOST_TEST_MODULE "ByteBuffer"

#include <boost/test/unit_test.hpp>
#include "Bluetooth.h"

#define checkBuffer(buffer, size, capacity, cursor) \
    D << "size=" << buffer.getSize() << ", capacity=" << buffer.getCapacity() << ", cursor=" << buffer.getCursor(); \
    BOOST_CHECK_EQUAL(buffer.getSize(), size); \
    BOOST_CHECK_EQUAL(buffer.getCapacity(), capacity); \
    BOOST_CHECK_EQUAL(buffer.getCursor(), cursor)

using namespace std;

template<typename T>
struct do_nothing {
    void operator()(T *) {
    }
};

class Bytes {
public:
    Bytes(size_t size) : capacity(size) {
        _bytes = shared_ptr<uint8_t>(new uint8_t[size]);
        uint8_t *tmp = (uint8_t *) (((uint64_t) &_bytes.get()[0x100]) & 0xffffffffffffff00);
        bytes = shared_ptr<uint8_t>(tmp, ::do_nothing<uint8_t>());
        for (int i = 0; i < size; i++) {
            bytes.get()[i] = (uint8_t) i;
        }
    }
    shared_ptr<uint8_t> bytes;
    shared_ptr<uint8_t> _bytes;
    size_t capacity;
};

BOOST_AUTO_TEST_CASE(empty_buffer) {
    Bytes b(0);
    ByteBuffer buffer(b.bytes, b.capacity);

    checkBuffer(buffer, 0, 0, 0);

    try {
        buffer.read8();
        BOOST_FAIL("Expected exception");
    } catch (ByteBufferException e) {
    }
}

BOOST_AUTO_TEST_CASE(basic) {
    Bytes b(1000);
    ByteBuffer buffer(b.bytes, 1000, 0, 1000);
    checkBuffer(buffer, 1000, 1000, 0);

    BOOST_CHECK_EQUAL(buffer.read8(), 0);
    checkBuffer(buffer, 1000, 1000, 1);

    for (int i = 1; i < b.capacity; i++) {
        BOOST_CHECK_EQUAL(buffer.read8(), b.bytes.get()[i]);
    }
}

BOOST_AUTO_TEST_CASE(setCursor) {
    Bytes b(1000);
    ByteBuffer buffer(b.bytes, 1000, 0, 10);
    checkBuffer(buffer, 10, 1000, 0);

    BOOST_CHECK_EQUAL(buffer.read8(), 0);
    checkBuffer(buffer, 10, 1000, 1);

    buffer.setCursor(0);
    checkBuffer(buffer, 10, 1000, 0);

    BOOST_CHECK_EQUAL(buffer.read8(), 0);
    checkBuffer(buffer, 10, 1000, 1);

    buffer.setCursor(9);
    checkBuffer(buffer, 10, 1000, 9);
}

BOOST_AUTO_TEST_CASE(view) {
    Bytes b(1000);
    ByteBuffer buffer(b.bytes, b.capacity, 0, 10);

    BOOST_CHECK_EQUAL(buffer.read8(), 0);
    ByteBuffer view1 = buffer.view();
    checkBuffer(view1, 9, 9, 0);

    BOOST_CHECK_EQUAL(view1.read8(), 1);
    BOOST_CHECK_EQUAL(view1.read8(), 2);

    ByteBuffer view2 = view1.view();
    checkBuffer(view2, 7, 7, 0);

    BOOST_CHECK_EQUAL(view1.read8(), 3);
}