import io.kotest.assertions.throwables.shouldThrowWithMessage import io.kotest.core.spec.style.FunSpec import io.kotest.matchers.collections.shouldBeEmpty import io.kotest.matchers.shouldBe import prog8.code.core.* import prog8.code.target.VMTarget import prog8.intermediate.* import prog8.vm.VirtualMachine import prog8.vm.VmRunner class TestVm: FunSpec( { fun getTestOptions(): CompilationOptions { val target = VMTarget() return CompilationOptions( OutputType.RAW, CbmPrgLauncherType.NONE, ZeropageType.DONTUSE, zpReserved = emptyList(), floats = true, noSysInit = false, compTarget = target, loadAddress = target.machine.PROGRAM_LOAD_ADDRESS ) } test("vm execution: empty program") { val program = IRProgram("test", IRSymbolTable(null), getTestOptions(), VMTarget()) val vm = VirtualMachine(program) vm.callStack.shouldBeEmpty() vm.valueStack.shouldBeEmpty() vm.pc shouldBe 0 vm.stepCount shouldBe 0 vm.run() vm.callStack.shouldBeEmpty() vm.valueStack.shouldBeEmpty() vm.pc shouldBe 0 vm.stepCount shouldBe 1 } test("vm execution: modify memory") { val program = IRProgram("test", IRSymbolTable(null), getTestOptions(), VMTarget()) val block = IRBlock("testmain", null, IRBlock.BlockAlignment.NONE, Position.DUMMY) val startSub = IRSubroutine("testmain.testsub", emptyList(), null, Position.DUMMY) val code = IRCodeChunk(Position.DUMMY) code += IRInstruction(Opcode.NOP) code += IRInstruction(Opcode.LOAD, IRDataType.WORD, reg1=1, value=12345) code += IRInstruction(Opcode.STOREM, IRDataType.WORD, reg1=1, value=1000) code += IRInstruction(Opcode.RETURN) startSub += code block += startSub program.addBlock(block) val vm = VirtualMachine(program) vm.memory.getUW(1000) shouldBe 0u vm.callStack.shouldBeEmpty() vm.valueStack.shouldBeEmpty() vm.pc shouldBe 0 vm.stepCount shouldBe 0 vm.run() vm.memory.getUW(1000) shouldBe 12345u vm.callStack.shouldBeEmpty() vm.valueStack.shouldBeEmpty() vm.pc shouldBe code.instructions.size-1 vm.stepCount shouldBe code.instructions.size } test("vm asmsub not supported") { val program = IRProgram("test", IRSymbolTable(null), getTestOptions(), VMTarget()) val block = IRBlock("testmain", null, IRBlock.BlockAlignment.NONE, Position.DUMMY) val startSub = IRSubroutine("testmain.testsub", emptyList(), null, Position.DUMMY) val code = IRCodeChunk(Position.DUMMY) code += IRInstruction(Opcode.BINARYDATA, binaryData = listOf(1u,2u,3u)) code += IRInstruction(Opcode.RETURN) startSub += code block += startSub program.addBlock(block) val vm = VirtualMachine(program) shouldThrowWithMessage("An operation is not implemented: BINARYDATA not yet supported in VM") { vm.run() } } test("vm asmbinary not supported") { val program = IRProgram("test", IRSymbolTable(null), getTestOptions(), VMTarget()) val block = IRBlock("main", null, IRBlock.BlockAlignment.NONE, Position.DUMMY) val startSub = IRAsmSubroutine( "main.asmstart", Position.DUMMY, 0x2000u, emptySet(), emptyList(), emptyList(), true, "inlined asm here" ) block += startSub program.addBlock(block) shouldThrowWithMessage("vm currently does not support asmsubs: main.asmstart") { VirtualMachine(program) } } test("vmrunner") { val runner = VmRunner() val irSource=""" """ runner.runProgram(irSource) } })