From 7b1ed968e4afc63d2ca54f6bdf1b42ef2b6576fa Mon Sep 17 00:00:00 2001 From: Ivan Orlov Date: Tue, 23 Apr 2024 16:52:45 +0100 Subject: lib: tests: Add test for spinlocks Implement the test which covers some of the functions from the `riscv_locks.h` file. This test consists of 3 test cases: 1) For lock/unlock functions 2) Unsuccessful trylock (the lock was previously taken) 3) Successful trylock (the lock is free and can be taken) Signed-off-by: Ivan Orlov Reviewed-by: Anup Patel --- lib/sbi/tests/objects.mk | 3 +++ lib/sbi/tests/riscv_locks_test.c | 41 ++++++++++++++++++++++++++++++++++++++++ 2 files changed, 44 insertions(+) create mode 100644 lib/sbi/tests/riscv_locks_test.c diff --git a/lib/sbi/tests/objects.mk b/lib/sbi/tests/objects.mk index ba588dc..8f27289 100644 --- a/lib/sbi/tests/objects.mk +++ b/lib/sbi/tests/objects.mk @@ -9,3 +9,6 @@ libsbi-objs-$(CONFIG_SBIUNIT) += tests/sbi_console_test.o carray-sbi_unit_tests-$(CONFIG_SBIUNIT) += atomic_test_suite libsbi-objs-$(CONFIG_SBIUNIT) += tests/riscv_atomic_test.o + +carray-sbi_unit_tests-$(CONFIG_SBIUNIT) += locks_test_suite +libsbi-objs-$(CONFIG_SBIUNIT) += tests/riscv_locks_test.o diff --git a/lib/sbi/tests/riscv_locks_test.c b/lib/sbi/tests/riscv_locks_test.c new file mode 100644 index 0000000..7894c4e --- /dev/null +++ b/lib/sbi/tests/riscv_locks_test.c @@ -0,0 +1,41 @@ +#include +#include + +static spinlock_t test_lock = SPIN_LOCK_INITIALIZER; + +static void spin_lock_test(struct sbiunit_test_case *test) +{ + /* We don't want to accidentally get locked */ + SBIUNIT_ASSERT(test, !spin_lock_check(&test_lock)); + + spin_lock(&test_lock); + SBIUNIT_EXPECT(test, spin_lock_check(&test_lock)); + spin_unlock(&test_lock); + + SBIUNIT_ASSERT(test, !spin_lock_check(&test_lock)); +} + +static void spin_trylock_fail(struct sbiunit_test_case *test) +{ + /* We don't want to accidentally get locked */ + SBIUNIT_ASSERT(test, !spin_lock_check(&test_lock)); + + spin_lock(&test_lock); + SBIUNIT_EXPECT(test, !spin_trylock(&test_lock)); + spin_unlock(&test_lock); +} + +static void spin_trylock_success(struct sbiunit_test_case *test) +{ + SBIUNIT_EXPECT(test, spin_trylock(&test_lock)); + spin_unlock(&test_lock); +} + +static struct sbiunit_test_case locks_test_cases[] = { + SBIUNIT_TEST_CASE(spin_lock_test), + SBIUNIT_TEST_CASE(spin_trylock_fail), + SBIUNIT_TEST_CASE(spin_trylock_success), + SBIUNIT_END_CASE, +}; + +SBIUNIT_TEST_SUITE(locks_test_suite, locks_test_cases); -- cgit v1.1