/*
 * Copyright (C) 2020 The Android Open Source Project
 *
 * 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.
 */

#include "minikin/Buffer.h"

#include <gtest/gtest.h>

namespace minikin {

class TestObject {
public:
    void writeTo(BufferWriter* writer) const {
        // Total size = 20
        // uint8_t (1)
        writer->write<uint8_t>(0xAB);
        // padding (1), uint16_t (2)
        writer->write<uint16_t>(0xCDEF);
        // uint8_t(1)
        writer->write<uint8_t>(0x01);
        // padding (3), array size (4), uint32_t (4) * 2
        uint32_t uint32Array[] = {0x98765432, 0x98765433};
        writer->writeArray<uint32_t>(uint32Array, 2);
        uint16_t* uint16Array = writer->reserve<uint16_t>(2 * sizeof(uint16_t));
        if (uint16Array != nullptr) {
            uint16Array[0] = 0x1234u;
            uint16Array[1] = 0x5678u;
        }
    }
};

TEST(BufferTest, testMeasureWriteRead) {
    TestObject testObject;
    BufferWriter fakeWriter(nullptr);
    testObject.writeTo(&fakeWriter);
    ASSERT_EQ(fakeWriter.size(), 24u);
    std::vector<uint8_t> buffer(fakeWriter.size());

    BufferWriter writer(buffer.data());
    testObject.writeTo(&writer);
    ASSERT_EQ(writer.size(), buffer.size());

    BufferReader reader(buffer.data());
    ASSERT_EQ(reader.current(), buffer.data());
    ASSERT_EQ(reader.read<uint8_t>(), 0xABu);
    ASSERT_EQ(reader.current(), buffer.data() + 1u);
    ASSERT_EQ(reader.read<uint16_t>(), 0xCDEFu);
    ASSERT_EQ(reader.current(), buffer.data() + 4u);
    ASSERT_EQ(reader.read<uint8_t>(), 0x01u);
    ASSERT_EQ(reader.current(), buffer.data() + 5u);
    auto [uint32Array, size] = reader.readArray<uint32_t>();
    ASSERT_EQ(size, 2u);
    ASSERT_EQ(uint32Array[0], 0x98765432u);
    ASSERT_EQ(uint32Array[1], 0x98765433u);
    ASSERT_EQ(reader.current(), buffer.data() + 20u);
    const uint16_t* uint16Array = reader.map<uint16_t>(4);
    ASSERT_EQ(uint16Array[0], 0x1234u);
    ASSERT_EQ(uint16Array[1], 0x5678u);
    ASSERT_EQ(reader.current(), buffer.data() + 24u);
}

TEST(BufferTest, testSkip) {
    TestObject testObject;
    BufferWriter fakeWriter(nullptr);
    testObject.writeTo(&fakeWriter);
    ASSERT_EQ(fakeWriter.size(), 24u);
    std::vector<uint8_t> buffer(fakeWriter.size());

    BufferWriter writer(buffer.data());
    testObject.writeTo(&writer);
    ASSERT_EQ(writer.size(), buffer.size());

    BufferReader reader(buffer.data());
    ASSERT_EQ(reader.current(), buffer.data());
    reader.skip<uint8_t>();
    ASSERT_EQ(reader.current(), buffer.data() + 1u);
    reader.read<uint16_t>();
    ASSERT_EQ(reader.current(), buffer.data() + 4u);
    reader.skip<uint8_t>();
    ASSERT_EQ(reader.current(), buffer.data() + 5u);
    reader.skipArray<uint32_t>();
    ASSERT_EQ(reader.current(), buffer.data() + 20u);
    // No skip function for mapped data.
    reader.map<uint16_t>(4);
    ASSERT_EQ(reader.current(), buffer.data() + 24u);
}

}  // namespace minikin