feat: add CCTests command

jellydn · Dec 23, 2023 · 1c7de7e · 1c7de7e
1 parent 199561c
commit 1c7de7e
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/lua/CopilotChat/init.lua b/lua/CopilotChat/init.lua
@@ -8,6 +8,16 @@ M.setup = function()
   utils.create_cmd('CChatExplain', function(opts)
     vim.cmd('CChat Explain how it works')
   end, { nargs = '*', range = true })
+
+  -- Add new command to generate unit tests with CopilotChat for selected text
+  utils.create_cmd('CChatTests', function(opts)
+    local cmd = 'CChat Generate unit tests for the code.'
+    -- Append the provided arguments to the command if any
+    if opts.args then
+      cmd = cmd .. ' ' .. opts.args
+    end
+    vim.cmd(cmd)
+  end, { nargs = '*', range = true })
 end
 
 return M