init.lua/lua/scratch/gotest.lua(view raw)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
local gotests = {}
function gotests.switch()
local bufnr = vim.api.nvim_get_current_buf()
local fname_parts = vim.split(vim.api.nvim_buf_get_name(bufnr), "/")
local test_fname = fname_parts[#fname_parts]
if test_fname:find "_test.go" then
test_fname = test_fname:gsub("_test.go", ".go")
else
test_fname = test_fname:gsub(".go", "_test.go")
end
table.remove(fname_parts, #fname_parts)
local path = table.concat(fname_parts, "/") .. "/" .. test_fname
vim.cmd("edit " .. path)
end
return gotests
|