test_queue_async.lua

local h = require("tests.helpers")

local new_set = MiniTest.new_set

local child = MiniTest.new_child_neovim()
T = new_set({
  hooks = {
    pre_case = function()
      h.child_start(child)

      -- Load helpers and set up the environment in the child process
      child.lua([[
        --require("tests.log")
        h = require('tests.helpers')
        chat, tools = h.setup_chat_buffer()

        -- Reset test globals
        _G._test_func = nil
        _G._test_exit = nil
        _G._test_order = nil
        _G._test_output = nil
        _G._test_setup = nil
      ]])
    end,
    post_case = function()
      child.lua([[h.teardown_chat_buffer()]])
    end,
    post_once = child.stop,
  },
})

T["Tools"] = new_set()
T["Tools"]["queue"] = new_set()

T["Tools"]["queue"]["can queue multiple async functions"] = function()
  h.eq(vim.NIL, child.lua_get([[_G._test_order]]))

  child.lua([[
    local tool_call = {
      {
        ["function"] = {
          arguments = { data = "Data 1" },
          name = "func_async_1",
        },
      },
      {
        ["function"] = {
          name = "cmd_queue",
        },
      },
      {
        ["function"] = {
          arguments = { data = "Data 2" },
          name = "func_async_2",
        },
      },
    }
    tools:execute(chat, tool_call)

    while (chat.tool_orchestrator) do
      vim.wait(100)
    end
  ]])

  -- Test order
  h.eq(
    "AsyncFunc[Setup]->AsyncFunc[Success]->AsyncFunc[Exit]->Cmd[Setup]->Cmd[Success]->Cmd[Exit]->AsyncFunc2[Setup]->AsyncFunc2[Success]->AsyncFunc2[Exit]",
    child.lua_get([[_G._test_order]])
  )

  -- Test that the function was called
  h.eq("Data 1 Data 2", child.lua_get([[_G._test_func]]))
end

T["Tools"]["queue"]["can queue async function with sync function"] = function()
  h.eq(vim.NIL, child.lua_get([[_G._test_order]]))

  child.lua([[
    local tool_call = {
      {
        ["function"] = {
          arguments = { data = "Data 1" },
          name = "func_queue",
        },
      },
      {
        ["function"] = {
          arguments = { data = "Data 2" },
          name = "func_async_2",
        },
      },
    }
    tools:execute(chat, tool_call)
    while (chat.tool_orchestrator) do
      vim.wait(100)
    end
  ]])

  -- Test order
  h.eq(
    "Func[Setup]->Func[Success]->Func[Exit]->AsyncFunc2[Setup]->AsyncFunc2[Success]->AsyncFunc2[Exit]",
    child.lua_get([[_G._test_order]])
  )

  -- Test that the function was called
  h.eq("Data 1 Data 2", child.lua_get([[_G._test_func]]))
end

return T