server-skynet-source-3rd-je.../test/unit/thread_event.c
Qi Wang 97dd79db6c Implement deallocation events.
Make the event module to accept two event types, and pass around the event
context.  Use bytes-based events to trigger tcache GC on deallocation, and get
rid of the tcache ticker.
2020-02-04 00:18:15 -08:00

71 lines
1.8 KiB
C

#include "test/jemalloc_test.h"
TEST_BEGIN(test_next_event_fast_roll_back) {
tsd_t *tsd = tsd_fetch();
event_ctx_t ctx;
event_ctx_get(tsd, &ctx, true);
event_ctx_last_event_set(&ctx, 0);
event_ctx_current_bytes_set(&ctx,
THREAD_NEXT_EVENT_FAST_MAX - 8U);
event_ctx_next_event_set(tsd, &ctx,
THREAD_NEXT_EVENT_FAST_MAX);
#define E(event, condition, is_alloc) \
if (is_alloc && condition) { \
event##_event_wait_set(tsd, THREAD_NEXT_EVENT_FAST_MAX);\
}
ITERATE_OVER_ALL_EVENTS
#undef E
void *p = malloc(16U);
assert_ptr_not_null(p, "malloc() failed");
free(p);
}
TEST_END
TEST_BEGIN(test_next_event_fast_resume) {
tsd_t *tsd = tsd_fetch();
event_ctx_t ctx;
event_ctx_get(tsd, &ctx, true);
event_ctx_last_event_set(&ctx, 0);
event_ctx_current_bytes_set(&ctx,
THREAD_NEXT_EVENT_FAST_MAX + 8U);
event_ctx_next_event_set(tsd, &ctx,
THREAD_NEXT_EVENT_FAST_MAX + 16U);
#define E(event, condition, is_alloc) \
if (is_alloc && condition) { \
event##_event_wait_set(tsd, \
THREAD_NEXT_EVENT_FAST_MAX + 16U); \
}
ITERATE_OVER_ALL_EVENTS
#undef E
void *p = malloc(SC_LOOKUP_MAXCLASS);
assert_ptr_not_null(p, "malloc() failed");
free(p);
}
TEST_END
TEST_BEGIN(test_event_rollback) {
tsd_t *tsd = tsd_fetch();
const uint64_t diff = THREAD_EVENT_MAX_INTERVAL >> 2;
size_t count = 10;
uint64_t thread_allocated = thread_allocated_get(tsd);
while (count-- != 0) {
thread_alloc_event_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_roll_back,
test_next_event_fast_resume,
test_event_rollback);
}