test/py: add TPMv2.x test suite
[oweals/u-boot.git] / test / py / tests / test_efi_selftest.py
1 # SPDX-License-Identifier: GPL-2.0
2 # Copyright (c) 2017, Heinrich Schuchardt <xypron.glpk@gmx.de>
3
4 # Test efi API implementation
5
6 import pytest
7 import u_boot_utils
8
9 @pytest.mark.buildconfigspec('cmd_bootefi_selftest')
10 def test_efi_selftest(u_boot_console):
11         """
12         Run bootefi selftest
13         """
14
15         u_boot_console.run_command(cmd='setenv efi_selftest')
16         u_boot_console.run_command(cmd='bootefi selftest', wait_for_prompt=False)
17         m = u_boot_console.p.expect(['Summary: 0 failures', 'Press any key'])
18         if m != 0:
19                 raise Exception('Failures occured during the EFI selftest')
20         u_boot_console.run_command(cmd='', wait_for_echo=False, wait_for_prompt=False);
21         m = u_boot_console.p.expect(['resetting', 'U-Boot'])
22         if m != 0:
23                 raise Exception('Reset failed during the EFI selftest')
24         u_boot_console.restart_uboot();
25
26 @pytest.mark.buildconfigspec('cmd_bootefi_selftest')
27 @pytest.mark.buildconfigspec('of_control')
28 def test_efi_selftest_device_tree(u_boot_console):
29         u_boot_console.run_command(cmd='setenv efi_selftest list')
30         output = u_boot_console.run_command('bootefi selftest')
31         assert '\'device tree\'' in output
32         u_boot_console.run_command(cmd='setenv efi_selftest device tree')
33         u_boot_console.run_command(cmd='setenv -f serial# Testing DT')
34         u_boot_console.run_command(cmd='bootefi selftest ${fdtcontroladdr}', wait_for_prompt=False)
35         m = u_boot_console.p.expect(['serial-number: Testing DT', 'U-Boot'])
36         if m != 0:
37                 raise Exception('Reset failed in \'device tree\' test')
38         u_boot_console.restart_uboot();
39
40 @pytest.mark.buildconfigspec('cmd_bootefi_selftest')
41 def test_efi_selftest_watchdog_reboot(u_boot_console):
42         u_boot_console.run_command(cmd='setenv efi_selftest list')
43         output = u_boot_console.run_command('bootefi selftest')
44         assert '\'watchdog reboot\'' in output
45         u_boot_console.run_command(cmd='setenv efi_selftest watchdog reboot')
46         u_boot_console.run_command(cmd='bootefi selftest', wait_for_prompt=False)
47         m = u_boot_console.p.expect(['resetting', 'U-Boot'])
48         if m != 0:
49                 raise Exception('Reset failed in \'watchdog reboot\' test')
50         u_boot_console.restart_uboot();