/* Copyright (C) 2024 Aryadev Chavali

 * You may distribute and modify this code under the terms of the
 * GPLv2 license.  You should have received a copy of the GPLv2
 * license with this file.  If not, please write to:
 * aryadev@aryadevchavali.com.

 * Created: 2024-04-28
 * Author: Aryadev Chavali
 * Description: Tests for base.h
 */

#ifndef TEST_BASE_H
#define TEST_BASE_H

#include <lib/base.h>
#include <testing.h>

void testing_lib_base_word_safe_sub(void)
{
  const struct TestCase
  {
    word_t a;
    word_t b;
    word_t expected;
  } tests[] = {
      {0xFF, 0, 0xFF},
      {1LU << 32, 1LU << 31, 1LU << 31},
      {0, 100, 0},
      {~(1LU << 62), ~(1LU << 63), 1LU << 62},
  };
  for (size_t i = 0; i < ARR_SIZE(tests); ++i)
  {
    const struct TestCase test = tests[i];
    const word_t res           = test.expected;
    const word_t got           = WORD_SAFE_SUB(test.a, test.b);
    INFO(__func__, "Testing(a=0x%lX, b=0x%lX)\n", test.a, test.b);
    if (got != res)
    {
      FAIL(__func__, "[%lu] -> Expected 0x%lX got 0x%lX\n", i, res, got);
      assert(false);
    }
  }
}

void testing_lib_base_word_nth_byte(void)
{
  const struct TestCase
  {
    word_t word;
    size_t n;
    byte_t expected;
  } tests[] = {
      {0, 0, 0},
      {0x0123456789abcdef, 0, 0xef},
      {0x0123456789abcdef, 1, 0xcd},
      {0x0123456789abcdef, 2, 0xab},
      {0x0123456789abcdef, 3, 0x89},
      {0x0123456789abcdef, 4, 0x67},
      {0x0123456789abcdef, 5, 0x45},
      {0x0123456789abcdef, 6, 0x23},
      {0x0123456789abcdef, 7, 0x01},
  };

  for (size_t i = 0; i < ARR_SIZE(tests); ++i)
  {
    const word_t w        = tests[i].word;
    const size_t n        = tests[i].n;
    const byte_t expected = tests[i].expected;
    INFO(__func__, "Testing(0x%lX, %lu)\n", w, n);
    const byte_t got = WORD_NTH_BYTE(w, n);
    if (got != expected)
    {
      FAIL(__func__, "[%lu] -> Expected 0x%x got 0x%x\n", i, expected, got);
      assert(false);
    }
  }
}

void testing_lib_base_word_nth_hword(void)
{
  const struct TestCase
  {
    word_t word;
    size_t n;
    hword_t expected;
  } tests[] = {
      {0, 0, 0},
      {0x0123456789abcdef, 0, 0x89abcdef},
      {0x0123456789abcdef, 1, 0x01234567},
      {~(1LU << 63), 0, 0xFFFFFFFF},
      {~(1LU << 63), 1, 0x7FFFFFFF},
  };

  for (size_t i = 0; i < ARR_SIZE(tests); ++i)
  {
    const word_t w         = tests[i].word;
    const size_t n         = tests[i].n;
    const hword_t expected = tests[i].expected;
    INFO(__func__, "Testing(0x%lX, %lu)\n", w, n);
    const hword_t got = WORD_NTH_HWORD(w, n);
    if (got != expected)
    {
      FAIL(__func__, "[%lu] -> Expected 0x%x got 0x%x\n", i, expected, got);
      assert(false);
    }
  }
}

void testing_lib_base_bytes_to_hword(void)
{
  const struct TestCase
  {
    byte_t bytes[4];
    hword_t expected;
  } tests[] = {{{0, 0, 0, 0}, 0},
               {{0xFF, 0xFF, 0xFF, 0xFF}, HWORD_MAX},
               {{1, 0, 0, 0}, 1},
               {{0, 0, 0, 0b10000000}, 1 << 31},
               {{0x89, 0xab, 0xcd, 0xef}, 0xefcdab89}};

  for (size_t i = 0; i < ARR_SIZE(tests); ++i)
  {
    const hword_t got = convert_bytes_to_hword(tests[i].bytes);
    INFO(__func__, "Testing(0x%X, 0x%X, 0x%X, 0x%X)\n", tests[i].bytes[0],
         tests[i].bytes[1], tests[i].bytes[2], tests[i].bytes[3]);
    if (tests[i].expected != got)
    {
      FAIL(__func__, "[%lu] -> Expected 0x%x got 0x%x\n", i, tests[i].expected,
           got);
      assert(false);
    }
  }
}

void testing_lib_base_bytes_to_word(void)
{
  const struct TestCase
  {
    byte_t bytes[8];
    word_t expected;
  } tests[] = {
      {{0, 0, 0, 0}, 0},
      {{0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF}, WORD_MAX},
      {{0x01, 0, 0, 0, 0, 0, 0, 0}, 1},
      {{0, 0, 0, 0, 0, 0, 0, 0b10000000}, 1LU << 63},
      {{0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef}, 0xefcdab8967452301}};

  for (size_t i = 0; i < ARR_SIZE(tests); ++i)
  {
    INFO(__func__, "Testing(0x%X, 0x%X, 0x%X, 0x%X, 0x%X, 0x%X, 0x%X, 0x%X)\n",
         tests[i].bytes[0], tests[i].bytes[1], tests[i].bytes[2],
         tests[i].bytes[3], tests[i].bytes[4], tests[i].bytes[5],
         tests[i].bytes[6], tests[i].bytes[7]);
    const word_t got = convert_bytes_to_word(tests[i].bytes);
    if (tests[i].expected != got)
    {
      FAIL(__func__, "[%lu] -> Expected 0x%lx got 0x%lx\n", i,
           tests[i].expected, got);
      assert(false);
    }
  }
}

TEST_SUITE(test_lib, CREATE_TEST(testing_lib_base_word_safe_sub),
           CREATE_TEST(testing_lib_base_word_nth_byte),
           CREATE_TEST(testing_lib_base_word_nth_hword),
           CREATE_TEST(testing_lib_base_bytes_to_hword),
           CREATE_TEST(testing_lib_base_bytes_to_word));

#endif