976cd70f4f
This will cause the system to automatically run all of the unit tests after a short pause on boot. Will be used by an automatic test script. At the moment there aren't a lot of unit tests in the list, but this should greatly increase the utility of them since they'll be automatically run.
132 lines
3.9 KiB
C
132 lines
3.9 KiB
C
/*
|
|
* Copyright (c) 2013, Google, Inc. All rights reserved
|
|
*
|
|
* Use of this source code is governed by a MIT-style
|
|
* license that can be found in the LICENSE file or at
|
|
* https://opensource.org/licenses/MIT
|
|
*/
|
|
/*
|
|
* All unit tests get registered here. A call to run_all_tests() will run
|
|
* them and provide results.
|
|
*/
|
|
#include <lib/unittest.h>
|
|
|
|
#include <assert.h>
|
|
#include <lk/console_cmd.h>
|
|
#include <lk/err.h>
|
|
#include <lk/init.h>
|
|
|
|
static struct list_node test_case_list = LIST_INITIAL_VALUE(test_case_list);
|
|
|
|
#if WITH_LIB_CONSOLE
|
|
#include <lib/console.h>
|
|
|
|
#if RUN_UNITTESTS_AT_BOOT
|
|
static void unittest_run_at_boot(uint level) {
|
|
const char run_at_boot_script[] = "sleep 5; ut all";
|
|
console_t *con = console_create(false);
|
|
if (con) {
|
|
console_run_script(con, run_at_boot_script);
|
|
// TODO: destroy console afterwards
|
|
}
|
|
}
|
|
|
|
LK_INIT_HOOK(unittest_at_boot, unittest_run_at_boot, LK_INIT_LEVEL_LAST)
|
|
#endif
|
|
#endif
|
|
|
|
/*
|
|
* Registers a test case with the unit test framework.
|
|
*/
|
|
void unittest_register_test_case(struct test_case_element *elem) {
|
|
DEBUG_ASSERT(elem);
|
|
DEBUG_ASSERT(!list_in_list(&elem->node));
|
|
list_add_tail(&test_case_list, &elem->node);
|
|
}
|
|
|
|
/*
|
|
* Runs all registered test cases.
|
|
*/
|
|
bool run_all_tests(void) {
|
|
unsigned int n_tests = 0;
|
|
unsigned int n_success = 0;
|
|
unsigned int n_failed = 0;
|
|
struct list_node failed_test_case_list = LIST_INITIAL_VALUE(failed_test_case_list);
|
|
|
|
bool all_success = true;
|
|
struct test_case_element *current;
|
|
list_for_every_entry(&test_case_list, current, struct test_case_element, node) {
|
|
if (!current->test_case()) {
|
|
list_add_tail(&failed_test_case_list, ¤t->failed_node);
|
|
all_success = false;
|
|
n_failed++;
|
|
} else {
|
|
n_success++;
|
|
}
|
|
n_tests++;
|
|
}
|
|
|
|
unittest_printf("\n====================================================\n");
|
|
unittest_printf(" CASES: %d SUCCESS: %d FAILED: %d ",
|
|
n_tests, n_success, n_failed);
|
|
unittest_printf("\n====================================================\n");
|
|
|
|
if (all_success) {
|
|
unittest_printf("\nSUCCESS! All test cases passed\n");
|
|
} else {
|
|
unittest_printf("\nFAILURE! Some test cases failed:\n");
|
|
struct test_case_element *failed;
|
|
struct test_case_element *temp;
|
|
list_for_every_entry_safe(&failed_test_case_list, failed, temp, struct test_case_element, failed_node) {
|
|
unittest_printf(" %s\n", failed->name);
|
|
list_delete(&failed->failed_node);
|
|
}
|
|
unittest_printf("\n");
|
|
}
|
|
|
|
return all_success;
|
|
}
|
|
|
|
static int do_unittests(int argc, const console_cmd_args *argv) {
|
|
|
|
if (argc < 2) {
|
|
usage:
|
|
printf("usage:\n");
|
|
printf("%s all : run all unit tests\n", argv[0].str);
|
|
printf("%s list : list all test cases\n", argv[0].str);
|
|
printf("%s <test name> : run specific test\n", argv[0].str);
|
|
return -1;
|
|
}
|
|
|
|
if (!strcmp(argv[1].str, "all")) {
|
|
bool result = run_all_tests();
|
|
printf("UNIT TEST: run_all_tests return %u\n", result);
|
|
} else if (!strcmp(argv[1].str, "list")) {
|
|
struct test_case_element *current;
|
|
list_for_every_entry(&test_case_list, current, struct test_case_element, node) {
|
|
puts(current->name);
|
|
}
|
|
} else {
|
|
// look for unit test that matches the string name
|
|
bool found_test = false;
|
|
struct test_case_element *current;
|
|
list_for_every_entry(&test_case_list, current, struct test_case_element, node) {
|
|
if (strcmp(argv[1].str, current->name) == 0) {
|
|
found_test = true;
|
|
current->test_case();
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!found_test) {
|
|
goto usage;
|
|
}
|
|
}
|
|
|
|
return NO_ERROR;
|
|
}
|
|
|
|
STATIC_COMMAND_START
|
|
STATIC_COMMAND("ut", "run some or all of the unit tests", do_unittests)
|
|
STATIC_COMMAND_END(unit_tests);
|