#!/usr/bin/env python

import ttfw_idf
from tiny_test_fw import Utility

mem_test = [
    ['IRAM0_SRAM', 'WRX'],
    ['IRAM0_RTCFAST', 'WRX'],
    ['DRAM0_SRAM', 'WR'],
    ['DRAM0_RTCFAST', 'WR'],
    ['PERI1_RTCSLOW', 'WR'],
    ['PERI2_RTCSLOW_0', 'WRX'],
    ['PERI2_RTCSLOW_1', 'WRX']
]


@ttfw_idf.idf_custom_test(env_tag='Example_GENERIC', target='esp32s2', group='test-apps')
def test_memprot(env, extra_data):

    dut = env.get_dut('memprot', 'tools/test_apps/system/memprot')
    dut.start_app()

    for i in mem_test:
        if 'R' in i[1]:
            dut.expect(i[0] + ' read low: OK')
            dut.expect(i[0] + ' read high: OK')
        if 'W' in i[1]:
            dut.expect(i[0] + ' write low: OK')
            dut.expect(i[0] + ' write high: OK')
        if 'X' in i[1]:
            dut.expect(i[0] + ' exec low: OK')
            dut.expect(i[0] + ' exec high: OK')

    Utility.console_log('Memprot test done')


if __name__ == '__main__':
    test_memprot()