1 files changed, 143 insertions, 0 deletions
diff --git a/plugins/editorconfig/tests/init.lua b/plugins/editorconfig/tests/init.lua
new file mode 100644
index 0000000..654067b
--- /dev/null
+++ b/plugins/editorconfig/tests/init.lua
@@ -0,0 +1,143 @@
+local Parser = require "plugins.editorconfig.parser"
+
+local tests = {}
+
+---@class tests.test
+---@field name string Name of test
+---@field config string Path to config file
+---@field in_match string A path to test against the config
+---@field out_match string A regex to match against the result
+
+---Registered tests
+---@type tests.test[]
+tests.list = {}
+
+--- parsers cache
+---@type table<string,plugins.editorconfig.parser>
+local parsers = {}
+setmetatable(parsers, {
+  __index = function(t, k)
+    local v = rawget(t, k)
+    if v == nil then
+      v = Parser.new(k)
+      rawset(t, k, v)
+    end
+    return v
+  end
+})
+
+---Adds color to given text on non windows systems.
+---@param text string
+---@param color "red" | "green" | "yellow"
+---@return string colorized_text
+local function colorize(text, color)
+  if PLATFORM ~= "Windows" then
+    if color == "green" then
+      return "\27[92m"..text.."\27[0m"
+    elseif color == "red" then
+      return "\27[91m"..text.."\27[0m"
+    elseif color == "yellow" then
+      return "\27[93m"..text.."\27[0m"
+    end
+  end
+  return text
+end
+
+local PASSED = colorize("PASSED", "green")
+local FAILED = colorize("FAILED", "red")
+
+---Runs an individual test (executed by tests.run())
+---@param name string Test name
+---@param config_path string Relative path to tests diretory for a [config].in
+---@param in_match string Filename to match
+---@param out_match string | table Result to match regex
+function tests.check_config(name, config_path, in_match, out_match, pos, total)
+  if type(out_match) == "string" then
+    out_match = { out_match }
+  end
+  local parser = parsers[USERDIR .. "/plugins/editorconfig/tests/" .. config_path]
+  local config = parser:getConfigString(in_match)
+  local passed = true
+  for _, match in ipairs(out_match) do
+    if not regex.match(match, config) then
+      passed = false
+      break
+    end
+  end
+  if pos then
+    pos = "[" .. pos .. "/" .. total .. "] "
+  else
+    pos = ""
+  end
+  if passed then
+    print(pos .. string.format("%s - %s - '%s': %s", name, in_match, config_path, PASSED))
+  else
+    print(pos .. string.format("%s - %s - '%s': %s", name, in_match, config_path, FAILED))
+    print(config)
+  end
+  return passed
+end
+
+---Register a new test to be run later.
+---@param name string Test name
+---@param config_path string Relative path to tests diretory for a [config].in
+---@param in_match string Filename to match
+---@param out_match string | table Result to match regex
+function tests.add(name, config_path, in_match, out_match)
+  table.insert(tests.list, {
+    name = name,
+    config = config_path,
+    in_match = in_match,
+    out_match = out_match
+  })
+end
+
+---Runs all registered tests and outputs the results to terminal.
+function tests.run()
+  print "========================================================="
+  print "Running Tests"
+  print "========================================================="
+  local failed = 0
+  local passed = 0
+  local total = #tests.list
+  for i, test in ipairs(tests.list) do
+    local res = tests.check_config(
+      test.name, test.config, test.in_match, test.out_match, i, total
+    )
+    if res then passed = passed + 1 else failed = failed + 1 end
+  end
+  print "========================================================="
+  print (
+    string.format(
+      "%s %s %s",
+      colorize("Total tests: " .. #tests.list, "yellow"),
+      colorize("Passed: " .. passed, "green"),
+      colorize("Failed: " .. failed, "red")
+    )
+  )
+  print "========================================================="
+end
+
+function tests.add_parser(config_path)
+  return parsers[config_path]
+end
+
+function tests.run_parsers()
+  print "========================================================="
+  print "Running Parsers"
+  print "========================================================="
+
+  for config, parser in pairs(parsers) do
+    print "---------------------------------------------------------"
+    print(string.format("%s results:", config))
+    for _, section in ipairs(parser.sections) do
+      print(string.format("\nPath expression: %s", section.rule.expression))
+      print(string.format("Regex: %s", section.rule.regex))
+      print(string.format("Negation: %s", section.rule.negation and "true" or "false"))
+      print(string.format("Ranges: %s\n", section.rule.ranges and #section.rule.ranges or "0"))
+    end
+    print "---------------------------------------------------------"
+  end
+end
+
+return tests