1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Tests that the PCI Maximum Payload Size (MPS) command can set the sandbox
4  * PCI Express device to safe mode and determine the correct payload size.
5  *
6  * Copyright 2023 Microsoft
7  * Written by Stephen Carlson <stcarlso@linux.microsoft.com>
8  */
9 
10 #include <common.h>
11 #include <console.h>
12 #include <test/suites.h>
13 #include <test/ut.h>
14 
15 #define PCI_MPS_TEST(_name, _flags) UNIT_TEST(_name, _flags, pci_mps_test)
16 
17 /* Test "pci_mps" command in safe "s" mode */
test_pci_mps_safe(struct unit_test_state * uts)18 static int test_pci_mps_safe(struct unit_test_state *uts)
19 {
20 	/* Enumerate PCI Express first */
21 	ut_assertok(run_command("pci e", 0));
22 	ut_assert_console_end();
23 
24 	/* Test pci_mps s */
25 	ut_assertok(run_command("pci_mps s", 0));
26 	ut_assert_nextline("Setting MPS of all devices to 256B");
27 	ut_assert_console_end();
28 
29 	return 0;
30 }
31 
32 PCI_MPS_TEST(test_pci_mps_safe, UT_TESTF_CONSOLE_REC);
33 
do_ut_pci_mps(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])34 int do_ut_pci_mps(struct cmd_tbl *cmdtp, int flag, int argc,
35 		  char * const argv[])
36 {
37 	struct unit_test *tests = UNIT_TEST_SUITE_START(pci_mps_test);
38 	const int n = UNIT_TEST_SUITE_COUNT(pci_mps_test);
39 
40 	return cmd_ut_category("cmd_pci_mps", "pci_mps_test_", tests, n,
41 			       argc, argv);
42 }
43