feat(gotests): add generate one test

docs(gotests): add generate one test

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

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

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