[people/ms/u-boot.git] / test / cmd_ut.c

/*
 * (C) Copyright 2015
 * Joe Hershberger, National Instruments, joe.hershberger@ni.com
 *
 * SPDX-License-Identifier:	GPL-2.0
 */

#include <common.h>
#include <command.h>
#include <test/suites.h>
#include <test/test.h>

static int do_ut_all(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]);

int cmd_ut_category(const char *name, struct unit_test *tests, int n_ents,
		    int argc, char * const argv[])
{
	struct unit_test_state uts = { .fail_count = 0 };
	struct unit_test *test;

	if (argc == 1)
		printf("Running %d %s tests\n", n_ents, name);

	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);

	return uts.fail_count ? CMD_RET_FAILURE : 0;
}

static cmd_tbl_t cmd_ut_sub[] = {
	U_BOOT_CMD_MKENT(all, CONFIG_SYS_MAXARGS, 1, do_ut_all, "", ""),
#if defined(CONFIG_UT_DM)
	U_BOOT_CMD_MKENT(dm, CONFIG_SYS_MAXARGS, 1, do_ut_dm, "", ""),
#endif
#if defined(CONFIG_UT_ENV)
	U_BOOT_CMD_MKENT(env, CONFIG_SYS_MAXARGS, 1, do_ut_env, "", ""),
#endif
#ifdef CONFIG_UT_OVERLAY
	U_BOOT_CMD_MKENT(overlay, CONFIG_SYS_MAXARGS, 1, do_ut_overlay, "", ""),
#endif
#ifdef CONFIG_UT_TIME
	U_BOOT_CMD_MKENT(time, CONFIG_SYS_MAXARGS, 1, do_ut_time, "", ""),
#endif
#ifdef CONFIG_SANDBOX
	U_BOOT_CMD_MKENT(compression, CONFIG_SYS_MAXARGS, 1, do_ut_compression,
			 "", ""),
#endif
};

static int do_ut_all(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	int i;
	int retval;
	int any_fail = 0;

	for (i = 1; i < ARRAY_SIZE(cmd_ut_sub); i++) {
		printf("----Running %s tests----\n", cmd_ut_sub[i].name);
		retval = cmd_ut_sub[i].cmd(cmdtp, flag, 1, &cmd_ut_sub[i].name);
		if (!any_fail)
			any_fail = retval;
	}

	return any_fail;
}

static int do_ut(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	cmd_tbl_t *cp;

	if (argc < 2)
		return CMD_RET_USAGE;

	/* drop initial "ut" arg */
	argc--;
	argv++;

	cp = find_cmd_tbl(argv[0], cmd_ut_sub, ARRAY_SIZE(cmd_ut_sub));

	if (cp)
		return cp->cmd(cmdtp, flag, argc, argv);

	return CMD_RET_USAGE;
}

#ifdef CONFIG_SYS_LONGHELP
static char ut_help_text[] =
	"all - execute all enabled tests\n"
#ifdef CONFIG_UT_DM
	"ut dm [test-name]\n"
#endif
#ifdef CONFIG_UT_ENV
	"ut env [test-name]\n"
#endif
#ifdef CONFIG_UT_OVERLAY
	"ut overlay [test-name]\n"
#endif
#ifdef CONFIG_UT_TIME
	"ut time - Very basic test of time functions\n"
#endif
#ifdef CONFIG_SANDBOX
	"ut compression - Test compressors and bootm decompression\n"
#endif
	;
#endif

U_BOOT_CMD(
	ut, CONFIG_SYS_MAXARGS, 1, do_ut,
	"unit tests", ut_help_text
);
Commit	Line	Data
c617ede0 JH	1	/*
	2	* (C) Copyright 2015
	3	* Joe Hershberger, National Instruments, joe.hershberger@ni.com
	4	*
	5	* SPDX-License-Identifier: GPL-2.0
	6	*/
	7
	8	#include <common.h>
	9	#include <command.h>
	10	#include <test/suites.h>
4d869c1e	11	#include <test/test.h>
c617ede0 JH	12
	13	static int do_ut_all(cmd_tbl_t cmdtp, int flag, int argc, char const argv[]);
	14
