From e93232e15ec9fb42a6f856b612c7b64792e533c2 Mon Sep 17 00:00:00 2001 From: Simon Glass Date: Sat, 25 Nov 2017 11:57:30 -0700 Subject: [PATCH] test: overlay: Use cmd_ut_category() Adjust the code to use the common test-execution function. Signed-off-by: Simon Glass --- test/overlay/cmd_ut_overlay.c | 21 ++------------------- 1 file changed, 2 insertions(+), 19 deletions(-) diff --git a/test/overlay/cmd_ut_overlay.c b/test/overlay/cmd_ut_overlay.c index c730a11f51..6279e6d0c2 100644 --- a/test/overlay/cmd_ut_overlay.c +++ b/test/overlay/cmd_ut_overlay.c @@ -14,6 +14,7 @@ #include #include +#include /* 4k ought to be enough for anybody */ #define FDT_COPY_SIZE (4 * SZ_1K) @@ -221,7 +222,6 @@ int do_ut_overlay(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) overlay_test); const int n_ents = ll_entry_count(struct unit_test, overlay_test); struct unit_test_state *uts; - struct unit_test *test; void *fdt_base = &__dtb_test_fdt_base_begin; void *fdt_overlay = &__dtb_test_fdt_overlay_begin; void *fdt_overlay_stacked = &__dtb_test_fdt_overlay_stacked_begin; @@ -280,24 +280,7 @@ int do_ut_overlay(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) /* Apply the stacked overlay */ ut_assertok(fdt_overlay_apply(fdt_base_copy, fdt_overlay_stacked_copy)); - if (argc == 1) - printf("Running %d environment tests\n", n_ents); - - for (test = tests; test < tests + n_ents; test++) { - if (argc > 1 && strcmp(argv[1], test->name)) - continue; - printf("Test: %s\n", test->name); - - uts->start = mallinfo(); - - test->func(uts); - } - - printf("Failures: %d\n", uts->fail_count); - if (!uts->fail_count) - ret = 0; - else - ret = CMD_RET_FAILURE; + ret = cmd_ut_category("overlay", tests, n_ents, argc, argv); free(fdt_overlay_stacked_copy); err3: -- 2.39.5