#include "test/jemalloc_test.h"

static const uint64_t interval = 1 << 20;

TEST_BEGIN(test_counter_accum) {
       uint64_t increment = interval >> 4;
       unsigned n = interval / increment;
       uint64_t accum = 0;

       counter_accum_t c;
       counter_accum_init(&c, interval);

       tsd_t *tsd = tsd_fetch();
       bool trigger;
       for (unsigned i = 0; i < n; i++) {
               trigger = counter_accum(tsd_tsdn(tsd), &c, increment);
               accum += increment;
               if (accum < interval) {
                       expect_b_eq(trigger, false, "Should not trigger");
               } else {
                       expect_b_eq(trigger, true, "Should have triggered");
               }
       }
       expect_b_eq(trigger, true, "Should have triggered");
}
TEST_END

void
expect_counter_value(counter_accum_t *c, uint64_t v) {
       uint64_t accum = locked_read_u64_unsynchronized(&c->accumbytes);
       expect_u64_eq(accum, v, "Counter value mismatch");
}

#define N_THDS (16)
#define N_ITER_THD (1 << 12)
#define ITER_INCREMENT (interval >> 4)

static void *
thd_start(void *varg) {
       counter_accum_t *c = (counter_accum_t *)varg;

       tsd_t *tsd = tsd_fetch();
       bool trigger;
       uintptr_t n_triggered = 0;
       for (unsigned i = 0; i < N_ITER_THD; i++) {
               trigger = counter_accum(tsd_tsdn(tsd), c, ITER_INCREMENT);
               n_triggered += trigger ? 1 : 0;
       }

       return (void *)n_triggered;
}


TEST_BEGIN(test_counter_mt) {
       counter_accum_t shared_c;
       counter_accum_init(&shared_c, interval);

       thd_t thds[N_THDS];
       unsigned i;
       for (i = 0; i < N_THDS; i++) {
               thd_create(&thds[i], thd_start, (void *)&shared_c);
       }

       uint64_t sum = 0;
       for (i = 0; i < N_THDS; i++) {
               void *ret;
               thd_join(thds[i], &ret);
               sum += (uintptr_t)ret;
       }
       expect_u64_eq(sum, N_THDS * N_ITER_THD / (interval / ITER_INCREMENT),
           "Incorrect number of triggers");
}
TEST_END

int
main(void) {
       return test(
           test_counter_accum,
           test_counter_mt);
}