#include "greatest.h"

#include "task.h"

SUITE(task_tests);

TEST task_test(void) {
  function_id func_id = {
      {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1}};
  task_spec* task = alloc_task_spec(func_id, 4, 2, 10);
  ASSERT(task_num_args(task) == 4);
  ASSERT(task_num_returns(task) == 2);

  unique_id arg1 = {
      {2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2}};
  ASSERT(task_args_add_ref(task, arg1) == 0);
  ASSERT(task_args_add_val(task, (uint8_t*) "hello", 5) == 1);
  unique_id arg2 = {
      {3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3}};
  ASSERT(task_args_add_ref(task, arg2) == 2);
  ASSERT(task_args_add_val(task, (uint8_t*) "world", 5) == 3);

  unique_id ret0 = {
      {4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4}};
  unique_id ret1 = {
      {5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5}};
  memcpy(task_return(task, 0), &ret0, sizeof(ret0));
  memcpy(task_return(task, 1), &ret1, sizeof(ret1));

  ASSERT(memcmp(task_arg_id(task, 0), &arg1, sizeof(arg1)) == 0);
  ASSERT(memcmp(task_arg_val(task, 1), (uint8_t*) "hello",
                task_arg_length(task, 1)) == 0);
  ASSERT(memcmp(task_arg_id(task, 2), &arg2, sizeof(arg2)) == 0);
  ASSERT(memcmp(task_arg_val(task, 3), (uint8_t*) "world",
                task_arg_length(task, 3)) == 0);

  ASSERT(memcmp(task_return(task, 0), &ret0, sizeof(unique_id)) == 0);
  ASSERT(memcmp(task_return(task, 1), &ret1, sizeof(unique_id)) == 0);

  free_task_spec(task);
  PASS();
}

SUITE(task_tests) {
  RUN_TEST(task_test);
}

GREATEST_MAIN_DEFS();

int main(int argc, char** argv) {
  GREATEST_MAIN_BEGIN();
  RUN_SUITE(task_tests);
  GREATEST_MAIN_END();
}