public-backup/bind9
1
0
Fork 1
mirror of https://gitlab.isc.org/isc-projects/bind9.git synced 2025-12-05 18:57:04 -06:00
Code Issues Packages Projects Releases Wiki Activity

Test counter increase

Browse Source
This commit is contained in:
Alessio Podda
2025-12-04 16:38:43 +01:00
parent 149411c0cc
commit 9cab11633f
1 changed files with 60 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
tests/isc/stats_test.c
View File

@@ -22,11 +22,16 @@
#define UNIT_TESTING
#include <cmocka.h>
#include <isc/async.h>
#include <isc/atomic.h>
#include <isc/lib.h>
#include <isc/loop.h>
#include <isc/mem.h>
#include <isc/result.h>
#include <isc/stats.h>
#include <isc/statsmulti.h>
#include <isc/time.h>
#include <isc/timer.h>
#include <isc/util.h>
#include <tests/isc.h>
@@ -141,10 +146,65 @@ ISC_RUN_TEST_IMPL(isc_statsmulti_basic) {
isc_statsmulti_detach(&stats);
}
/* test statsmulti with multiple threads */
static isc_statsmulti_t *mt_stats = NULL;
static atomic_uint_fast32_t mt_workers_completed = 0;
static int mt_counter_id = 0; /* Global counter ID */
#define MT_INCREMENTS_PER_THREAD 100000
static void
mt_increment_worker(void *arg ISC_ATTR_UNUSED) {
/* Do exactly 100,000 increments */
for (int i = 0; i < MT_INCREMENTS_PER_THREAD; i++) {
isc_statsmulti_increment(mt_stats, mt_counter_id);
}
/* Signal completion and check if we're the last one */
uint32_t completed = atomic_fetch_add(&mt_workers_completed, 1) + 1;
if (completed == isc_loopmgr_nloops()) {
/* Last worker shuts down the loop manager */
isc_loopmgr_shutdown();
}
}
static void
mt_setup_workers(void *arg ISC_ATTR_UNUSED) {
/* Start workers on each loop */
for (size_t i = 0; i < isc_loopmgr_nloops(); i++) {
isc_async_run(isc_loop_get(i), mt_increment_worker, NULL);
}
}
ISC_RUN_TEST_IMPL(isc_statsmulti_multithread) {
atomic_store(&mt_workers_completed, 0);
/* Create stats with 1 additive counter */
isc_statsmulti_create(isc_g_mctx, &mt_stats, 1, 0);
isc_loop_setup(isc_loop_main(), mt_setup_workers, NULL);
isc_loopmgr_run();
/* Check results - should be exactly threads * increments per thread */
uint64_t actual_count = isc_statsmulti_get_counter(mt_stats, 0);
uint64_t expected_total = (uint64_t)isc_loopmgr_nloops() * MT_INCREMENTS_PER_THREAD;
/* Verify no increments were lost */
assert_int_equal(actual_count, expected_total);
assert_true(actual_count > 0);
/* Verify all workers completed */
assert_int_equal(atomic_load(&mt_workers_completed), isc_loopmgr_nloops());
/* Cleanup */
isc_statsmulti_detach(&mt_stats);
}
ISC_TEST_LIST_START
ISC_TEST_ENTRY(isc_stats_basic)
ISC_TEST_ENTRY(isc_statsmulti_basic)
ISC_TEST_ENTRY_CUSTOM(isc_statsmulti_multithread, setup_loopmgr, teardown_loopmgr)
ISC_TEST_LIST_END