""" This module contains functions used by the test cases to hide the
architecture and/or the platform dependent nature of the tests. """

def check_first_register_readable(test_case):
    if test_case.getArchitecture() in ['x86_64', 'i386']:
        test_case.expect("register read eax", substrs = ['eax = 0x'])
    elif test_case.getArchitecture() in ['arm']:
    	test_case.expect("register read r0", substrs = ['r0 = 0x'])
    elif test_case.getArchitecture() in ['aarch64']:
        test_case.expect("register read x0", substrs = ['x0 = 0x'])
    else:
        # TODO: Add check for other architectures
        test_case.fail("Unsupported architecture for test case (arch: %s)" % test_case.getArchitecture())