#include "utf32.hh"

#include "utf_error.hh"

#include <array>
#include <gtest/gtest.h>

TEST(utf32be, sanity) {
  size_t offset = 0;
  auto ret = utf::read32be(
      std::array<uint8_t, 4>({0x00, 0x00, 0x00, 0x24}), offset);
  EXPECT_EQ('$', ret);
  EXPECT_EQ(4, offset);

  offset = 0;
  ret = utf::read32be(
      std::array<uint8_t, 4>({0x00, 0x00, 0x20, 0xAC}), offset);
  EXPECT_EQ(0x20AC, ret);
  EXPECT_EQ(4, offset);

  offset = 0;
  ret = utf::read32be(
      std::array<uint8_t, 4>({0x00, 0x01, 0x04, 0x37}), offset);
  EXPECT_EQ(0x10437, ret);
  EXPECT_EQ(4, offset);
}

TEST(utf32le, sanity) {
  size_t offset = 0;
  auto ret = utf::read32le(
      std::array<uint8_t, 4>({0x24, 0x00, 0x00, 0x00}), offset);
  EXPECT_EQ('$', ret);
  EXPECT_EQ(4, offset);

  offset = 0;
  ret = utf::read32le(
      std::array<uint8_t, 4>({0xAC, 0x20, 0x00, 0x00}), offset);
  EXPECT_EQ(0x20AC, ret);
  EXPECT_EQ(4, offset);

  offset = 0;
  ret = utf::read32le(
      std::array<uint8_t, 4>({0x37, 0x04, 0x01, 0x00}), offset);
  EXPECT_EQ(0x10437, ret);
  EXPECT_EQ(4, offset);
}

TEST(utf32be, invalid) {
  size_t offset = 0;
  auto ret = utf::read32be(
      std::array<uint8_t, 4>({0xFF, 0xFF, 0xFF, 0xFF}), offset);
  EXPECT_EQ(utf::INVALID, ret);
  EXPECT_EQ(0, offset);

  offset = 0;
  ret = utf::read32be(
      std::array<uint8_t, 4>({0x00, 0x00, 0xD8, 0x00}), offset);
  EXPECT_EQ(utf::INVALID, ret);
  EXPECT_EQ(0, offset);

  offset = 0;
  ret = utf::read32be(std::array<uint8_t, 1>({}), offset);
  EXPECT_EQ(utf::NEED_MORE, ret);
  EXPECT_EQ(0, offset);

  offset = 0;
  ret = utf::read32be(std::array<uint8_t, 1>({0x00}), offset);
  EXPECT_EQ(utf::NEED_MORE, ret);
  EXPECT_EQ(0, offset);

  offset = 0;
  ret = utf::read32be(std::array<uint8_t, 2>({0x00, 0x00}), offset);
  EXPECT_EQ(utf::NEED_MORE, ret);
  EXPECT_EQ(0, offset);

  offset = 0;
  ret = utf::read32be(std::array<uint8_t, 3>({0x00, 0x00, 0x00}), offset);
  EXPECT_EQ(utf::NEED_MORE, ret);
  EXPECT_EQ(0, offset);
}

TEST(utf32le, invalid) {
  size_t offset = 0;
  auto ret = utf::read32le(
      std::array<uint8_t, 4>({0xFF, 0xFF, 0xFF, 0xFF}), offset);
  EXPECT_EQ(utf::INVALID, ret);
  EXPECT_EQ(0, offset);

  offset = 0;
  ret = utf::read32le(
      std::array<uint8_t, 4>({0x00, 0xD8, 0x00, 0x00}), offset);
  EXPECT_EQ(utf::INVALID, ret);
  EXPECT_EQ(0, offset);

  offset = 0;
  ret = utf::read32le(std::array<uint8_t, 0>(), offset);
  EXPECT_EQ(utf::NEED_MORE, ret);
  EXPECT_EQ(0, offset);

  offset = 0;
  ret = utf::read32le(std::array<uint8_t, 1>({0x00}), offset);
  EXPECT_EQ(utf::NEED_MORE, ret);
  EXPECT_EQ(0, offset);

  offset = 0;
  ret = utf::read32le(std::array<uint8_t, 2>({0x00, 0x00}), offset);
  EXPECT_EQ(utf::NEED_MORE, ret);
  EXPECT_EQ(0, offset);

  offset = 0;
  ret = utf::read32le(std::array<uint8_t, 3>({0x00, 0x00, 0x00}), offset);
  EXPECT_EQ(utf::NEED_MORE, ret);
  EXPECT_EQ(0, offset);
}

TEST(utf32be, bom) {
  std::array<uint8_t, 8> data({0x00, 0x00, 0xFF, 0xFE, 0x00, 0x00, 0x20, 0xAC});
  size_t offset = 0;
  auto ret = utf::read32be(data, offset);
  EXPECT_EQ(0xFFFE, ret);
  ret = utf::read32be(data, offset);
  EXPECT_EQ(0x20AC, ret);
  ret = utf::read32be(data, offset);
  EXPECT_EQ(utf::NEED_MORE, ret);
  EXPECT_EQ(data.size(), offset);
}

TEST(utf32le, bom) {
  std::array<uint8_t, 8> data({0xFE, 0xFF, 0x00, 0x00, 0xAC, 0x20, 0x00, 0x00});
  size_t offset = 0;
  auto ret = utf::read32le(data, offset);
  EXPECT_EQ(0xFFFE, ret);
  ret = utf::read32le(data, offset);
  EXPECT_EQ(0x20AC, ret);
  ret = utf::read32le(data, offset);
  EXPECT_EQ(utf::NEED_MORE, ret);
  EXPECT_EQ(data.size(), offset);
}