SPDX: Convert all of our single license tags to Linux Kernel style

[oweals/u-boot.git] / test / py / tests / test_efi_selftest.py

diff --git a/test/py/tests/test_efi_selftest.py b/test/py/tests/test_efi_selftest.py
index 76e282a6c7261d6b2a8009bae24285857fadaf76..747d52d935a4142af1f816f4473a3c343ede1105 100644 (file)
--- a/test/py/tests/test_efi_selftest.py
+++ b/test/py/tests/test_efi_selftest.py
@@ -1,7 +1,5 @@
-# Copyright (c) 2016, NVIDIA CORPORATION. All rights reserved.
-# Copyright (c) 2017, Heinrich Schuchardt <xypron.glpk@gmx.de>
-#
 # SPDX-License-Identifier: GPL-2.0
+# Copyright (c) 2017, Heinrich Schuchardt <xypron.glpk@gmx.de>
 
 # Test efi API implementation
 
@@ -14,6 +12,7 @@ def test_efi_selftest(u_boot_console):
        Run bootefi selftest
        """
 
+       u_boot_console.run_command(cmd='setenv efi_selftest')
        u_boot_console.run_command(cmd='bootefi selftest', wait_for_prompt=False)
        m = u_boot_console.p.expect(['Summary: 0 failures', 'Press any key'])
        if m != 0:
@@ -23,3 +22,29 @@ def test_efi_selftest(u_boot_console):
        if m != 0:
                raise Exception('Reset failed during the EFI selftest')
        u_boot_console.restart_uboot();
+
+@pytest.mark.buildconfigspec('cmd_bootefi_selftest')
+@pytest.mark.buildconfigspec('of_control')
+def test_efi_selftest_device_tree(u_boot_console):
+       u_boot_console.run_command(cmd='setenv efi_selftest list')
+       output = u_boot_console.run_command('bootefi selftest')
+       assert '\'device tree\'' in output
+       u_boot_console.run_command(cmd='setenv efi_selftest device tree')
+       u_boot_console.run_command(cmd='setenv -f serial# Testing DT')
+       u_boot_console.run_command(cmd='bootefi selftest ${fdtcontroladdr}', wait_for_prompt=False)
+       m = u_boot_console.p.expect(['serial-number: Testing DT', 'U-Boot'])
+       if m != 0:
+               raise Exception('Reset failed in \'device tree\' test')
+       u_boot_console.restart_uboot();
+
+@pytest.mark.buildconfigspec('cmd_bootefi_selftest')
+def test_efi_selftest_watchdog_reboot(u_boot_console):
+       u_boot_console.run_command(cmd='setenv efi_selftest list')
+       output = u_boot_console.run_command('bootefi selftest')
+       assert '\'watchdog reboot\'' in output
+       u_boot_console.run_command(cmd='setenv efi_selftest watchdog reboot')
+       u_boot_console.run_command(cmd='bootefi selftest', wait_for_prompt=False)
+       m = u_boot_console.p.expect(['resetting', 'U-Boot'])
+       if m != 0:
+               raise Exception('Reset failed in \'watchdog reboot\' test')
+       u_boot_console.restart_uboot();