summaryrefslogtreecommitdiff
path: root/lua/99/test/test_utils.lua
diff options
context:
space:
mode:
Diffstat (limited to 'lua/99/test/test_utils.lua')
-rw-r--r--lua/99/test/test_utils.lua98
1 files changed, 49 insertions, 49 deletions
diff --git a/lua/99/test/test_utils.lua b/lua/99/test/test_utils.lua
index 29b7344..2f60b86 100644
--- a/lua/99/test/test_utils.lua
+++ b/lua/99/test/test_utils.lua
@@ -1,14 +1,14 @@
local M = {}
function M.next_frame()
- local next = false
- vim.schedule(function()
- next = true
- end)
+ local next = false
+ vim.schedule(function()
+ next = true
+ end)
- vim.wait(1000, function()
- return next
- end)
+ vim.wait(1000, function()
+ return next
+ end)
end
M.created_files = {}
@@ -25,64 +25,64 @@ local TestProvider = {}
TestProvider.__index = TestProvider
function TestProvider.new()
- return setmetatable({}, TestProvider)
+ return setmetatable({}, TestProvider)
end
--- @param query string
---@param request _99.Request
---@param observer _99.ProviderObserver?
function TestProvider:make_request(query, request, observer)
- local logger = request.context.logger:set_area("TestProvider")
- logger:debug("make_request", "tmp_file", request.context.tmp_file)
- self.request = {
- query = query,
- request = request,
- observer = observer,
- logger = logger,
- }
+ local logger = request.context.logger:set_area("TestProvider")
+ logger:debug("make_request", "tmp_file", request.context.tmp_file)
+ self.request = {
+ query = query,
+ request = request,
+ observer = observer,
+ logger = logger,
+ }
end
--- @param status _99.Request.ResponseState
--- @param result string
function TestProvider:resolve(status, result)
- assert(self.request, "you cannot call resolve until make_request is called")
- local obs = self.request.observer
- if obs then
- --- to match the behavior expected from the OpenCodeProvider
- if self.request.request:is_cancelled() then
- obs.on_complete("cancelled", result)
- else
- obs.on_complete(status, result)
- end
+ assert(self.request, "you cannot call resolve until make_request is called")
+ local obs = self.request.observer
+ if obs then
+ --- to match the behavior expected from the OpenCodeProvider
+ if self.request.request:is_cancelled() then
+ obs.on_complete("cancelled", result)
+ else
+ obs.on_complete(status, result)
end
- self.request = nil
+ end
+ self.request = nil
end
--- @param line string
function TestProvider:stdout(line)
- assert(self.request, "you cannot call stdout until make_request is called")
- local obs = self.request.observer
- if obs then
- obs.on_stdout(line)
- end
+ assert(self.request, "you cannot call stdout until make_request is called")
+ local obs = self.request.observer
+ if obs then
+ obs.on_stdout(line)
+ end
end
--- @param line string
function TestProvider:stderr(line)
- assert(self.request, "you cannot call stderr until make_request is called")
- local obs = self.request.observer
- if obs then
- obs.on_stderr(line)
- end
+ assert(self.request, "you cannot call stderr until make_request is called")
+ local obs = self.request.observer
+ if obs then
+ obs.on_stderr(line)
+ end
end
M.TestProvider = TestProvider
function M.clean_files()
- for _, bufnr in ipairs(M.created_files) do
- vim.api.nvim_buf_delete(bufnr, { force = true })
- end
- M.created_files = {}
+ for _, bufnr in ipairs(M.created_files) do
+ vim.api.nvim_buf_delete(bufnr, { force = true })
+ end
+ M.created_files = {}
end
---@param contents string[]
@@ -90,17 +90,17 @@ end
---@param row number?
---@param col number?
function M.create_file(contents, file_type, row, col)
- assert(type(contents) == "table", "contents must be a table of strings")
- file_type = file_type or "lua"
- local bufnr = vim.api.nvim_create_buf(false, false)
+ assert(type(contents) == "table", "contents must be a table of strings")
+ file_type = file_type or "lua"
+ local bufnr = vim.api.nvim_create_buf(false, false)
- vim.api.nvim_set_current_buf(bufnr)
- vim.bo[bufnr].ft = file_type
- vim.api.nvim_buf_set_lines(bufnr, 0, -1, false, contents)
- vim.api.nvim_win_set_cursor(0, { row or 1, col or 0 })
+ vim.api.nvim_set_current_buf(bufnr)
+ vim.bo[bufnr].ft = file_type
+ vim.api.nvim_buf_set_lines(bufnr, 0, -1, false, contents)
+ vim.api.nvim_win_set_cursor(0, { row or 1, col or 0 })
- table.insert(M.created_files, bufnr)
- return bufnr
+ table.insert(M.created_files, bufnr)
+ return bufnr
end
return M