diff options
| author | Richard Qin <richard.qin.x@gmail.com> | 2026-07-10 11:31:46 +0800 |
|---|---|---|
| committer | Richard Qin <richard.qin.x@gmail.com> | 2026-07-10 11:31:46 +0800 |
| commit | e90b72ec3082588b296956730bd2930cf8d5d671 (patch) | |
| tree | 36ff6916af9b4176ea5a9662219d11e721f0b691 /test/test_waitqueue.cc | |
| parent | 1767d390fbf3d0c65389de46caf0b6d43ef3b6ad (diff) | |
| download | lume-e90b72ec3082588b296956730bd2930cf8d5d671.tar.gz lume-e90b72ec3082588b296956730bd2930cf8d5d671.tar.xz | |
refactor: promote new architecture to repository root
Archive the deprecated implementation under lume-legacy, make the new kernel the default root-level build, and update repository entry points without changing kernel behavior.
Diffstat (limited to 'test/test_waitqueue.cc')
| -rw-r--r-- | test/test_waitqueue.cc | 91 |
1 files changed, 91 insertions, 0 deletions
diff --git a/test/test_waitqueue.cc b/test/test_waitqueue.cc new file mode 100644 index 0000000..7f50d6b --- /dev/null +++ b/test/test_waitqueue.cc @@ -0,0 +1,91 @@ +/* SPDX-License-Identifier: GPL-2.0-or-later */ +/* Copyright (C) 2026 Richard Qin */ + +#include <lume/selftest.h> +#include <lume/waitqueue.h> +#include <lume/task.h> +#include <lume/sched.h> + +static WaitQueue g_test_wq; +static volatile bool g_thread_woken = false; +static volatile bool g_thread_started = false; +static volatile int g_wake_count = 0; + +extern TaskControlBlock* kthread_create(const char* name, void (*entry)(void*), void* arg, uint8 priority); + +static void waitqueue_test_thread(void*) +{ + g_thread_started = true; + g_test_wq.sleep(); + g_thread_woken = true; + + extern void task_exit(int code); + task_exit(0); +} + +static void waitqueue_test_thread_all(void*) +{ + g_test_wq.sleep(); + __sync_fetch_and_add(&g_wake_count, 1); + + extern void task_exit(int code); + task_exit(0); +} + +void selftest_waitqueue() +{ + st_begin("waitqueue: sleep and wake_one"); + + g_test_wq.init("test_wq"); + g_thread_started = false; + g_thread_woken = false; + + TaskControlBlock* child = kthread_create("test_wq_child", waitqueue_test_thread, nullptr, 0); + ST_ASSERT(child != nullptr); + + sched_yield(); + + ST_ASSERT(g_thread_started == true); + ST_ASSERT(g_thread_woken == false); + ST_ASSERT(!g_test_wq.is_empty()); + + kprintf("[test_wq] waking child\n"); + bool woken = g_test_wq.wake_one(); + ST_ASSERT(woken == true); + ST_ASSERT(g_test_wq.is_empty()); + + kprintf("[test_wq] yielding to child again\n"); + sched_yield(); + kprintf("[test_wq] returned from yield 2\n"); + + ST_ASSERT(g_thread_woken == true); + + st_pass(); + + st_begin("waitqueue: wake_all"); + g_wake_count = 0; + for (int i = 0; i < 3; i++) { + kthread_create("test_wq_all", waitqueue_test_thread_all, nullptr, 0); + } + + /* Give all 3 threads a chance to start and sleep */ + kprintf("[test_wq] yielding 1 for wake_all\n"); + sched_yield(); + sched_yield(); + sched_yield(); + + ST_ASSERT(!g_test_wq.is_empty()); + + int woken_count = g_test_wq.wake_all(); + + ST_ASSERT_EQ(woken_count, 3); + ST_ASSERT(g_test_wq.is_empty()); + + /* Give all 3 threads a chance to wake up and run */ + sched_yield(); + sched_yield(); + sched_yield(); + + ST_ASSERT_EQ(g_wake_count, 3); + st_pass(); +} |
