from __future__ import unicode_literals
from ttfw_idf import Utility
import os
import ttfw_idf


@ttfw_idf.idf_example_test(env_tag="test_jtag_arm")
def test_examples_gcov(env, extra_data):

    rel_project_path = os.path.join('examples', 'system', 'gcov')
    dut = env.get_dut('gcov', rel_project_path)
    idf_path = dut.app.get_sdk_path()
    proj_path = os.path.join(idf_path, rel_project_path)
    openocd_cmd_log = os.path.join(proj_path, 'openocd_cmd.log')

    with ttfw_idf.OCDBackend(os.path.join(proj_path, 'openocd.log'), dut.app.target) as oocd:
        dut.start_app()

        def expect_counter_output(loop, timeout=10):
            dut.expect_all('blink_dummy_func: Counter = {}'.format(loop),
                           'some_dummy_func: Counter = {}'.format(loop * 2),
                           timeout=timeout)

        expect_counter_output(0, timeout=20)
        dut.expect('Ready to dump GCOV data...', timeout=5)

        def dump_coverage():
            try:
                response = oocd.cmd_exec('esp gcov dump')
                with open(openocd_cmd_log, 'a') as f:
                    f.write(response)

                assert all(x in response for x in ['Targets connected.',
                                                   'gcov_example_main.c.gcda',
                                                   'gcov_example_func.c.gcda',
                                                   'some_funcs.c.gcda',
                                                   'Targets disconnected.',
                                                   ])

            except AssertionError:
                # Print what is happening with DUT. Limit the size if it is in loop and generating output.
                Utility.console_log(dut.read(size=1000))
                raise

        dump_coverage()
        dut.expect('GCOV data have been dumped.', timeout=5)
        expect_counter_output(1)
        dut.expect('Ready to dump GCOV data...', timeout=5)
        dump_coverage()
        dut.expect('GCOV data have been dumped.', timeout=5)

        for i in range(2, 6):
            expect_counter_output(i)


if __name__ == '__main__':
    test_examples_gcov()