feat(summary): count tests

Author: rcarriga

doc/neotest.txt

@@ -161,6 +161,7 @@ Default values:
     },
     summary = {
       animated = true,
+      count = true,
       enabled = true,
       expand_errors = true,
       follow = true,
@@ -299,6 +300,7 @@ Fields~
 window
 {open} `(string)` | fun(): integer A command or function to open a window for
 the summary
+{count} `(boolean)` Display number of tests found beside the adapter name
 
                                                *neotest.Config.summary.mappings*
 Fields~
@@ -1324,9 +1326,9 @@ https://github.com/neovim/neovim/blob/master/runtime/lua/vim/treesitter/language
 Inherits: `neotest.lib.positions.ParseOptions`
 
 Fields~
-{fast} `(boolean)` Use faster parsing (Should be unchanged unless injections
+{fast?} `(boolean)` Use faster parsing (Should be unchanged unless injections
 are needed)
-{build_position} `(fun(file_path: string, source: string, captured_nodes:
+{build_position?} `(fun(file_path: string, source: string, captured_nodes:
 table<string, userdata>): neotest.Position|neotest.Position[]|nil)` Builds one
 or more positions from the captured nodes from a query match.
 
@@ -1409,7 +1411,7 @@ outputs. Use vim.jobstart instead.
 
 Parameters~
 {command} `(string[])`
-{args} `(neotest.lib.process.RunArgs)`
+{args?} `(neotest.lib.process.RunArgs)`
 Return~
 `(integer,neotest.lib.process.RunResult)` Exit code and table containing stdout/stderr keys if requested
 

lua/neotest/config/init.lua

@@ -99,6 +99,7 @@ local js_watch_query = [[
 ---@field expand_errors boolean Expand all failed positions
 ---@field mappings neotest.Config.summary.mappings Buffer mappings for summary window
 ---@field open string | fun(): integer A command or function to open a window for the summary
+---@field count boolean Display number of tests found beside the adapter name
 
 ---@class neotest.Config.summary.mappings
 ---@field expand string|string[] Expand currently selected position
@@ -229,6 +230,7 @@ local default_config = {
   },
   summary = {
     enabled = true,
+    count = true,
     animated = true,
     follow = true,
     expand_errors = true,

lua/neotest/consumers/summary/summary.lua

@@ -89,16 +89,25 @@ function Summary:run()
       if self._starting then
         for _, adapter_id in ipairs(self.client:get_adapters()) do
           local tree = assert(self.client:get_position(nil, { adapter = adapter_id }))
+          local count = 0
+          if config.summary.count then
+            for _, pos in tree:iter() do
+              if pos.type == "test" then
+                count = count + 1
+              end
+            end
+          end
           canvas:write(
-            vim.split(adapter_id, ":", { trimempty = true })[1] .. "\n",
+            vim.split(adapter_id, ":", { trimempty = true })[1]
+            .. (count == 0 and "" or string.format(" %d Tests Found", count)) .. "\n",
             { group = config.highlights.adapter_name }
           )
           if tree:data().path ~= cwd then
             local root_dir = nio.fn.fnamemodify(tree:data().path, ":.")
             canvas:write(root_dir .. "\n", { group = config.highlights.dir })
           end
           self.components[adapter_id] = self.components[adapter_id]
-            or SummaryComponent(self.client, adapter_id)
+              or SummaryComponent(self.client, adapter_id)
           if config.summary.animated then
             if self.components[adapter_id]:render(canvas, tree, all_expanded, self.focused) then
               self.render_ready.set()

lua/neotest/lib/process/init.lua

@@ -22,7 +22,7 @@ neotest.lib.process = {}
 --- outputs. Use vim.jobstart instead.
 ---@async
 ---@param command string[]
----@param args neotest.lib.process.RunArgs
+---@param args? neotest.lib.process.RunArgs
 ---@return integer,neotest.lib.process.RunResult Exit code and table containing stdout/stderr keys if requested
 function neotest.lib.process.run(command, args)
   args = args or {}

lua/neotest/lib/treesitter/init.lua

@@ -91,8 +91,8 @@ function neotest.lib.treesitter.fast_parse(lang_tree)
 end
 
 ---@class neotest.lib.treesitter.ParseOptions : neotest.lib.positions.ParseOptions
----@field fast boolean Use faster parsing (Should be unchanged unless injections are needed)
----@field build_position fun(file_path: string, source: string, captured_nodes: table<string, userdata>): neotest.Position|neotest.Position[]|nil Builds one or more positions from the captured nodes from a query match.
+---@field fast? boolean Use faster parsing (Should be unchanged unless injections are needed)
+---@field build_position? fun(file_path: string, source: string, captured_nodes: table<string, userdata>): neotest.Position|neotest.Position[]|nil Builds one or more positions from the captured nodes from a query match.
 
 --- Build a parsed Query object from a string
 ---@param lang string