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 <ivan.orlov0322@gmail.com>
Reviewed-by: Anup Patel <anup@brainfault.org>

diff --git a/lib/sbi/tests/objects.mk b/lib/sbi/tests/objects.mk
index ba588dce318f674a864732134ea97988bf40a34a..8f272897ade48a0ab31e5423d5c5713cc97c9fd1 100644 (file)
--- 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 (file)
index 0000000..7894c4e
--- /dev/null
+++ b/lib/sbi/tests/riscv_locks_test.c
@@ -0,0 +1,41 @@
+#include <sbi/sbi_unit_test.h>
+#include <sbi/riscv_locks.h>
+
+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);