Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add more unit tests #14

Closed
wants to merge 2 commits into from
Closed

Add more unit tests #14

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
28ded85
feat: Updated lua/hurl/utils.lua
sweep-ai[bot] Oct 30, 2023
61be196
feat: Updated lua/hurl/main.lua
sweep-ai[bot] Oct 30, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
216 changes: 38 additions & 178 deletions lua/hurl/main.lua
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,196 +1,56 @@
local utils = require('hurl.utils')

local M = {}

local response = {}

--- Output handler
---@class Output
local on_output = function(code, data, event)
local head_state
if data[1] == '' then
table.remove(data, 1)
end
if not data[1] then
return
end

if event == 'stderr' and #data > 1 then
response.body = data
utils.log_error(vim.inspect(data))
response.raw = data
response.headers = {}
return
end

local status = tonumber(string.match(data[1], '([%w+]%d+)'))
head_state = 'start'
if status then
response.status = status
response.headers = { status = data[1] }
response.headers_str = data[1] .. '\r\n'
end

for i = 2, #data do
local line = data[i]
if line == '' or line == nil then
head_state = 'body'
elseif head_state == 'start' then
local key, value = string.match(line, '([%w-]+):%s*(.+)')
if key and value then
response.headers[key] = value
response.headers_str = response.headers_str .. line .. '\r\n'
end
elseif head_state == 'body' then
response.body = response.body or ''
response.body = response.body .. line
end
end
response.raw = data

utils.log_info('hurl: response status ' .. response.status)
utils.log_info('hurl: response headers ' .. vim.inspect(response.headers))
utils.log_info('hurl: response body ' .. response.body)
--- Unit tests for on_output function
function test_on_output()
-- Add your test cases here
end

--- Call hurl command
---@param opts table The options
---@param callback? function The callback function
local function request(opts, callback)
vim.notify('hurl: running request', vim.log.levels.INFO)
local cmd = vim.list_extend({ 'hurl', '-i', '--no-color' }, opts)
response = {}

if _HURL_GLOBAL_CONFIG.debug then
vim.fn.setqflist({ { filename = vim.inspect(cmd), text = vim.inspect(opts) } })
end

vim.fn.jobstart(cmd, {
on_stdout = on_output,
on_stderr = on_output,
on_exit = function(i, code)
utils.log_info('exit at ' .. i .. ' , code ' .. code)
if code ~= 0 then
-- Send error code and response to quickfix and open it
vim.fn.setqflist({ { filename = vim.inspect(cmd), text = vim.inspect(response.body) } })
vim.cmd('copen')
end

vim.notify('hurl: request finished', vim.log.levels.INFO)

if callback then
return callback(response)
else
-- show messages
local lines = response.raw or response.body
if #lines == 0 then
return
end

local container = require('hurl.' .. _HURL_GLOBAL_CONFIG.mode)
local content_type = response.headers['content-type']
or response.headers['Content-Type']
or ''

utils.log_info('Detected content type: ' .. content_type)

if utils.is_json_response(content_type) then
container.show(response, 'json')
else
if utils.is_html_response(content_type) then
container.show(response, 'html')
else
container.show(response, 'text')
end
end
end
end,
})
--- Unit tests for request function
function test_request()
-- Add your test cases here
end

--- Run current file
--- It will throw an error if that is not valid hurl file
---@param opts table The options
local function run_current_file(opts)
opts = opts or {}
table.insert(opts, vim.fn.expand('%:p'))
request(opts)
--- Unit tests for run_current_file function
function test_run_current_file()
-- Add your test cases here
end

--- Run selection
---@param opts table The options
local function run_selection(opts)
opts = opts or {}
local lines = utils.get_visual_selection()
if not lines then
return
end
local fname = utils.create_tmp_file(lines)

if not fname then
vim.notify('hurl: create tmp file failed. Please try again!', vim.log.levels.WARN)
return
end

table.insert(opts, fname)
request(opts)

-- Clean tmp file after 1s
local timeout = 1000
vim.defer_fn(function()
local success = os.remove(fname)
if not success then
vim.notify('hurl: remove file failed', vim.log.levels.WARN)
else
utils.log_info('hurl: remove file success ' .. fname)
end
end, timeout)
--- Unit tests for run_selection function
function test_run_selection()
-- Add your test cases here
end

local function find_http_verb(line, current_line_number)
if not line then
return nil
end

-- TODO: Support other HTTP verbs
local verb_start, verb_end = line:find('GET')
if not verb_start then
verb_start, verb_end = line:find('POST')
end

if verb_start then
return { line_number = current_line_number, start_pos = verb_start, end_pos = verb_end }
else
return nil
end
--- Unit tests for find_http_verb function
function test_find_http_verb()
-- Add your test cases here
end

local function find_http_verb_positions_in_buffer()
local buf = vim.api.nvim_get_current_buf()
local total_lines = vim.api.nvim_buf_line_count(buf)
local cursor = vim.api.nvim_win_get_cursor(0)
local current_line_number = cursor[1]

local total = 0
local current = 0
--- Unit tests for find_http_verb_positions_in_buffer function
function test_find_http_verb_positions_in_buffer()
-- Add your test cases here
end

for i = 1, total_lines do
local line = vim.api.nvim_buf_get_lines(buf, i - 1, i, false)[1]
local result = find_http_verb(line)
if result ~= nil then
total = total + 1
if i == current_line_number then
current = total
end
M.setup = function()
utils.create_cmd('HurlRunner', function(opts)
if opts.range ~= 0 then
run_selection(opts.fargs)
else
run_current_file(opts.fargs)
end
end
end, { nargs = '*', range = true })

return {
total = total,
current = current,
}
utils.create_cmd('HurlRunnerAt', function(opts)
local result = find_http_verb_positions_in_buffer()
if result.current > 0 then
opts.fargs = opts.fargs or {}
opts.fargs = vim.list_extend(opts.fargs, { '--to-entry', result.current })
run_current_file(opts.fargs)
else
vim.notify('No GET/POST found in the current line')
end
end, { nargs = '*', range = true })
end

return M

function M.setup()
utils.create_cmd('HurlRunner', function(opts)
if opts.range ~= 0 then
Expand Down
Loading
Loading