1 files changed, 76 insertions, 0 deletions
diff --git a/libcr_ipc/tests/test_sema.c b/libcr_ipc/tests/test_sema.c
new file mode 100644
index 0000000..435c01a
--- /dev/null
+++ b/libcr_ipc/tests/test_sema.c
@@ -0,0 +1,76 @@
+/* libcr_ipc/tests/test_sema.c - Tests for <libcr_ipc/sema.h>
+ *
+ * Copyright (C) 2025  Luke T. Shumaker <lukeshu@lukeshu.com>
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ */
+
+#include <libcr/coroutine.h>
+
+#define IMPLEMENTATION_FOR_LIBCR_IPC_SEMA_H YES /* so we can access .cnt */
+#include <libcr_ipc/sema.h>
+
+#include "test.h"
+
+int counter = 0;
+
+COROUTINE first_cr(void *_sema) {
+	cr_sema_t *sema = _sema;
+	cr_begin();
+
+	cr_sema_wait(sema);
+	counter++;
+	cr_sema_signal(sema);
+
+	cr_exit();
+}
+
+COROUTINE second_cr(void *_sema) {
+	cr_sema_t *sema = _sema;
+	cr_begin();
+
+	cr_sema_signal(sema); /* should be claimed by cr_first, which has been waiting */
+	cr_sema_wait(sema); /* should block, because cr_first claimed it */
+	test_assert(counter == 1);
+
+	cr_exit();
+}
+
+COROUTINE producer_cr(void *_sema) {
+	cr_sema_t *sema = _sema;
+	cr_begin();
+
+	for (int i = 0; i < 10; i++)
+		cr_sema_signal(sema);
+
+	cr_end();
+}
+
+COROUTINE consumer_cr(void *_sema) {
+	cr_sema_t *sema = _sema;
+	cr_begin();
+
+	for (int i = 0; i < 5; i++)
+		cr_sema_wait(sema);
+
+	cr_end();
+}
+
+int main() {
+	cr_sema_t sema = {};
+
+	printf("== test 1 =========================================\n");
+	coroutine_add("first", first_cr, &sema);
+	coroutine_add("second", second_cr, &sema);
+	coroutine_main();
+	test_assert(sema.cnt == 0);
+
+	printf("== test 2 =========================================\n");
+	coroutine_add("consumer", consumer_cr, &sema);
+	coroutine_add("producer", producer_cr, &sema);
+	coroutine_main();
+	coroutine_add("consumer", consumer_cr, &sema);
+	coroutine_main();
+	test_assert(sema.cnt == 0);
+
+	return 0;
+}