/*
 * Copyright (c) 2024, Alliance for Open Media. All rights reserved
 *
 * This source code is subject to the terms of the BSD 3-Clause Clear License
 * and the Alliance for Open Media Patent License 1.0. If the BSD 3-Clause Clear
 * License was not distributed with this source code in the LICENSE file, you
 * can obtain it at www.aomedia.org/license/software-license/bsd-3-c-c. If the
 * Alliance for Open Media Patent License 1.0 was not distributed with this
 * source code in the PATENTS file, you can obtain it at
 * www.aomedia.org/license/patent.
 */
#include <sys/types.h>

#include <cstdint>

#include "absl/types/span.h"
#include "fuzztest/fuzztest.h"
#include "gtest/gtest.h"
#include "iamf/common/write_bit_buffer.h"

namespace iamf_tools {
namespace {

void WriteSignedLiteral(uint32_t data, int num_bits) {
  WriteBitBuffer wb(0);
  auto status = wb.WriteUnsignedLiteral(data, num_bits);
  if (status.ok()) {
    EXPECT_EQ(wb.bit_offset(), num_bits);
  } else {
    EXPECT_EQ(wb.bit_offset(), 0);
  }
}

FUZZ_TEST(WriteBitBufferFuzzTest, WriteSignedLiteral);

void WriteSignedLiteral64(uint64_t data, int num_bits) {
  WriteBitBuffer wb(0);
  auto status = wb.WriteUnsignedLiteral(data, num_bits);
  if (status.ok()) {
    EXPECT_EQ(wb.bit_offset(), num_bits);
  } else {
    EXPECT_EQ(wb.bit_offset(), 0);
  }
}

FUZZ_TEST(WriteBitBufferFuzzTest, WriteSignedLiteral64);

void WriteSigned16(int16_t data) {
  WriteBitBuffer wb(0);
  auto status = wb.WriteSigned16(data);
  if (status.ok()) {
    EXPECT_EQ(wb.bit_offset(), 16);
  } else {
    EXPECT_EQ(wb.bit_offset(), 0);
  }
}

FUZZ_TEST(WriteBitBufferFuzzTest, WriteSigned16);

void WriteStringAndValidate(const std::string& data) {
  WriteBitBuffer wb(0);
  auto status = wb.WriteString(data);
  if (status.ok()) {
    const auto number_of_bytes =
        data.size() + 1;  // +1 for the NULL terminator.
    const auto number_of_bits = number_of_bytes * 8;
    EXPECT_EQ(wb.bit_offset(), number_of_bits);
    std::vector<uint8_t> expected_data = {data.begin(), data.end()};
    expected_data.push_back(0);  // NULL terminator.
    EXPECT_EQ(wb.bit_buffer(), expected_data);
  } else {
    EXPECT_EQ(wb.bit_offset(), 0);
  }
}

FUZZ_TEST(WriteBitBufferFuzzTest, WriteStringAndValidate);

void WriteUint8Span(const std::vector<uint8_t>& data) {
  WriteBitBuffer wb(0);
  auto status = wb.WriteUint8Span(absl::MakeConstSpan(data));
  if (status.ok()) {
    EXPECT_EQ(wb.bit_offset(), data.size() * 8);
    EXPECT_EQ(wb.bit_buffer(), data);
  } else {
    EXPECT_EQ(wb.bit_offset(), 0);
  }
}

FUZZ_TEST(WriteBitBufferFuzzTest, WriteUint8Span);

void WriteUleb128(uint32_t data) {
  WriteBitBuffer wb(0);
  auto status = wb.WriteUleb128(data);
  if (status.ok()) {
    EXPECT_NE(wb.bit_offset(), 0);
  } else {
    EXPECT_EQ(wb.bit_offset(), 0);
  }
}

FUZZ_TEST(WriteBitBufferFuzzTest, WriteUleb128);

void WriteIso14496_1Expanded(uint32_t data) {
  WriteBitBuffer wb(0);
  auto status = wb.WriteIso14496_1Expanded(data);
  if (status.ok()) {
    EXPECT_NE(wb.bit_offset(), 0);
  } else {
    EXPECT_EQ(wb.bit_offset(), 0);
  }
}

FUZZ_TEST(WriteBitBufferFuzzTest, WriteIso14496_1Expanded);

}  // namespace
}  // namespace iamf_tools