forked from Mirrors/opensbi
5c992a115a
Move all of the SBIUnit-related code into the lib/sbi/tests directory. Update 'Makefile' to index objects from the tests subdirectory. I don't think creating the full separate list of Makefile variables (libsbitests-objs-path-y, libsbitests-object-mks, etc. as it is done for libsbiutils) is necessary for the tests because: 1) `lib/sbi/tests/objects.mk` is already indexed into 'libsbi-objects-mks' since the find expression for the libsbi-object-mks variable looks for objects.mk files in the nested directories as well). 2) Tests are tightly coupled with the `lib/sbi/` sources, therefore it may be reasonable to store the list of lib/sbi and lib/sbi/tests object files together in the libsbi-objs-path-y variable. Additionally, update relative paths in the tests where necessary. Signed-off-by: Ivan Orlov <ivan.orlov0322@gmail.com> Reviewed-by: Anup Patel <anup@brainfault.org>
108 lines
3.1 KiB
C
108 lines
3.1 KiB
C
/*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*
|
|
* Author: Ivan Orlov <ivan.orlov0322@gmail.com>
|
|
*/
|
|
#include <sbi/riscv_locks.h>
|
|
#include <sbi/sbi_console.h>
|
|
#include <sbi/sbi_unit_test.h>
|
|
|
|
#define TEST_CONSOLE_BUF_LEN 1024
|
|
|
|
static const struct sbi_console_device *old_dev;
|
|
static char test_console_buf[TEST_CONSOLE_BUF_LEN];
|
|
static u32 test_console_buf_pos;
|
|
static spinlock_t test_console_lock = SPIN_LOCK_INITIALIZER;
|
|
|
|
static void test_console_putc(char c)
|
|
{
|
|
test_console_buf[test_console_buf_pos] = c;
|
|
test_console_buf_pos = (test_console_buf_pos + 1) % TEST_CONSOLE_BUF_LEN;
|
|
}
|
|
|
|
static void clear_test_console_buf(void)
|
|
{
|
|
test_console_buf_pos = 0;
|
|
test_console_buf[0] = '\0';
|
|
}
|
|
|
|
static const struct sbi_console_device test_console_dev = {
|
|
.name = "Test console device",
|
|
.console_putc = test_console_putc,
|
|
};
|
|
|
|
/* Mock the console device */
|
|
static inline void test_console_begin(const struct sbi_console_device *device)
|
|
{
|
|
old_dev = sbi_console_get_device();
|
|
sbi_console_set_device(device);
|
|
}
|
|
|
|
static inline void test_console_end(void)
|
|
{
|
|
sbi_console_set_device(old_dev);
|
|
}
|
|
|
|
static void putc_test(struct sbiunit_test_case *test)
|
|
{
|
|
clear_test_console_buf();
|
|
test_console_begin(&test_console_dev);
|
|
sbi_putc('a');
|
|
test_console_end();
|
|
SBIUNIT_ASSERT_EQ(test, test_console_buf[0], 'a');
|
|
}
|
|
|
|
#define PUTS_TEST(test, expected, str) do { \
|
|
spin_lock(&test_console_lock); \
|
|
clear_test_console_buf(); \
|
|
test_console_begin(&test_console_dev); \
|
|
sbi_puts(str); \
|
|
test_console_end(); \
|
|
SBIUNIT_ASSERT_STREQ(test, test_console_buf, expected, \
|
|
sbi_strlen(expected)); \
|
|
spin_unlock(&test_console_lock); \
|
|
} while (0)
|
|
|
|
static void puts_test(struct sbiunit_test_case *test)
|
|
{
|
|
PUTS_TEST(test, "Hello, OpenSBI!", "Hello, OpenSBI!");
|
|
PUTS_TEST(test, "Hello,\r\nOpenSBI!", "Hello,\nOpenSBI!");
|
|
}
|
|
|
|
#define PRINTF_TEST(test, expected, format, ...) do { \
|
|
spin_lock(&test_console_lock); \
|
|
clear_test_console_buf(); \
|
|
test_console_begin(&test_console_dev); \
|
|
size_t __res = sbi_printf(format, ##__VA_ARGS__); \
|
|
test_console_end(); \
|
|
SBIUNIT_ASSERT_EQ(test, __res, sbi_strlen(expected)); \
|
|
SBIUNIT_ASSERT_STREQ(test, test_console_buf, expected, \
|
|
sbi_strlen(expected)); \
|
|
spin_unlock(&test_console_lock); \
|
|
} while (0)
|
|
|
|
static void printf_test(struct sbiunit_test_case *test)
|
|
{
|
|
PRINTF_TEST(test, "Hello", "Hello");
|
|
PRINTF_TEST(test, "3 5 7", "%d %d %d", 3, 5, 7);
|
|
PRINTF_TEST(test, "Hello", "%s", "Hello");
|
|
PRINTF_TEST(test, "-1", "%d", -1);
|
|
PRINTF_TEST(test, "FF", "%X", 255);
|
|
PRINTF_TEST(test, "ff", "%x", 255);
|
|
PRINTF_TEST(test, "A", "%c", 'A');
|
|
PRINTF_TEST(test, "1fe", "%p", (void *)0x1fe);
|
|
PRINTF_TEST(test, "4294967295", "%u", 4294967295U);
|
|
PRINTF_TEST(test, "-2147483647", "%ld", -2147483647l);
|
|
PRINTF_TEST(test, "-9223372036854775807", "%lld", -9223372036854775807LL);
|
|
PRINTF_TEST(test, "18446744073709551615", "%llu", 18446744073709551615ULL);
|
|
}
|
|
|
|
static struct sbiunit_test_case console_test_cases[] = {
|
|
SBIUNIT_TEST_CASE(putc_test),
|
|
SBIUNIT_TEST_CASE(puts_test),
|
|
SBIUNIT_TEST_CASE(printf_test),
|
|
SBIUNIT_END_CASE,
|
|
};
|
|
|
|
SBIUNIT_TEST_SUITE(console_test_suite, console_test_cases);
|