rest.nvim/spec/utils_spec.lua at bddbc186bae3df575f017d83801acb4d14c95d7f · rest-nvim/rest.nvim · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
---@module 'luassert'

local utils = require("rest-nvim.utils")

local function open(path)
    vim.cmd.edit(path)
    return 0
end

describe("basic utils", function()
    it("parse_url", function()
        local function parse_url(url)
            local domain, path = utils.parse_url(url)
            return { domain = domain, path = path }
        end
        assert.same({
            domain = "domain.com",
            path = "/some/path",
        }, parse_url("http://domain.com/some/path?query=value"))
        assert.same({
            domain = "domain.com",
            path = "/some/path",
        }, parse_url("domain.com/some/path?query=value"))
        assert.same({
            domain = "localhost:8000",
            path = "/some/path",
        }, parse_url("localhost:8000/some/path?query=value"))
    end)
end)

describe("tree-sitter utils", function()
    local source = open("spec/examples/script/post_request_script.http")
    it("ts_parse_source", function()
        local _, tree = utils.ts_parse_source(source)
        local url_node = assert(tree:root():child(0):field("request")[1]:field("url")[1])
        assert.same("target_url", url_node:type())
        assert.is_false(tree:root():has_error())
    end)
    it("ts_find", function()
        local start_node = assert(vim.treesitter.get_node({ pos = { 4, 3 }, lang = "http" }))
        local script_node = assert(utils.ts_find(start_node, "script"))
        assert.same("script", script_node:type())
        local sr, sc, er, ec = script_node:range()
        assert.same({ 4, 2, 7, 2 }, { sr, sc, er, ec })
    end)
    it("ts_upper_node", function()
        local start_node = assert(vim.treesitter.get_node({ pos = { 4, 3 }, lang = "http" }))
        local comment_node = assert(utils.ts_upper_node(start_node))
        assert.same("comment", comment_node:type())
    end)
end)

describe("gq_lines", function()
    it("plain text", function()
        local lines = {
            "Lorem",
            "ipsum dolor sit amet, consectetur adipiscing elit. Nulla id nisl ut sapien ullamcorper congue non in ipsum",
            ". Phasellus efficitur metus lectus, sed placerat eros mollis varius. Praesent egestas sapien vel auctor egestas. Praesent ac lacus consequat, rhoncus libero et, ultricies urna. Maecenas vitae tortor ut mi convallis volutpat. Nam.",
        }
        vim.api.nvim_create_autocmd("FileType", {
            pattern = "text",
            callback = function(ev)
                vim.bo[ev.buf].formatprg = "fmt"
            end,
        })
        assert.same({
            "Lorem ipsum dolor sit amet, consectetur adipiscing elit. Nulla id nisl",
            "ut sapien ullamcorper congue non in ipsum . Phasellus efficitur metus",
            "lectus, sed placerat eros mollis varius. Praesent egestas sapien vel",
            "auctor egestas. Praesent ac lacus consequat, rhoncus libero et, ultricies",
            "urna. Maecenas vitae tortor ut mi convallis volutpat. Nam.",
        }, utils.gq_lines(lines, "text"))
    end)
    it("json with jq", function()
        local lines = {
            " {",
            '         "foo"    : 123      }',
        }
        vim.api.nvim_create_autocmd("FileType", {
            pattern = "json",
            callback = function(ev)
                vim.bo[ev.buf].formatprg = "jq --indent 4"
            end,
        })
        assert.same({
            "{",
            '    "foo": 123',
            "}",
        }, utils.gq_lines(lines, "json"))
    end)
    it("xml with xmlformat#Format()", function()
        local lines = {
            "<note>",
            "<to>User</to>",
            "<from>Bob</from>",
            "<heading>Reminder</heading>",
            "<body>Don't forget to complete your tasks today!</body>",
            "</note>",
        }
        assert.same({
            "<note>",
            "        <to>User</to>",
            "        <from>Bob</from>",
            "        <heading>Reminder</heading>",
            "        <body>Don't forget to complete your tasks today!</body>",
            "</note>",
        }, utils.gq_lines(lines, "xml"))
    end)
    it("handle empty content #473", function()
        local lines = {}
        assert.same({}, utils.gq_lines(lines, "json"))
    end)
    it("handle format error #478 (neovim/neovim#30593)", function()
        vim.api.nvim_create_autocmd("FileType", {
            pattern = "json",
            callback = function(ev)
                vim.bo[ev.buf].formatprg = "jq --indent 4"
            end,
        })
        local lines = { "" }
        assert.same({ "" }, utils.gq_lines(lines, "json"))
    end)
    it("returns original lines when json parsing fails", function()
        vim.api.nvim_create_autocmd("FileType", {
            pattern = "json",
            callback = function(ev)
                vim.bo[ev.buf].formatprg = "jq --indent 4"
            end,
        })
        local lines = { "this is not valid json {{{" }
        local result, ok = utils.gq_lines(lines, "json")
        assert.is_false(ok)
        assert.same(lines, result)
    end)
end)