105 lines
2.9 KiB
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})
|