test: use pytest for examples/custom_bootloader

2021-12-01 11:32:50 +08:00
parent 2cf17f802f
commit 5c3bc247cb
7 changed files with 58 additions and 41 deletions
@@ -1,23 +0,0 @@
-from __future__ import print_function
-
-import ttfw_idf
-
-
-@ttfw_idf.idf_example_test(env_tag='Example_GENERIC', target=['esp32', 'esp32s2', 'esp32c3'])
-def test_custom_bootloader_impl_example(env, _):  # type: ignore
-    # Test with default build configurations
-    dut = env.get_dut('main', 'examples/custom_bootloader/bootloader_override')
-    dut.start_app()
-
-    # Expect to read a message from the custom bootloader
-    # This message is defined in the Kconfig file, retrieve it while deleting
-    # leading and trailing quotes (")
-    welcome_message = dut.app.get_sdkconfig()['CONFIG_EXAMPLE_BOOTLOADER_WELCOME_MESSAGE'].strip("\"")
-    dut.expect(welcome_message)
-
-    # Expect to read a message from the user application
-    dut.expect('Application started!')
-
-
-if __name__ == '__main__':
-    test_custom_bootloader_impl_example()
@@ -0,0 +1,21 @@
+# SPDX-FileCopyrightText: 2021 Espressif Systems (Shanghai) CO LTD
+# SPDX-License-Identifier: CC0-1.0
+
+import pytest
+from pytest_embedded.dut import Dut
+from pytest_embedded_idf.app import IdfApp
+
+
+@pytest.mark.esp32
+@pytest.mark.esp32s2
+@pytest.mark.esp32c3
+@pytest.mark.generic
+def test_custom_bootloader_impl_example(app: IdfApp, dut: Dut) -> None:
+    # Expect to read a message from the custom bootloader
+    # This message is defined in the Kconfig file, retrieve it while deleting
+    # leading and trailing quotes (")
+    welcome_message = app.sdkconfig['EXAMPLE_BOOTLOADER_WELCOME_MESSAGE']
+    dut.expect_exact(welcome_message)
+
+    # Expect to read a message from the user application
+    dut.expect_exact('Application started!')