server-skynet-source-3rd-je.../test/unit/thread_event.c
2020-03-10 09:31:55 -07:00

51 lines
1.3 KiB
C

#include "test/jemalloc_test.h"
TEST_BEGIN(test_next_event_fast) {
tsd_t *tsd = tsd_fetch();
te_ctx_t ctx;
te_ctx_get(tsd, &ctx, true);
te_ctx_last_event_set(&ctx, 0);
te_ctx_current_bytes_set(&ctx, TE_NEXT_EVENT_FAST_MAX - 8U);
te_ctx_next_event_set(tsd, &ctx, TE_NEXT_EVENT_FAST_MAX);
#define E(event, condition, is_alloc) \
if (is_alloc && condition) { \
event##_event_wait_set(tsd, TE_NEXT_EVENT_FAST_MAX); \
}
ITERATE_OVER_ALL_EVENTS
#undef E
/* Test next_event_fast rolling back to 0. */
void *p = malloc(16U);
assert_ptr_not_null(p, "malloc() failed");
free(p);
/* Test next_event_fast resuming to be equal to next_event. */
void *q = malloc(SC_LOOKUP_MAXCLASS);
assert_ptr_not_null(q, "malloc() failed");
free(q);
}
TEST_END
TEST_BEGIN(test_event_rollback) {
tsd_t *tsd = tsd_fetch();
const uint64_t diff = TE_MAX_INTERVAL >> 2;
size_t count = 10;
uint64_t thread_allocated = thread_allocated_get(tsd);
while (count-- != 0) {
te_alloc_rollback(tsd, diff);
uint64_t thread_allocated_after = thread_allocated_get(tsd);
assert_u64_eq(thread_allocated - thread_allocated_after, diff,
"thread event counters are not properly rolled back");
thread_allocated = thread_allocated_after;
}
}
TEST_END
int
main(void) {
return test(
test_next_event_fast,
test_event_rollback);
}