libuv/test/test-timeout.c

/* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to
 * deal in the Software without restriction, including without limitation the
 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
 * sell copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
 * IN THE SOFTWARE.
 */

#include "../oio.h"
#include "test.h"


static int expected = 0;
static int timeouts = 0;

static int64_t start_time;

static void timeout_cb(oio_req *req) {
  ASSERT(req != NULL);
  free(req);
  timeouts++;

  /* Just call this randomly for the code coverage. */
  oio_update_time();
}

static void exit_timeout_cb(oio_req *req) {
  int64_t now = oio_now();
  ASSERT(req != NULL);
  ASSERT(timeouts == expected);
  ASSERT(start_time < now);
  exit(0);
}

static void dummy_timeout_cb(oio_req *req) {
  /* Should never be called */
  FATAL("dummy_timer_cb should never be called");
}


TEST_IMPL(timeout) {
  oio_req *req;
  oio_req exit_req;
  oio_req dummy_req;
  int i;

  oio_init();

  start_time = oio_now();
  ASSERT(0 < start_time);

  /* Let 10 timers time out in 500 ms total. */
  for (i = 0; i < 10; i++) {
    req = (oio_req*)malloc(sizeof(*req));
    ASSERT(req != NULL);

    oio_req_init(req, NULL, timeout_cb);

    if (oio_timeout(req, i * 50) < 0) {
      FATAL("oio_timeout failed");
    }

    expected++;
  }

  /* The 11th timer exits the test and runs after 1 s. */
  oio_req_init(&exit_req, NULL, exit_timeout_cb);
  if (oio_timeout(&exit_req, 1000) < 0) {
    FATAL("oio_timeout failed");
  }

  /* The 12th timer should never run. */
  oio_req_init(&dummy_req, NULL, dummy_timeout_cb);
  if (oio_timeout(&dummy_req, 2000)) {
    FATAL("oio_timeout failed");
  }

  oio_run();

  FATAL("should never get here");
  return 2;
}