4d869c1e SG	15	int cmd_ut_category(const char name, struct unit_test tests, int n_ents,
	16	int argc, char * const argv[])
	17	{
	18	struct unit_test_state uts = { .fail_count = 0 };
	19	struct unit_test *test;
	20
	21	if (argc == 1)
	22	printf("Running %d %s tests\n", n_ents, name);
	23
	24	for (test = tests; test < tests + n_ents; test++) {
	25	if (argc > 1 && strcmp(argv[1], test->name))
	26	continue;
	27	printf("Test: %s\n", test->name);
	28
	29	uts.start = mallinfo();
	30
	31	test->func(&uts);
	32	}
	33
	34	printf("Failures: %d\n", uts.fail_count);
	35
	36	return uts.fail_count ? CMD_RET_FAILURE : 0;
	37	}
	38
c617ede0 JH	39	static cmd_tbl_t cmd_ut_sub[] = {
c617ede0 JH	40	U_BOOT_CMD_MKENT(all, CONFIG_SYS_MAXARGS, 1, do_ut_all, "", ""),
40441e0b JH	41	#if defined(CONFIG_UT_DM)
	42	U_BOOT_CMD_MKENT(dm, CONFIG_SYS_MAXARGS, 1, do_ut_dm, "", ""),
	43	#endif
421f86f3 JH	44	#if defined(CONFIG_UT_ENV)
	45	U_BOOT_CMD_MKENT(env, CONFIG_SYS_MAXARGS, 1, do_ut_env, "", ""),
	46	#endif
f2a9942f MR	47	#ifdef CONFIG_UT_OVERLAY
	48	U_BOOT_CMD_MKENT(overlay, CONFIG_SYS_MAXARGS, 1, do_ut_overlay, "", ""),
	49	#endif
c812f722 JH	50	#ifdef CONFIG_UT_TIME
	51	U_BOOT_CMD_MKENT(time, CONFIG_SYS_MAXARGS, 1, do_ut_time, "", ""),
	52	#endif
0aac10f2 SG	53	#ifdef CONFIG_SANDBOX
	54	U_BOOT_CMD_MKENT(compression, CONFIG_SYS_MAXARGS, 1, do_ut_compression,
	55	"", ""),
	56	#endif
c617ede0 JH	57	};
	58
	59	static int do_ut_all(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
	60	{
	61	int i;
	62	int retval;
	63	int any_fail = 0;
	64
	65	for (i = 1; i < ARRAY_SIZE(cmd_ut_sub); i++) {
	66	printf("----Running %s tests----\n", cmd_ut_sub[i].name);
	67	retval = cmd_ut_sub[i].cmd(cmdtp, flag, 1, &cmd_ut_sub[i].name);
	68	if (!any_fail)
	69	any_fail = retval;
	70	}
	71
	72	return any_fail;
	73	}
	74
	75	static int do_ut(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
	76	{
	77	cmd_tbl_t *cp;
	78
	79	if (argc < 2)
	80	return CMD_RET_USAGE;
	81
	82	/* drop initial "ut" arg */
	83	argc--;
	84	argv++;
	85
	86	cp = find_cmd_tbl(argv[0], cmd_ut_sub, ARRAY_SIZE(cmd_ut_sub));
	87
	88	if (cp)
	89	return cp->cmd(cmdtp, flag, argc, argv);
	90
	91	return CMD_RET_USAGE;
	92	}
	93
	94	#ifdef CONFIG_SYS_LONGHELP
	95	static char ut_help_text[] =
	96	"all - execute all enabled tests\n"
40441e0b JH	97	#ifdef CONFIG_UT_DM
40441e0b JH	98	"ut dm [test-name]\n"
c812f722	99	#endif
421f86f3 JH	100	#ifdef CONFIG_UT_ENV
	101	"ut env [test-name]\n"
	102	#endif
f2a9942f MR	103	#ifdef CONFIG_UT_OVERLAY
	104	"ut overlay [test-name]\n"
	105	#endif
c812f722 JH	106	#ifdef CONFIG_UT_TIME
c812f722 JH	107	"ut time - Very basic test of time functions\n"
0aac10f2 SG	108	#endif
	109	#ifdef CONFIG_SANDBOX
	110	"ut compression - Test compressors and bootm decompression\n"
40441e0b	111	#endif
c617ede0 JH	112	;
	113	#endif
	114
	115	U_BOOT_CMD(
	116	ut, CONFIG_SYS_MAXARGS, 1, do_ut,
	117	"unit tests", ut_help_text
	118	);