18-测试

return {
  { "marilari88/neotest-vitest" },
  {
    "nvim-neotest/neotest",
    opts = { adapters = { "neotest-vitest" } },
  },
}

$ mkdir -p neotest-bun/lua/neotest-bun

print("Hello, Lua!")

return {
  { dir = "~/Desktop/Code/neotest-bun/" }, -- 请确保路径指向你实际创建的 neotest-bun 目录
  {
    "nvim-neotest/neotest",
    opts = { adapters = { "neotest-bun" } },
  },
}

local BunNeotestAdapter = { name = "neotest-bun" }

-- 查找项目根目录
function BunNeotestAdapter.root(dir) end

-- 过滤不应扫描的目录
function BunNeotestAdapter.filter_dir(name, rel_path, root) end

-- 判断一个文件是否是测试文件
function BunNeotestAdapter.is_test_file(file_path) end

-- 在文件中发现测试的位置（例如，具体的 test 或 describe 块）
function BunNeotestAdapter.discover_positions(file_path) end

-- 构建运行测试所需的具体命令和上下文
function BunNeotestAdapter.build_spec(args) end

-- 解析测试运行的输出结果
function BunNeotestAdapter.results(spec, result, tree) end

return BunNeotestAdapter

local lib = require("neotest.lib") -- 引入 neotest 库

local BunNeotestAdapter = { name = "neotest-bun" }

function BunNeotestAdapter.root(dir)
  -- 使用 neotest 库函数查找包含 bun.lockb 文件的根目录
  return lib.files.match_root_pattern("bun.lockb")(dir)
end

-- ... (其他函数保持不变) ...

return BunNeotestAdapter

-- ... (root 函数如上) ...

function BunNeotestAdapter.filter_dir(name, rel_path, root)
  -- 如果目录名不是 node_modules，则返回 true（不过滤）
  return name ~= "node_modules"
end

-- ... (其他函数保持不变) ...

return BunNeotestAdapter

-- ... (root 和 filter_dir 函数如上) ...

function BunNeotestAdapter.is_test_file(file_path)
  -- 使用 Lua 的字符串匹配检查文件名是否以 .test.js 或 .test.ts 结尾
  return string.match(file_path, ".*.test.[tj]s$") ~= nil
end

-- ... (其他函数保持不变) ...

return BunNeotestAdapter

-- ... (之前的函数实现) ...
-- 确保在文件顶部引入 lib
-- local lib = require("neotest.lib")

