130 lines
2.7 KiB
Lua
130 lines
2.7 KiB
Lua
local M = {}
|
|
|
|
M.logged_messages = {}
|
|
|
|
local mock_logger = {
|
|
log = function(msg, level)
|
|
table.insert(M.logged_messages, { msg = msg, level = level })
|
|
end,
|
|
set_config = function() end,
|
|
}
|
|
|
|
local function setup_vim_mocks()
|
|
if not vim.fn then
|
|
vim.fn = {}
|
|
end
|
|
vim.fn.expand = vim.fn.expand or function()
|
|
return '/tmp/test.cpp'
|
|
end
|
|
vim.fn.mkdir = vim.fn.mkdir or function() end
|
|
vim.fn.fnamemodify = vim.fn.fnamemodify or function(path)
|
|
return path
|
|
end
|
|
vim.fn.tempname = vim.fn.tempname or function()
|
|
return '/tmp/session'
|
|
end
|
|
if not vim.api then
|
|
vim.api = {}
|
|
end
|
|
vim.api.nvim_get_current_buf = vim.api.nvim_get_current_buf or function()
|
|
return 1
|
|
end
|
|
vim.api.nvim_buf_get_lines = vim.api.nvim_buf_get_lines or function()
|
|
return { '' }
|
|
end
|
|
if not vim.cmd then
|
|
vim.cmd = {}
|
|
end
|
|
vim.cmd.e = function() end
|
|
vim.cmd.only = function() end
|
|
vim.cmd.split = function() end
|
|
vim.cmd.vsplit = function() end
|
|
if not vim.system then
|
|
vim.system = function(cmd)
|
|
return {
|
|
wait = function()
|
|
return { code = 0 }
|
|
end,
|
|
}
|
|
end
|
|
end
|
|
end
|
|
|
|
function M.setup()
|
|
M.logged_messages = {}
|
|
package.loaded['cp.log'] = mock_logger
|
|
end
|
|
|
|
function M.setup_full()
|
|
M.setup()
|
|
setup_vim_mocks()
|
|
|
|
local cache = require('cp.cache')
|
|
cache.load = function() end
|
|
cache.set_test_cases = function() end
|
|
cache.set_file_state = function() end
|
|
cache.get_file_state = function()
|
|
return nil
|
|
end
|
|
cache.get_contest_data = function()
|
|
return nil
|
|
end
|
|
cache.get_test_cases = function()
|
|
return {}
|
|
end
|
|
end
|
|
|
|
function M.mock_scraper_success()
|
|
package.loaded['cp.scrape'] = {
|
|
scrape_problem = function(ctx)
|
|
return {
|
|
success = true,
|
|
problem_id = ctx.problem_id,
|
|
test_cases = {
|
|
{ input = '1 2', expected = '3' },
|
|
{ input = '3 4', expected = '7' },
|
|
},
|
|
test_count = 2,
|
|
}
|
|
end,
|
|
scrape_contest_metadata = function(platform, contest_id)
|
|
return {
|
|
success = true,
|
|
problems = {
|
|
{ id = 'a' },
|
|
{ id = 'b' },
|
|
{ id = 'c' },
|
|
},
|
|
}
|
|
end,
|
|
scrape_problems_parallel = function()
|
|
return {}
|
|
end,
|
|
}
|
|
end
|
|
|
|
function M.has_error_logged()
|
|
for _, log_entry in ipairs(M.logged_messages) do
|
|
if log_entry.level == vim.log.levels.ERROR then
|
|
return true
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
function M.find_logged_message(pattern)
|
|
for _, log_entry in ipairs(M.logged_messages) do
|
|
if log_entry.msg and log_entry.msg:match(pattern) then
|
|
return log_entry
|
|
end
|
|
end
|
|
return nil
|
|
end
|
|
|
|
function M.teardown()
|
|
package.loaded['cp.log'] = nil
|
|
package.loaded['cp.scrape'] = nil
|
|
M.logged_messages = {}
|
|
end
|
|
|
|
return M
|