/* This software was written by Dirk Engling <erdgeist@erdgeist.org>
   It is considered beerware. Prost. Skol. Cheers or whatever.

   $id$ */

/* System */
#include <stdlib.h>
#include <string.h>

/* Opentracker */
#include "ot_mutex.h"
#include "test.h"

/* A worker asks for a class of task and blocks until one turns up, so a task
   whose class matches no worker is never claimed and whoever is waiting for
   its answer waits forever. These cases pin the matching. */

static void case_task_goes_to_its_own_class(void) {
  ot_tasktype asked;
  ot_taskid   taskid;

  test_case("a worker is handed the task of the class it asked for");
  CHECK_INT(mutex_workqueue_pushtask(41, TASK_FULLSCRAPE_TPB_ASCII), 0);
  asked  = TASK_FULLSCRAPE;
  taskid = mutex_workqueue_poptask(&asked);
  CHECK(taskid != 0);

  test_case("and the full task type, not just its class");
  CHECK_INT(asked, TASK_FULLSCRAPE_TPB_ASCII);
  mutex_workqueue_canceltask(41);
}

static void case_flags_survive_the_queue(void) {
  ot_tasktype asked;

  test_case("flags outside the class mask are carried through");
  CHECK_INT(mutex_workqueue_pushtask(42, TASK_FULLSCRAPE_TPB_URLENCODED | TASK_FLAG_GZIP), 0);
  asked = TASK_FULLSCRAPE;
  CHECK(mutex_workqueue_poptask(&asked) != 0);
  CHECK_INT(asked, TASK_FULLSCRAPE_TPB_URLENCODED | TASK_FLAG_GZIP);
  mutex_workqueue_canceltask(42);
}

static void case_classes_do_not_cross(void) {
  ot_tasktype asked;
  ot_taskid   taskid;

  test_case("a stats worker passes over a fullscrape task");
  CHECK_INT(mutex_workqueue_pushtask(43, TASK_FULLSCRAPE_TPB_BINARY), 0);
  CHECK_INT(mutex_workqueue_pushtask(44, TASK_STATS_PEERS), 0);

  asked  = TASK_STATS;
  taskid = mutex_workqueue_poptask(&asked);
  CHECK(taskid != 0);
  CHECK_INT(asked, TASK_STATS_PEERS);

  test_case("and leaves it for the one that wants it");
  asked  = TASK_FULLSCRAPE;
  taskid = mutex_workqueue_poptask(&asked);
  CHECK(taskid != 0);
  CHECK_INT(asked, TASK_FULLSCRAPE_TPB_BINARY);

  mutex_workqueue_canceltask(43);
  mutex_workqueue_canceltask(44);
}

static void case_every_task_gets_its_own_id(void) {
  ot_tasktype asked;
  ot_taskid   first, second;

  test_case("two queued tasks are handed out under different ids");
  CHECK_INT(mutex_workqueue_pushtask(45, TASK_FULLSCRAPE), 0);
  CHECK_INT(mutex_workqueue_pushtask(46, TASK_FULLSCRAPE), 0);

  asked  = TASK_FULLSCRAPE;
  first  = mutex_workqueue_poptask(&asked);
  asked  = TASK_FULLSCRAPE;
  second = mutex_workqueue_poptask(&asked);
  CHECK(first != 0);
  CHECK(second != 0);
  CHECK(first != second);

  mutex_workqueue_canceltask(45);
  mutex_workqueue_canceltask(46);
}

static void case_cancelled_task_is_not_handed_out(void) {
  ot_tasktype asked;

  test_case("a cancelled task is gone before any worker sees it");
  CHECK_INT(mutex_workqueue_pushtask(47, TASK_FULLSCRAPE_TPB_ASCII_PLUS), 0);
  CHECK_INT(mutex_workqueue_pushtask(48, TASK_FULLSCRAPE), 0);
  mutex_workqueue_canceltask(47);

  asked = TASK_FULLSCRAPE;
  CHECK(mutex_workqueue_poptask(&asked) != 0);
  CHECK_INT(asked, TASK_FULLSCRAPE);
  mutex_workqueue_canceltask(48);
}

void test_mutex(void) {
  case_task_goes_to_its_own_class();
  case_flags_survive_the_queue();
  case_classes_do_not_cross();
  case_every_task_gets_its_own_id();
  case_cancelled_task_is_not_handed_out();
}