Ignore Early-Canceled Tests Example

import openhtf as htf
from openhtf.core import test_record
from openhtf.output.callbacks import json_factory
from openhtf.plugs import user_input
from openhtf.util import console_output

DEFAULT_DUT_ID = '<UNSET_DUT_ID>'


class CustomOutputToJSON(json_factory.OutputToJSON):

    def __call__(self, record):
        if (record.outcome == test_record.Outcome.ABORTED and
                record.dut_id == DEFAULT_DUT_ID):
            console_output.cli_print('Test was aborted at test start. Skipping output to JSON.')
        else:
            console_output.cli_print('Outputting test record to JSON.')
            super().__call__(record)


@htf.plug(user=user_input.UserInput)
def HelloWorldPhase(test, user):
    test.logger.info('Hello World!')
    user.prompt('The DUT ID is `%s`. Press enter to continue.' % test.test_record.dut_id)


def main():
    test = htf.Test(HelloWorldPhase)
    test.configure(default_dut_id=DEFAULT_DUT_ID)
    test.add_output_callbacks(CustomOutputToJSON('./{dut_id}.hello_world.json', indent=2))
    test.execute(test_start=user_input.prompt_for_test_start())


if __name__ == '__main__':
    main()

$ python ignore_early_canceled_tests.py
Enter a DUT ID in order to start the test.
--> ^C
Test was aborted at test start. Skipping output to JSON.

===================== test: openhtf_test  outcome: ABORTED =====================