local export = {}

local Array = require "Module:array"

local function collect(iter)
	local vals = Array()
	for val in iter do
		vals:insert(val)
	end
	return vals
end

local total_tests, successes = 0, 0
local function assert(val)
	total_tests = total_tests + 1
	if val then
		successes = successes + 1
	end
end

function export.run_tests()
	local ok, ret = pcall(function ()
		assert(Array(1, 2, 3):map(function (x) return x * x end):concat ", " == "1, 4, 9")
		
		assert(Array(1, 2, 3, 4, 5):filter(function (x) return x % 2 == 0 end):concat ", " == "2, 4")
		
		assert(Array():type() == "array")
		
		assert(Array(1, nil, 2, nil, 3):compress():concat ", " == "1, 2, 3")
		assert(collect(Array(1, nil, 2, nil, 3):sparseIpairs()):concat ", " == "1, 3, 5")
		
		assert(Array("a", "b", "c"):toSet()["a"] == true)
		assert(Array("a", "b", "c"):to_set()["a"] == true)
		
		assert(Array(2, 4, 8):all(function (x) return x % 2 == 0 end) == true)
		assert(Array(1, 2, 3):some(function (x) return x % 2 == 0 end) == true)
		
		assert(Array(1, 2, 3, nil, 4):length() == 3)
		
		assert(Array(1, 2, 3, nil, 4):contains(3) == true)
		assert(Array(1, 2, 3, nil, 4):contains(4) == false)
		
		assert(Array.keysToList(Array("a", "b", "c"):invert()):concat ", " == "a, b, c")
		
		assert(Array(1, nil, 2, nil, 3):maxIndex() == 5)
		assert(Array(1, nil, 2, nil, 3):max_index() == 5)
		
		assert(Array(1, nil, 2, nil, 3):num_keys():concat ", " == "1, 3, 5")
		
		assert(collect(Array("a", "b", "c"):reverseIpairs()):concat ", " == "3, 2, 1")
		
		assert(Array("a", "b", "c"):reverse():concat ", " == "c, b, a")
		
		assert(Array("a", "b", "c"):serial_comma_join { dontTag = true } == "a, b, and c")
		
		assert(Array(1, 1, 2, 3, 1, 2, 3):removeDuplicates():concat ", " == "1, 2, 3")
		
		assert(Array(1, 2, 3, 4, 5):slice(1, 3):concat ", " == "1, 2, 3")
		assert(Array(1, 2, 3, 4, 5):slice(1, -3):concat ", " == "1, 2, 3")
		
		assert(Array(1, 2, 3):adjust_index(-1) == 3)
		
		assert((Array(1, 2, 3) + Array(4, 5, 6)):concat ", " == "1, 2, 3, 4, 5, 6")
	end)
	
	local err
	if not ok then
		err = ret
	end
	
	local color = successes < total_tests and "red" or "green"
	
	return ('<span style="color: %s;">%d out of %d assertions passed</span>%s')
		:format(color, successes, total_tests, err and '\n\n<span class="error">' .. err .. '</span>' or "")
end

return export