diff --git a/README.md b/README.md index 058f9b8..3452a9d 100644 --- a/README.md +++ b/README.md @@ -3,7 +3,6 @@ Minimalistic plugin for Go development in Neovim written in Lua. It's not an LSP tool, the main goal of this plugin add go tooling support in neovim. - ## Install Pre-dependency: [go](https://github.com/golang/go) (tested on 1.17 and 1.18) @@ -32,6 +31,7 @@ This will install next tools: - [gomodifytags](https://github.com/fatih/gomodifytags) - [impl](https://github.com/josharian/impl) +- [gotests](https://github.com/cweill/gotests) 2. Modify struct tags: By default be added/removed `json` tag, if not set. @@ -78,7 +78,15 @@ Example of usage: :GoImpl io.Reader ``` -5. Run `go generate` command +6. Generate tests with [gotests](https://github.com/cweill/gotests) + +Generate one test for spesific function/method + +```vim +:GoTestAdd +``` + +7. Run `go generate` command ```vim " Run `go generate` in cwd path diff --git a/lua/gopher/_utils/ts/init.lua b/lua/gopher/_utils/ts/init.lua index 3fbd1ff..91455c1 100644 --- a/lua/gopher/_utils/ts/init.lua +++ b/lua/gopher/_utils/ts/init.lua @@ -3,6 +3,8 @@ local M = { querys = { struct_block = [[((type_declaration (type_spec name:(type_identifier) @struct.name type: (struct_type)))@struct.declaration)]], em_struct_block = [[(field_declaration name:(field_identifier)@struct.name type: (struct_type)) @struct.declaration]], + method_name = [[((method_declaration receiver: (parameter_list)@method.receiver name: (field_identifier)@method.name body:(block))@method.declaration)]], + func = [[((function_declaration name: (identifier)@function.name) @function.declaration)]], }, } @@ -31,4 +33,15 @@ function M.get_struct_node_at_pos(row, col, bufnr) end end +function M.get_func_method_node_at_pos(row, col, bufnr) + local query = M.querys.func .. " " .. M.querys.method_name + local bufn = bufnr or vim.api.nvim_get_current_buf() + local ns = nodes.nodes_at_cursor(query, get_name_defaults(), bufn, row, col) + if ns == nil then + print "func not found" + else + return ns[#ns] + end +end + return M diff --git a/lua/gopher/gotests.lua b/lua/gopher/gotests.lua new file mode 100644 index 0000000..55ba454 --- /dev/null +++ b/lua/gopher/gotests.lua @@ -0,0 +1,48 @@ +local Job = require "plenary.job" +local ts_utils = require "gopher._utils.ts" +local M = {} + +---@param cmd_args table +local function run(cmd_args) + Job + :new({ + command = "gotests", + args = cmd_args, + on_exit = function(_, retval) + if retval ~= 0 then + print("command exited with code " .. retval) + return + end + + print "unit test(s) generated" + end, + }) + :start() +end + +---@param args table +local function add_test(args) + local fpath = vim.fn.expand "%" ---@diagnostic disable-line: missing-parameter + table.insert(args, "-w") + table.insert(args, fpath) + run(args) +end + +---generate unit test for one function +---@param parallel boolean +function M.func_test(parallel) + local ns = ts_utils.get_func_method_node_at_pos(unpack(vim.api.nvim_win_get_cursor(0))) + if ns == nil or ns.name == nil then + print "cursor on func/method and execute the command again" + return + end + + local cmd_args = { "-only", ns.name } + if parallel then + table.insert(cmd_args, "-parallel") + end + + add_test(cmd_args) +end + +return M diff --git a/lua/gopher/init.lua b/lua/gopher/init.lua index dddb7f2..6961470 100644 --- a/lua/gopher/init.lua +++ b/lua/gopher/init.lua @@ -1,4 +1,5 @@ local tags = require "gopher.struct_tags" +local gotests = require "gopher.gotests" local gopher = {} gopher.install_deps = require "gopher.installer" @@ -8,5 +9,6 @@ gopher.mod = require "gopher.gomod" gopher.get = require "gopher.goget" gopher.impl = require "gopher.impl" gopher.generate = require "gopher.gogenerate" +gopher.test_add = gotests.one_test return gopher diff --git a/plugin/gopher.vim b/plugin/gopher.vim index 6e1224f..a919cfb 100644 --- a/plugin/gopher.vim +++ b/plugin/gopher.vim @@ -1,5 +1,6 @@ command! -nargs=* GoTagAdd :lua require"gopher".tags_add() command! -nargs=* GoTagRm :lua require"gopher".tags_rm() +command! -nargs=* GoTestAdd :lua require"gopher".test_add() command! -nargs=* GoMod :lua require"gopher".mod() command! -nargs=* GoGet :lua require"gopher".get() command! -nargs=* GoImpl :lua require"gopher".impl()