Quelle fiber_context_test.cc
Sprache: C
/* Unit tests for fiber_context — the coroutine / fiber primitives used by DuckDB cross - engine predicate pushdown . Build ( standalone , from the runtime / directory ) : cc - c fiber_context . c - o fiber_context . o c + + - std = c + + 17 fiber_context_test . cc fiber_context . o - o fiber_context_test . / fiber_context_test Or via CMake : see the ADD_EXECUTABLE block in CMakeLists . txt #include "fiber_context.h" #include <cassert>#include <cstdio>#include <cstring>#include <vector>#define FIBER_STACK_SIZE (64 * 1024 )/* ---------------------------------------------------------------- Helpers static int test_count = 0 ;static int pass_count = 0 ;#define TEST(name) \static void test_## name(); \static struct Register_## name { \## name() { tests.push_back({#name , test_## name}); } \## name; \static void test_## name()#define EXPECT(cond) do { \if (!(cond)) { \" FAIL: %s:%d: %s\n" , __FILE__, __LINE__, #cond ); \else { \while (0 )struct TestEntry {const char *name;void (*func)();static std::vector<TestEntry> tests;/* ---------------------------------------------------------------- Test 1 : init / destroy — basic lifecycle struct fiber_context ctx;0 xAB, sizeof (ctx)); /* poison */ int rc = fiber_context_init(&ctx, FIBER_STACK_SIZE);0 );/* ---------------------------------------------------------------- Test 2 : spawn a fiber that runs to completion without yielding static int simple_run_flag = 0 ;static void simple_func(void *arg)int *)arg;struct fiber_context ctx;0 );0 ;int val = 42 ;/* spawn should return 0 when the user function completes without yielding */ int rc = fiber_context_spawn(&ctx, simple_func, &val);0 );42 );/* ---------------------------------------------------------------- Test 3 : spawn + single yield + continue → completion struct single_yield_data {struct fiber_context *ctx;int phase;static void single_yield_func(void *arg)auto *d = (single_yield_data *)arg;1 ;2 ;struct fiber_context ctx;0 );0 };int rc = fiber_context_spawn(&ctx, single_yield_func, &d);1 ); /* 1 = suspended */ 1 );0 ); /* 0 = completed */ 2 );/* ---------------------------------------------------------------- Test 4 : multiple yields — simulate chunk - based streaming struct multi_yield_data {struct fiber_context *ctx;int chunks_produced;static constexpr int TOTAL_CHUNKS = 5 ;static void multi_yield_func(void *arg)auto *d = (multi_yield_data *)arg;for (int i = 0 ; i < multi_yield_data::TOTAL_CHUNKS; i++)struct fiber_context ctx;0 );0 };int rc = fiber_context_spawn(&ctx, multi_yield_func, &d);1 );1 );for (int i = 2 ; i <= multi_yield_data::TOTAL_CHUNKS; i++)1 ); /* still suspended */ /* One more continue — fiber function returns */ 0 ); /* completed */ /* ---------------------------------------------------------------- Test 5 : continue on a completed context returns 0 struct fiber_context ctx;0 );int val = 1 ;int rc = fiber_context_spawn(&ctx, simple_func, &val);0 );/* Calling continue on a finished fiber should return 0 (not crash). */ 0 );/* ---------------------------------------------------------------- Test 6 : large stack usage — verify the fiber stack is adequate static void deep_stack_func(void *arg)auto *d = (single_yield_data *)arg;/* Allocate ~16KB on the fiber stack. */ volatile char buf[16384 ];char *)buf, 0 xCC, sizeof (buf));0 ] == (char )0 xCC && buf[16383 ] == (char )0 xCC) ? 1 : -1 ;2 ;struct fiber_context ctx;0 );0 };int rc = fiber_context_spawn(&ctx, deep_stack_func, &d);1 );1 );0 );2 );/* ---------------------------------------------------------------- Test 7 : two independent fibers — verify no cross - contamination struct two_fiber_data {struct fiber_context *ctx;int id;int phase;static void two_fiber_func(void *arg)auto *d = (two_fiber_data *)arg;10 + 1 ;10 + 2 ;struct fiber_context ctx_a, ctx_b;0 );0 );1 , 0 };2 , 0 };/* Spawn A, it yields at phase 11 */ int rc = fiber_context_spawn(&ctx_a, two_fiber_func, &a);1 );11 );/* Spawn B, it yields at phase 21 */ 1 );21 );/* A's state is not corrupted by B */ 11 );/* Resume B first */ 0 );22 );/* Resume A — should still work */ 0 );12 );/* ---------------------------------------------------------------- Test 8 : reuse context — spawn again after completion struct fiber_context ctx;0 );0 };int rc = fiber_context_spawn(&ctx, single_yield_func, &d1);1 && d1.phase == 1 );0 && d1.phase == 2 );/* Spawn again on the same context */ 0 };1 && d2.phase == 1 );0 && d2.phase == 2 );/* ---------------------------------------------------------------- Test 9 : simulate DuckDB scan pattern — chunked row streaming struct scan_sim_data {struct fiber_context *ctx;int total_rows;int chunk_size;int rows_produced;int yields;static void scan_sim_func(void *arg)auto *d = (scan_sim_data *)arg;int count = 0 ;for (int i = 0 ; i < d->total_rows; i++)if (count >= d->chunk_size)0 ;/* Final partial chunk — no yield, just return */ struct fiber_context ctx;0 );1000 , 128 , 0 , 0 };int rc = fiber_context_spawn(&ctx, scan_sim_func, &d);int continues = 0 ;while (rc == 1 )0 );1000 );7 ); /* 1000/128 = 7 full chunks, remainder returns */ 7 );/* ---------------------------------------------------------------- main int main()"fiber_context unit tests\n" );"========================\n" );#ifdef FIBER_CONTEXT_DISABLE"SKIPPED: fiber context is disabled on this platform.\n" );return 0 ;#endif int failures = 0 ;for (auto &t : tests)int before = test_count;int before_pass = pass_count;" %-30s " , t.name);int ran = test_count - before;int passed = pass_count - before_pass;if (passed == ran)"OK (%d checks)\n" , ran);else "FAILED (%d/%d)\n" , passed, ran);"------------------------\n" );"%d/%d tests passed, %d/%d checks passed\n" ,int )tests.size() - failures, (int )tests.size(),return failures ? 1 : 0 ;
Messung V0.5 in Prozent C=93 H=88 G=90
¤ Dauer der Verarbeitung: 0.3 Sekunden
(vorverarbeitet am 2026-10-08)
¤ *© Formatika GbR, Deutschland
2026-10-09
Neuigkeiten Aktuelles Motto des Tages Open Source Software
Jenseits des Üblichen ....
Besucherstatistik
Statistik #Sources=1126438 #Domains=1867298
