Rewrite

test/run.lua

@@ -0,0 +1,204 @@
+local lfs = require("lfs")
+local anselme = require("anselme")
+local ser = require("test.ser")
+local inspect = require("test.inspect")
+
+local function format_text(t, prefix)
+	prefix = prefix or " "
+	local r = ""
+	for _, l in ipairs(t) do
+		r = r .. prefix
+		local tags = ""
+		for k, v in ipairs(l.tags) do
+			tags = tags .. ("[%q]=%q"):format(k, v)
+		end
+		if tags ~= "" then
+			r = r .. ("[%s]%s"):format(tags, l.data)
+		else
+			r = r .. l.data
+		end
+	end
+	return r
+end
+
+local function compare(a, b)
+	if type(a) == "table" and type(b) == "table" then
+		for k, v in pairs(a) do
+			if not compare(v, b[k]) then
+				return false
+			end
+		end
+		for k, v in pairs(b) do
+			if not compare(v, a[k]) then
+				return false
+			end
+		end
+		return true
+	else
+		return a == b
+	end
+end
+
+-- parse args
+local args = {}
+local i=1
+while i <= #arg do
+	if arg[i+1] and not arg[i+1]:match("^%-%-") then
+		args[arg[i]:gsub("^%-%-", "")] = arg[i+1]
+		i = i + 2
+	else
+		args[arg[i]:gsub("^%-%-", "")] = true
+		i = i + 1
+	end
+end
+
+-- list tests
+local files = {}
+for item in lfs.dir("test/tests/") do
+	if item:match("%.ans$") and item:match(args.filter or "") then
+		table.insert(files, "test/tests/"..item)
+	end
+end
+table.sort(files)
+
+-- test script
+if args.script then
+	local vm = anselme()
+	local state, err = vm:loadfile("test.ans", "test")
+	if state then
+		local istate, e = vm:run("test")
+		if not istate then
+			print("error", e)
+		else
+			repeat
+				local t, d = istate:step()
+				if t == "text" then
+					print(format_text(d))
+				elseif t == "choice" then
+					print(format_text(d, "\n> "))
+					istate:choose(io.read())
+				else
+					print(t, inspect(d))
+				end
+			until t == "return" or t == "error"
+		end
+	else
+		print("error", err)
+	end
+-- run tests
+else
+	local total, success = #files, 0
+	for _, file in ipairs(files) do
+		local filebase = file:match("^(.*)%.ans$")
+		local namespace = filebase:match("([^/]*)$")
+		math.randomseed(0)
+		local vm = anselme()
+		vm:loadalias {
+			seen = "👁️",
+			checkpoint = "🏁"
+		}
+		vm:loadfunction {
+			-- custom event test
+			["wait"] = {
+				{
+					arity = 1, types = { "number" },
+					value = function(duration)
+						coroutine.yield("wait", duration)
+					end
+				}
+			},
+			-- run another function in parallel
+			["run"] = {
+				{
+					arity =  1, types = { "string" },
+					value = function(str)
+						local istate, e = anselme.running.vm:run(str, anselme.running:current_namespace())
+						if not istate then coroutine.yield("error", e) end
+						local event, data = istate:step()
+						coroutine.yield(event, data)
+					end
+				}
+			},
+			-- manual choice
+			choose = {
+				{
+					arity = 1, types = { "number" },
+					value = function(c)
+						anselme.running:choose(c)
+					end
+				}
+			},
+			-- manual interrupt
+			interrupt = {
+				{
+					arity = 1, types = { "string" },
+					value = function(str)
+						anselme.running:interrupt(str)
+						coroutine.yield("wait", 0)
+					end
+				},
+				{
+					arity = 0,
+					value = function()
+						anselme.running:interrupt()
+						coroutine.yield("wait", 0)
+					end
+				}
+			}
+		}
+		local state, err = vm:loadfile(file, namespace)
+
+		local result = {}
+		if state then
+			local istate, e = vm:run(namespace)
+			if not istate then
+				table.insert(result, { "error", e })
+			else
+				repeat
+					local t, d = istate:step()
+					table.insert(result, { t, d })
+				until t == "return" or t == "error"
+			end
+		else
+			table.insert(result, { "error", err })
+		end
+
+		if args.write then
+			local o = assert(io.open(filebase..".lua", "w"))
+			o:write(ser(result))
+			o:write("\n--[[\n")
+			for _, v in ipairs(result) do
+				o:write(inspect(v).."\n")
+			end
+			o:write("]]--")
+			o:close()
+		else
+			local o, e = loadfile(filebase..".lua")
+			if o then
+				local output = o()
+				if not compare(result, output) then
+					if not args.silent then
+						print("> "..namespace)
+						print(inspect(result))
+						print("is not equal to")
+						print(inspect(output))
+					end
+				else
+					success = success + 1
+				end
+			else
+				if not args.silent then
+					print("> "..namespace)
+					print(e)
+					print("result was:")
+					print(inspect(result))
+				end
+			end
+		end
+	end
+	if args.write then
+		print("Wrote test results.")
+	else
+		print(("%s/%s tests success."):format(success, total))
+	end
+end