You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
go.nvim/lua/go/gotests.lua

86 lines
2.3 KiB
Lua

-- Table driven tests based on its target source files' function and method signatures.
-- https://github.com/cweill/gotests
local ut = {}
local gotests = "gotests"
local test_dir = vim.g.go_nvim_test_dir or ""
local test_template = vim.go_nvim_test_template or ""
local utils = require("go").utils
local run = function(setup)
print(vim.inspect(setup))
local j =
vim.fn.jobstart(
setup,
{
on_stdout = function(jobid, data, event)
print("unit tests generate " .. vim.inspect(data))
end,
on_stderr = function(_, data, _)
print("failed to generate tests for " .. vim.inspect(setup) .. "error: " .. vim.inspect(data))
end
}
)
end
local add_test = function(args)
require("go.install").install(gotests)
if string.len(test_template) > 1 then
table.insert(args, "-template")
table.insert(args, test_template)
if string.len(test_dir) > 1 then
table.insert(args, "-template_dir")
table.insert(args, test_dir)
end
end
local row, col = unpack(vim.api.nvim_win_get_cursor(0))
row, col = row + 1, col + 1
local ns = require("go.ts.go").get_func_method_node_at_pos(row, col)
if ns == nil or ns == {} then
return
end
utils.log("parnode" .. vim.inspect(ns))
run(args)
end
ut.fun_test = function(parallel)
parallel = parallel or false
local row, col = unpack(vim.api.nvim_win_get_cursor(0))
row, col = row + 1, col + 1
local ns = require("go.ts.go").get_func_method_node_at_pos(row, col)
if ns == nil or ns == {} then
return
end
-- utils.log("parnode" .. vim.inspect(ns))
local funame = ns.name
-- local rs, re = ns.dim.s.r, ns.dim.e.r
local gofile = vim.fn.expand("%")
local args = {gotests, "-w", "-only", funame, gofile}
if parallel then
table.insert(args, "-parallel")
end
add_test(args)
end
ut.all_test = function(parallel)
parallel = parallel or false
local gofile = vim.fn.expand("%")
local args = {gotests, "-all", "-w", gofile}
if parallel then
table.insert(args, "-parallel")
end
add_test(args)
end
ut.exported_test = function(parallel)
parallel = parallel or false
local gofile = vim.fn.expand("%")
local args = {gotests, "-exported", "-w", gofile}
if parallel then
table.insert(args, "-parallel")
end
add_test(args)
end
return ut