dotfiles/vim/lua/test_runner.lua

105 lines
2.9 KiB
Lua

_G._test_cmd_to_wins = {}
_G._build_test_cmd = function()
local bufnr = vim.fn.bufnr("%")
local filetype = vim.fn.getbufvar(bufnr, "&filetype")
if filetype == "" then
vim.api.nvim_err_writeln("cannot build test command for filetype:" .. filetype)
return nil
end
local path = vim.fn.expand("%:p")
-- TODO: allow line number to be passed
-- TODO: check file exists before running command
if filetype == "ruby" then
if not path:find("_spec.rb") then
path = path:gsub("/app/", "/spec/"):gsub([[.rb$]], "_spec.rb")
end
return "bundle exec rspec --format=progress --no-profile " .. path
elseif filetype == "go" then
return "go test -v " .. path:gsub("^(.*)/(.*go)$", "%1/...")
else
vim.api.nvim_err_writeln("filetype not supported: " .. filetype)
return nil
end
end
_G.run_tests = function()
local cmd = _G._build_test_cmd()
if cmd == nil then
return
end
vim.api.nvim_command([[silent up]])
local winid = _G._test_cmd_to_wins[cmd]
local current_winid = vim.fn.win_getid()
if winid == nil or not vim.api.nvim_win_is_valid(winid) or winid == current_winid or not vim.fn.win_gotoid(winid) == -1 then
vim.api.nvim_command([[10split]])
winid = vim.fn.win_getid()
_G._test_cmd_to_wins[cmd] = winid
end
local buf = vim.api.nvim_create_buf(false, true)
vim.api.nvim_buf_set_option(buf, "bufhidden", "delete")
vim.api.nvim_win_set_buf(winid, buf)
vim.fn.termopen(cmd)
vim.api.nvim_command([[normal! G]])
vim.api.nvim_command([[wincmd p]])
end
_G.focus_tests = function()
local cmd = _G._build_test_cmd()
if cmd == nil then
return
end
local winid = _G._test_cmd_to_wins[cmd]
if winid == nil or not vim.api.nvim_win_is_valid(winid) then
return
end
vim.fn.win_gotoid(winid)
vim.api.nvim_command([[wincmd =]])
end
_G.close_tests = function()
local current_winid = vim.api.nvim_get_current_win()
for _, winid in pairs(_G._test_cmd_to_wins) do
if current_winid == winid and vim.api.nvim_win_is_valid(winid) then
vim.api.nvim_win_close(winid, false)
return
end
end
local cmd = _G._build_test_cmd()
if cmd == nil then
return
end
local winid = _G._test_cmd_to_wins[cmd]
if winid == nil or not vim.api.nvim_win_is_valid(winid) then
return
end
vim.api.nvim_win_close(winid, false)
end
_G.copy_test_cmd = function()
local cmd = _G._build_test_cmd()
if cmd == nil then
return
end
vim.fn.setreg("+", cmd)
print("Copied: " .. cmd)
end
vim.api.nvim_set_keymap("n", "<leader>cr", [[:lua _G.run_tests()<cr>]], {noremap = true, silent = true})
vim.api.nvim_set_keymap("n", "<leader>cf", [[:lua _G.focus_tests()<cr>]], {noremap = true, silent = true})
vim.api.nvim_set_keymap("n", "<leader>cq", [[:lua _G.close_tests()<cr>]], {noremap = true, silent = true})
vim.api.nvim_set_keymap("n", "<leader>cc", [[:lua _G.copy_test_cmd()<cr>]], {noremap = true, silent = true})