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
    dut.expect('Custom bootloader has been initialized correctly.')

    # Expect to read a message from the user application
    dut.expect('Application started!')


if __name__ == '__main__':
    test_custom_bootloader_impl_example()