openssl/test/event_queue_test.c
Pauli 0eb2765943 Add unit test for event queue
Reviewed-by: Tomas Mraz <tomas@openssl.org>
Reviewed-by: Matt Caswell <matt@openssl.org>
(Merged from https://github.com/openssl/openssl/pull/18345)
2022-07-06 10:33:56 +10:00

102 lines
3.3 KiB
C

/*
* Copyright 2022 The OpenSSL Project Authors. All Rights Reserved.
*
* Licensed under the Apache License 2.0 (the "License"). You may not use
* this file except in compliance with the License. You can obtain a copy
* in the file LICENSE in the source distribution or at
* https://www.openssl.org/source/license.html
*/
#include "internal/event_queue.h"
#include "internal/nelem.h"
#include "testutil.h"
static OSSL_TIME cur_time = 100;
OSSL_TIME ossl_time_now(void)
{
return cur_time;
}
#define PAYLOAD(s) s, strlen(s) + 1
static int event_test(void)
{
int res = 0;
size_t len = 0;
OSSL_EVENT *e1, *e2, e3, *e4 = NULL, *ep = NULL;
OSSL_EVENT_QUEUE *q = NULL;
void *p;
static char payload[] = "payload";
/* Create an event queue and add some events */
if (!TEST_ptr(q = ossl_event_queue_new())
|| !TEST_ptr(e1 = ossl_event_queue_add_new(q, 1, 10, 1100, "ctx 1",
PAYLOAD(payload)))
|| !TEST_ptr(e2 = ossl_event_queue_add_new(q, 2, 5, 1100, "ctx 2",
PAYLOAD("data")))
|| !TEST_true(ossl_event_queue_add(q, &e3, 3, 20, 1200, "ctx 3",
PAYLOAD("more data")))
|| !TEST_ptr(e4 = ossl_event_queue_add_new(q, 2, 5, 1150, "ctx 2",
PAYLOAD("data")))
/* Verify some event details */
|| !TEST_uint_eq(ossl_event_get_type(e1), 1)
|| !TEST_uint_eq(ossl_event_get_priority(e1), 10)
|| !TEST_uint64_t_eq(ossl_event_get_when(e1), 1100)
|| !TEST_str_eq(ossl_event_get0_ctx(e1), "ctx 1")
|| !TEST_ptr(p = ossl_event_get0_payload(e1, &len))
|| !TEST_str_eq((char *)p, payload)
|| !TEST_uint64_t_eq(ossl_event_time_until(&e3), 1100)
|| !TEST_uint64_t_eq(ossl_event_queue_time_until_next(q), 1000)
/* Modify an event's time */
|| !TEST_true(ossl_event_queue_postpone_until(q, e1, 1200))
|| !TEST_uint64_t_eq(ossl_event_get_when(e1), 1200)
|| !TEST_true(ossl_event_queue_remove(q, e4)))
goto err;
ossl_event_free(e4);
/* Execute the queue */
cur_time = 1000;
if (!TEST_true(ossl_event_queue_get1_next_event(q, &ep))
|| !TEST_ptr_null(ep))
goto err;
cur_time = 1100;
if (!TEST_true(ossl_event_queue_get1_next_event(q, &ep))
|| !TEST_ptr_eq(ep, e2))
goto err;
ossl_event_free(ep);
ep = e2 = NULL;
if (!TEST_true(ossl_event_queue_get1_next_event(q, &ep))
|| !TEST_ptr_null(ep))
goto err;
cur_time = 1250;
if (!TEST_true(ossl_event_queue_get1_next_event(q, &ep))
|| !TEST_ptr_eq(ep, &e3))
goto err;
ossl_event_free(ep);
ep = NULL;
if (!TEST_true(ossl_event_queue_get1_next_event(q, &ep))
|| !TEST_ptr_eq(ep, e1))
goto err;
ossl_event_free(ep);
ep = e1 = NULL;
if (!TEST_true(ossl_event_queue_get1_next_event(q, &ep))
|| !TEST_ptr_null(ep))
goto err;
res = 1;
err:
ossl_event_free(ep);
ossl_event_queue_free(q);
return res;
}
int setup_tests(void)
{
ADD_TEST(event_test);
return 1;
}