145 lines
4.1 KiB
Lua
145 lines
4.1 KiB
Lua
local verify = {}
|
|
|
|
verify.eventually = function(timeout, assertfunc, failmsg, ...)
|
|
local success, args = false, { ... }
|
|
vim.wait(timeout or 1000, function()
|
|
success = assertfunc(unpack(args))
|
|
return success
|
|
end)
|
|
assert(success, failmsg)
|
|
end
|
|
|
|
local id = 0
|
|
---Waits until the next vim.schedule before running assertfunc
|
|
verify.schedule = function(assertfunc, timeout, failmsg)
|
|
id = id + 1
|
|
local scheduled_func_ran = false
|
|
local success = false
|
|
local args
|
|
vim.schedule(function()
|
|
args = { assertfunc() }
|
|
success = args[1]
|
|
scheduled_func_ran = true
|
|
end)
|
|
local notimeout, errcode = vim.wait(timeout or 1000, function()
|
|
return scheduled_func_ran
|
|
end)
|
|
assert(success, failmsg)
|
|
end
|
|
|
|
verify.after = function(timeout, assertfunc, failmsg)
|
|
vim.wait(timeout, function()
|
|
return false
|
|
end)
|
|
assert(assertfunc(), failmsg)
|
|
end
|
|
|
|
verify.bufnr_is = function(bufnr, timeout)
|
|
verify.eventually(timeout or 500, function()
|
|
return bufnr == vim.api.nvim_get_current_buf()
|
|
end, string.format("Current buffer is expected to be '%s' but is not", bufnr))
|
|
end
|
|
|
|
verify.bufnr_is_not = function(bufnr, timeout)
|
|
verify.eventually(timeout or 500, function()
|
|
return bufnr ~= vim.api.nvim_get_current_buf()
|
|
end, string.format("Current buffer is '%s' when expected to not be", bufnr))
|
|
end
|
|
|
|
verify.buf_name_endswith = function(buf_name, timeout)
|
|
verify.eventually(
|
|
timeout or 500,
|
|
function()
|
|
if buf_name == "" then
|
|
return true
|
|
end
|
|
local n = vim.api.nvim_buf_get_name(0)
|
|
if n:sub(-#buf_name) == buf_name then
|
|
return true
|
|
else
|
|
return false
|
|
end
|
|
end,
|
|
string.format("Current buffer name is expected to be end with '%s' but it does not", buf_name)
|
|
)
|
|
end
|
|
|
|
verify.buf_name_is = function(buf_name, timeout)
|
|
verify.eventually(timeout or 500, function()
|
|
return buf_name == vim.api.nvim_buf_get_name(0)
|
|
end, string.format("Current buffer name is expected to be '%s' but is not", buf_name))
|
|
end
|
|
|
|
verify.tree_focused = function(timeout)
|
|
verify.eventually(timeout or 1000, function()
|
|
if not verify.get_state() then
|
|
return false
|
|
end
|
|
return vim.bo[0].filetype == "neo-tree"
|
|
end, "Current buffer is not a 'neo-tree' filetype")
|
|
end
|
|
|
|
verify.get_state = function(source_name, winid)
|
|
if source_name == nil then
|
|
local success
|
|
success, source_name = pcall(vim.api.nvim_buf_get_var, 0, "neo_tree_source")
|
|
if not success then
|
|
return nil
|
|
end
|
|
end
|
|
local state = require("neo-tree.sources.manager").get_state(source_name, nil, winid)
|
|
if not state.tree then
|
|
return nil
|
|
end
|
|
if not state._ready then
|
|
return nil
|
|
end
|
|
return state
|
|
end
|
|
|
|
verify.tree_node_is = function(source_name, expected_node_id, winid, timeout)
|
|
verify.eventually(timeout or 500, function()
|
|
local state = verify.get_state(source_name, winid)
|
|
if not state then
|
|
return false
|
|
end
|
|
local success, node = pcall(state.tree.get_node, state.tree)
|
|
if not success then
|
|
return false
|
|
end
|
|
if not node then
|
|
return false
|
|
end
|
|
local node_id = node:get_id()
|
|
if node_id == expected_node_id then
|
|
return true
|
|
end
|
|
return false
|
|
end, string.format("Tree node '%s' not focused", expected_node_id))
|
|
end
|
|
|
|
verify.filesystem_tree_node_is = function(expected_node_id, winid, timeout)
|
|
verify.tree_node_is("filesystem", expected_node_id, winid, timeout)
|
|
end
|
|
|
|
verify.buffers_tree_node_is = function(expected_node_id, winid, timeout)
|
|
verify.tree_node_is("buffers", expected_node_id, winid, timeout)
|
|
end
|
|
|
|
verify.git_status_tree_node_is = function(expected_node_id, winid, timeout)
|
|
verify.tree_node_is("git_status", expected_node_id, winid, timeout)
|
|
end
|
|
|
|
verify.window_handle_is = function(winid, timeout)
|
|
verify.eventually(timeout or 500, function()
|
|
return winid == vim.api.nvim_get_current_win()
|
|
end, string.format("Current window handle is expected to be '%s' but is not", winid))
|
|
end
|
|
|
|
verify.window_handle_is_not = function(winid, timeout)
|
|
verify.eventually(timeout or 500, function()
|
|
return winid ~= vim.api.nvim_get_current_win()
|
|
end, string.format("Current window handle is not expected to be '%s' but it is", winid))
|
|
end
|
|
|
|
return verify
|