function BunNeotestAdapter.discover_positions(file_path)
  -- Treesitter 查询，用于查找 describe (命名空间) 和 test/it (测试) 块
  local query = [[
    ; -- Namespaces --
    ; Matches: `describe('context', () => {})`
    ((call_expression
      function: (identifier) @func_name (
        #eq? @func_name "describe"
      )
      arguments: (arguments (
        string (string_fragment) @namespace.name
      ) (arrow_function))
    )) @namespace.definition
    ; Matches: `describe('context', function() {})`
    ((call_expression
      function: (identifier) @func_name (
        #eq? @func_name "describe"
      )
      arguments: (arguments (
        string (string_fragment) @namespace.name
      ) (function_expression))
    )) @namespace.definition
    ; Matches: `describe.only('context', () => {})`
    ((call_expression
      function: (member_expression
        object: (identifier) @func_name (
          #any-of? @func_name "describe"
        )
      )
      arguments: (
        arguments (string (string_fragment) @namespace.name
      ) (arrow_function))
    )) @namespace.definition
    ; Matches: `describe.only('context', function() {})`
    ((call_expression
      function: (member_expression
        object: (identifier) @func_name (
          #any-of? @func_name "describe"
        )
      )
      arguments: (arguments (
        string (string_fragment) @namespace.name
      ) (function_expression))
    )) @namespace.definition
    ; Matches: `describe.each(['data'])('context', () => {})`
    ((call_expression
      function: (call_expression
        function: (member_expression
          object: (identifier) @func_name (
            #any-of? @func_name "describe"
          )
        )
      )
      arguments: (arguments (
        string (string_fragment) @namespace.name
      ) (arrow_function))
    )) @namespace.definition
    ; Matches: `describe.each(['data'])('context', function() {})`
    ((call_expression
      function: (call_expression
        function: (member_expression
          object: (identifier) @func_name (
            #any-of? @func_name "describe"
          )
        )
      )
      arguments: (arguments (
        string (string_fragment) @namespace.name
      ) (function_expression))
    )) @namespace.definition

    ; -- Tests --
    ; Matches: `test('test') / it('test')`
    ((call_expression
      function: (identifier) @func_name (
        #any-of? @func_name "it" "test"
      )
      arguments: (arguments (
        string (string_fragment) @test.name
      ) [(arrow_function) (function_expression)])
    )) @test.definition
    ; Matches: `test.only('test') / it.only('test')`
    ((call_expression
      function: (member_expression
        object: (identifier) @func_name (
          #any-of? @func_name "test" "it"
        )
      )
      arguments: (arguments (
        string (string_fragment) @test.name
      ) [(arrow_function) (function_expression)])
    )) @test.definition
    ; Matches: `test.each(['data'])('test')
    ((call_expression
      function: (call_expression
        function: (member_expression
          object: (identifier) @func_name (
            #any-of? @func_name "it" "test"
          )
          property: (property_identifier) @each_property (
            #eq? @each_property "each"
          )
        )
      )
      arguments: (arguments (
        string (string_fragment) @test.name
      ) [(arrow_function) (function_expression)])
    )) @test.definition
  ]]

  -- 使用 neotest 库函数和 Treesitter 查询来解析文件，找到测试位置
  local positions = lib.treesitter.parse_positions(
    file_path, query, {
    nested_tests = false, -- Bun/Jest 不支持嵌套测试的标准方式
  })

  return positions
end

-- ... (build_spec 和 results 函数保持不变) ...

return BunNeotestAdapter

-- 确保在文件顶部引入 async
local async = require("neotest.async")
-- 并且确保 lib 已经引入
-- local lib = require("neotest.lib")

-- ... (之前的函数实现) ...

function BunNeotestAdapter.build_spec(args)
  -- 创建一个临时文件路径来存储测试结果
  local results_path = async.fn.tempname()
  -- 获取当前测试节点（测试、命名空间、文件或目录）的数据
  local position = args.tree:data()
  -- 获取项目根目录
  local cwd = assert(
    BunNeotestAdapter.root(
      position.path -- position.path 是文件路径
    ),
    "无法定位根目录 " .. position.path)
  local command = nil

  -- 根据触发测试的类型构建不同的 bun test 命令
  if position.type == "test" or position.type == "namespace" then
    -- 运行单个测试或命名空间下的所有测试
    command = "bun test " ..
      position.path .. -- 文件路径
      " --test-name-pattern " .. -- Bun 用于匹配测试名称的参数
      "\"" .. position.name .. "\"" -- 测试或命名空间的名称 (加引号处理特殊字符)
  elseif position.type == "file" then
    -- 运行文件中的所有测试
    command = "bun test " .. position.path -- 这里用 position.path 更准确
  elseif position.type == "dir" then
    -- 运行目录（通常是项目根目录）下的所有测试
    command = "bun test"
  end

  -- 返回一个包含命令、上下文和工作目录的对象
  return {
    -- 将标准错误重定向到临时文件，因为 bun test 把结果输出到 stderr
    command = command .. " 2>" .. results_path,
    context = {
      results_path = results_path, -- 将临时文件路径传递给 results 函数
    },
    cwd = cwd, -- 指定命令运行的工作目录
  }
end

-- ... (results 函数保持不变) ...

return BunNeotestAdapter

-- ... (之前的函数实现) ...
-- 确保 io 和 string 在作用域内（Lua 标准库，通常不需要显式 require）

function BunNeotestAdapter.results(spec, result, tree)
  local results = {} -- 初始化结果表
  -- 打开由 build_spec 创建的包含测试输出的临时文件
  local file = io.open(spec.context.results_path, "r")
  if not file then
    print("错误：无法打开结果文件 " .. spec.context.results_path)
    return results -- 如果文件打不开，返回空结果
  end

  local line = file:read("l") -- 读取第一行 ("l" 表示读取一行)
  local current_file_absolute_path = ""

  while line do
    -- 尝试匹配通过的测试行，例如: (pass) Describe block > Test name [0.40ms]
    -- 改进了匹配，允许浮点数时间，并捕获测试名称
    local pass_match = string.match(line, "^%(pass%) (.*) %[%.%d+ms%]$")
    if pass_match then
      -- 将 Bun 输出中的 " > " 替换为 Neotest 使用的 "::"
      local test_name_neotest_format = string.gsub(pass_match, " > ", "::")
      -- 构建 Neotest 需要的完整测试 ID：绝对路径::命名空间::测试名
      results[current_file_absolute_path .. "::" .. test_name_neotest_format] = { status = "passed" }
    end

    -- 尝试匹配失败的测试行，例如: (fail) Describe block > Test name [1.24ms]
    local fail_match = string.match(line, "^%(fail%) (.*) %[%.%d+ms%]$")
    if fail_match then
      local test_name_neotest_format = string.gsub(fail_match, " > ", "::")
      results[current_file_absolute_path .. "::" .. test_name_neotest_format] = { status = "failed" }
    end

    -- 尝试匹配文件名行，例如: src/clients/stability/stability.test.ts:
    -- 改进了匹配，处理更广泛的文件名字符
    local file_match = string.match(line, "^([^:]+%.test%.[tj]s):$")
    if file_match then
      -- 存储当前文件的绝对路径，spec.cwd 是项目根目录
      -- 确保路径分隔符正确 (在 Unix/Linux/Mac 上是 /)
      current_file_absolute_path = spec.cwd .. "/" .. file_match
      -- 如果在 Windows 上运行，可能需要处理路径分隔符
      -- current_file_absolute_path = spec.cwd .. "\\" .. file_match
    end

    line = file:read("l") -- 读取下一行
  end

  file:close() -- 关闭文件

  -- 清理临时文件 (可选但推荐)
  os.remove(spec.context.results_path)

  return results -- 返回解析后的结果表
end

return BunNeotestAdapter

{
  ["/.../project_root/src/clients/stability/stability.test.ts::Stability::generates a reference image"] = {
    status = "passed"
  },
  ["/.../project_root/src/clients/passage/passage.test.ts::Passage::GetUserTier::Undefined"] = {
    status = "passed"
  },
  ["/.../project_root/src/clients/passage/passage.test.ts::Passage::GetUserTier::with tier free"] = {
    status = "passed"
  },
  ["/.../project_root/src/clients/passage/passage.test.ts::Passage::GetUserTier::with tier hobby"] = {
    status = "failed"
  },
}