import os, sys, inspect
import traceback
sys.path.insert(1, os.path.join(sys.path[0], '..'))

import infernal

################################################################################

tests = []

def add_test(name, result, register, code):
	tests.append((name, result, register, code))

def printf (str, *args):
	print(str.format(*args))

def execute_tests():
	print("Executing tests!")
	total_tests = 0
	failed_tests = 0
	error_tests = 0
	for name, result, register, code in tests:
		total_tests += 1
		line_nr = None
		try:
			emu = infernal.Emulator(code)
			emu.setStack("junk...", "calling eip")
			emu.setRegs( rip = 0, rbp = 'old bp')
		except infernal.CodeParseException as e:
			error_tests += 1
			printf("Encountered error when parsing {}, at line {}: {}",
					name, e.line_nr, e.str)
		try:
			for line_nr in emu:
				pass
			if isinstance(result, BaseException):
				printf("Error should have happened in {}, but did not", name)
				failed_tests += 1
			output = emu.getVal(register)
			if output != result:
				failed_tests += 1
				printf("Failed in {}. {} was {}, should be {}",
						name, register, result, output)
		except BaseException as e:
			if not isinstance(e, result):
				error_tests += 1
				printf("Encountered error in {}, at operation {}: {}",
						name, emu.getVal('%rip'), e)
				traceback.print_exc()
	printf("Tests done! {}/{}.",
		total_tests - failed_tests - error_tests, total_tests)
	printf("{} failed, and {} encountered a python error",
		failed_tests, error_tests)

################################################################################

add_test("constant $255", 255, "%rsi", """
movq $255, %rsi
""")

add_test("static addition 10+$20", 30, "%rsi", """
movq $10, %rsi
addq $20, %rsi
""")

add_test("register addition 10+20", 30, "%rsi", """
movq $10, %rsi
movq $20, %rax
addq %rax, %rsi
""")

################################################################################

add_test("invalid comparison 1", infernal.JunkComparisonException, "None", """
start:	movq $100, %rsp   # Set stack pointer to a random position.
		popq %rsi		  # Move a Junk value into %rsi
		cmpq $10, %rsi    # Do a Junk comparison, which triggers the `i` comp
						  # virtual register.
		jg start		  # Attempt to do a jump to the start, if (Junk-10)>0,
						  # which makes no sense, and thus throws an error.
""")

add_test("invalid addition 1", infernal.JunkComparisonException, "None", """
start:	movq $100, %rsp   # Set stack pointer to a random position.
		popq %rsi		  # Move a Junk value into %rsi
		addq %rsi, %rsp   # Adds Junk to 101, which produces Junk.
		cmpq $10, %rsp    # Do a Junk comparison, which triggers the `i` comp
						  # virtual register.
		jg start		  # Attempt to do a jump to the start, if (Junk-10)>0,
						  # which makes no sense, and thus throws an error.
""")

add_test("invalid addition 2", infernal.JunkComparisonException, "None", """
start:	movq $100, %rsp   # Set stack pointer to a random position.
		popq %rsi		  # Move a Junk value into %rsi
		addq %rsp, %rsi   # Adds 101 to Junk, which produces Junk.
		cmpq $10, %rsi    # Do a Junk comparison, which triggers the `i` comp
						  # virtual register.
		jg start		  # Attempt to do a jump to the start, if (Junk-10)>0,
						  # which makes no sense, and thus throws an error.
""")

add_test("invalid subtraction 1", infernal.JunkComparisonException, "None", """
start:	movq $100, %rsp   # Set stack pointer to a random position.
		popq %rsi		  # Move a Junk value into %rsi
		subq %rsi, %rsp   # Adds Junk to 101, which produces Junk.
		cmpq $10, %rsp    # Do a Junk comparison, which triggers the `i` comp
						  # virtual register.
		jg start		  # Attempt to do a jump to the start, if (Junk-10)>0,
						  # which makes no sense, and thus throws an error.
""")

add_test("invalid subtraction 2", infernal.JunkComparisonException, "None", """
start:	movq $100, %rsp   # Set stack pointer to a random position.
		popq %rsi		  # Move a Junk value into %rsi
		subq %rsp, %rsi   # Adds 101 to Junk, which produces Junk.
		cmpq $10, %rsi    # Do a Junk comparison, which triggers the `i` comp
						  # virtual register.
		jg start		  # Attempt to do a jump to the start, if (Junk-10)>0,
						  # which makes no sense, and thus throws an error.
""")

################################################################################

if __name__ == "__main__":
	execute_tests()