From 315c6248afbf9e5b85c06519e01ad5f9cc61dfd0 Mon Sep 17 00:00:00 2001
From: BoltlessEngineer <boltlessengineer@gmail.com>
Date: Mon, 19 Aug 2024 22:51:18 +0900
Subject: [PATCH] test: organize test cases

---
 lua/rest-nvim/parser/init.lua                 |  50 ++--
 spec/api_spec.lua                             | 235 ---------------
 spec/client/curl/command_spec.lua             |   1 +
 spec/{ => examples}/examples_spec.lua         |  36 +++
 spec/examples/script/pre_request_script.http  |   4 +
 .../variables/in_place_variables.http         |  21 ++
 spec/parser/http_parser_spec.lua              | 276 +++++++++++++++---
 spec/script_spec.lua                          |  19 ++
 spec/utils_spec.lua                           |  38 ++-
 9 files changed, 386 insertions(+), 294 deletions(-)
 delete mode 100644 spec/api_spec.lua
 rename spec/{ => examples}/examples_spec.lua (54%)
 create mode 100644 spec/examples/variables/in_place_variables.http

diff --git a/lua/rest-nvim/parser/init.lua b/lua/rest-nvim/parser/init.lua
index 6af977f..938509b 100644
--- a/lua/rest-nvim/parser/init.lua
+++ b/lua/rest-nvim/parser/init.lua
@@ -23,12 +23,18 @@ local NAMED_REQUEST_QUERY = vim.treesitter.query.parse("http", [[
   request: (_)) @request
 (section
   (comment
-    name: (_) @keyword
+    name: (_) @_keyword
     value: (_) @name
-    (#eq? @keyword "name"))
+    (#eq? @_keyword "name"))
   request: (_)) @request
 ]])
 
+local IN_PLACE_VARIABLE_QUERY = vim.treesitter.query.parse("http", [[
+(section
+  !request
+  (variable_declaration)+ @inplace_variable)
+]])
+
 ---@param node TSNode
 ---@param field string
 ---@param source Source
@@ -96,7 +102,7 @@ local function parse_urlencoded_form(str)
   return vim.iter(query_pairs):map(function (query)
     local key, value = query:match("([^=]+)=?(.*)")
     if not key then
-      -- TODO: error
+      logger.error(("Error while parsing query '%s' from urlencoded form '%s'"):format(query_pairs, str))
       return nil
     end
     return vim.trim(key) .. "=" .. vim.trim(value)
@@ -147,13 +153,12 @@ function parser.parse_body(content_type, body_node, source, context)
       return nil
     end
   elseif node_type == "raw_body" then
-    -- TODO: exclude comments from text
     local text = vim.treesitter.get_node_text(body_node, source)
     if content_type and vim.startswith(content_type, "application/x-www-form-urlencoded") then
       body.__TYPE = "raw"
       body.data = parse_urlencoded_form(text)
       if not body.data then
-        -- TODO: parsing urlencoded form failed
+        logger.error("Error while parsing urlencoded form")
         return nil
       end
     else
@@ -170,16 +175,13 @@ function parser.parse_body(content_type, body_node, source, context)
   return body
 end
 
-local IN_PLACE_VARIABLE_QUERY = "(variable_declaration) @inplace_variable"
-
 ---parse all in-place variables from source
 ---@param source Source
 ---@return rest.Context ctx
 function parser.create_context(source)
-  local query = vim.treesitter.query.parse("http", IN_PLACE_VARIABLE_QUERY)
+  local query = IN_PLACE_VARIABLE_QUERY
   local ctx = Context:new()
   local _, tree = utils.ts_parse_source(source)
-  -- TODO: capture variable_decalarations in section without request
   for _, node in query:iter_captures(tree:root(), source) do
     if node:type() == "variable_declaration" then
       parser.parse_variable_declaration(node, source, ctx)
@@ -213,7 +215,7 @@ function parser.get_all_request_nodes(source)
   local _, tree = utils.ts_parse_source(source)
   local result = {}
   for node, _ in tree:root():iter_children() do
-    if node:type() == "section" then
+    if node:type() == "section" and #node:field("request") > 0 then
       table.insert(result, node)
     end
   end
@@ -327,28 +329,34 @@ function parser.parse(node, source, ctx)
     logger.info("no method provided, falling back to 'GET'")
     method = "GET"
   end
-  local url = expand_variables(
-    assert(get_node_field_text(req_node, "url", source)),
-    ctx,
-    utils.escape
-  )
-  url = url:gsub("\n%s+", "")
+  -- NOTE: url will be parsed after because in-place variables should be parsed
+  -- first
+  local url
 
   local name
   local handlers = {}
   for child, _ in node:iter_children() do
-    local node_type = child:type()
-    if node_type == "pre_request_script" then
+    local child_type = child:type()
+    if child_type == "request" then
+      url = expand_variables(
+        assert(get_node_field_text(req_node, "url", source)),
+        ctx,
+        utils.escape
+      )
+      url = url:gsub("\n%s+", "")
+    elseif child_type == "pre_request_script" then
       parser.parse_pre_request_script(child, source, ctx)
-    elseif node_type == "res_handler_script" then
+    elseif child_type == "res_handler_script" then
       local handler = parser.parse_request_handler(child, source, ctx)
       if handler then
         table.insert(handlers, handler)
       end
-    elseif node_type == "request_separator" then
+    elseif child_type == "request_separator" then
       name = get_node_field_text(child, "value", source)
-    elseif node_type == "comment" and get_node_field_text(child, "name", source) == "name" then
+    elseif child_type == "comment" and get_node_field_text(child, "name", source) == "name" then
       name = get_node_field_text(child, "value", source) or name
+    elseif child_type == "variable_declaration" then
+      parser.parse_variable_declaration(child, source, ctx)
     end
   end
   if not name then
diff --git a/spec/api_spec.lua b/spec/api_spec.lua
deleted file mode 100644
index b0ceb98..0000000
--- a/spec/api_spec.lua
+++ /dev/null
@@ -1,235 +0,0 @@
----@module 'luassert'
-
-require("spec.minimum_init")
-
-local parser = require("rest-nvim.parser")
-local utils = require("rest-nvim.utils")
-local context = require("rest-nvim.context").Context
-local logger = require("rest-nvim.logger")
-
-local spy = require("luassert.spy")
-
-local function open(path)
-  vim.cmd.edit(path)
-  vim.cmd.source("ftplugin/http.lua")
-  return 0
-end
-
----@diagnostic disable-next-line: duplicate-set-field
-vim.notify = function () end
-
-describe("parser", function()
-  it("validate http parser", function()
-    assert.same("http", vim.treesitter.language.get_lang("http"))
-  end)
-  it("parse from http string", function()
-    local source="GET https://github.com\n"
-    local _, tree = utils.ts_parse_source(source)
-    local req_node = assert(tree:root():child(0))
-    assert.same({
-      method = "GET",
-      url = "https://github.com",
-      headers = {},
-      cookies = {},
-      handlers = {},
-    }, parser.parse(req_node, source))
-  end)
-  it("parse from http file", function()
-    local source = open "spec/examples/basic_get.http"
-    local _, tree = utils.ts_parse_source(source)
-    local req_node = assert(tree:root():child(0))
-    assert.same({
-      name = "basic_get#1",
-      method = "GET",
-      url = "https://api.github.com/users/boltlessengineer",
-      headers = {
-        ["user-agent"] = { "neovim" }
-      },
-      cookies = {},
-      handlers = {},
-    }, parser.parse(req_node, source))
-  end)
-  it("parse json", function ()
-    local source = "POST https://example.com\n\n{\n\t\"blah\": 1}\n"
-    local _, tree = utils.ts_parse_source(source)
-    local req_node = assert(tree:root():child(0))
-    assert.same({
-      method = "POST",
-      url = "https://example.com",
-      headers = {},
-      cookies = {},
-      handlers = {},
-      body = {
-        __TYPE = "json",
-        data = "{\n\t\"blah\": 1}"
-      },
-    }, parser.parse(req_node, source))
-  end)
-  it("parse invalid json", function ()
-    local source = "POST https://example.com\n\n{\n\t\"blah\": 1\n"
-    local _, tree = utils.ts_parse_source(source)
-    local req_node = assert(tree:root():child(0))
-    local spy_log_warn = spy.on(logger, "warn")
-    parser.parse(req_node, source)
-    ---@diagnostic disable-next-line: undefined-field
-    assert.spy(spy_log_warn).called_with("invalid json: '{\n\t\"blah\": 1'")
-  end)
-  it("parse xml", function ()
-    local source = [[POST https://example.com
-
-<?xml version="1.0" encoding="utf-8"?>
-<Request>
-  <Login>login</Login>
-  <Password>password</Password>
-</Request>
-]]
-    local _, tree = utils.ts_parse_source(source)
-    local req_node = assert(tree:root():child(0))
-    assert.same({
-      method = "POST",
-      url = "https://example.com",
-      headers = {},
-      cookies = {},
-      handlers = {},
-      body = {
-        __TYPE = "xml",
-        data = [[<?xml version="1.0" encoding="utf-8"?>
-<Request>
-  <Login>login</Login>
-  <Password>password</Password>
-</Request>]],
-      },
-    }, parser.parse(req_node, source))
-  end)
-  it("parse invalid xml", function ()
-    logger.info("hi")
-    local source = "POST https://example.com\n\n<?xml\n"
-    local _, tree = utils.ts_parse_source(source)
-    local req_node = assert(tree:root():child(0))
-    local spy_log_warn = spy.on(logger, "warn")
-    parser.parse(req_node, source)
-    ---@diagnostic disable-next-line: undefined-field
-    assert.spy(spy_log_warn).called_with("invalid xml: '<?xml'")
-  end)
-  it("parse with variables in header", function ()
-    local source = [[POST https://example.com
-X-DATE: {{$date}}
-]]
-    local _, tree = utils.ts_parse_source(source)
-    local req_node = assert(tree:root():child(0))
-    local req = parser.parse(req_node, source)
-    assert.not_nil(req)
-    ---@cast req rest.Request
-    assert.same({
-      ["x-date"] = { os.date("%Y-%m-%d") }
-    }, req.headers)
-  end)
-  it("parse with variables in body", function ()
-    vim.env["DATE"] = "2024-07-28"
-    local source = [[POST https://example.com
-
-{
-  "name": "{{DATE}}"
-}
-]]
-    local _, tree = utils.ts_parse_source(source)
-    local req_node = assert(tree:root():child(0))
-    local req = parser.parse(req_node, source)
-    assert.not_nil(req)
-    ---@cast req rest.Request
-    assert.same({
-      __TYPE = "json",
-      data = [[{
-  "name": "2024-07-28"
-}]]
-    }, req.body)
-  end)
-  it("parse variable declaration", function ()
-    local source = "@foo = bar\n"
-    local _, tree = utils.ts_parse_source(source)
-    local vd_node = assert(tree:root():child(0):child(0))
-    assert.same("variable_declaration", vd_node:type())
-    local c = context:new()
-    parser.parse_variable_declaration(vd_node, source, c)
-    assert.same({
-      foo = "bar"
-    }, c.vars)
-  end)
-  it("parse variable declaration with other variable", function ()
-    local source = "@foo = bar\n@baz = {{foo}} {{$date}}"
-    local _, tree = utils.ts_parse_source(source)
-    local c = context:new()
-    parser.parse_variable_declaration(assert(tree:root():child(0):child(0)), source, c)
-    parser.parse_variable_declaration(assert(tree:root():child(0):child(1)), source, c)
-    assert.same({
-      foo = "bar",
-      baz = "bar " .. os.date("%Y-%m-%d"),
-    }, c.vars)
-  end)
-  it("parse pre-request script", function ()
-    local source = "# @lang=lua\n< {%\nrequest.variables.set('foo', 'bar')\n%}\n"
-    local _, tree = utils.ts_parse_source(source)
-    local c = context:new()
-    local script_node = tree:root():child(0):child(1)
-    assert(script_node)
-    assert.same("pre_request_script", script_node:type())
-    parser.parse_pre_request_script(script_node, source, c)
-    assert.same({
-      foo = "bar",
-    }, c.vars)
-  end)
-  it("create context from source", function ()
-    local source = [[
-@foo=bar
-@bar=1234
-@baz={{foo}} adsf
-]]
-    local ctx = parser.create_context(source)
-    assert.same({
-      foo = "bar",
-      bar = "1234",
-      baz = "bar adsf",
-    }, ctx.vars)
-  end)
-  it("handler script", function ()
-    local ctx = context:new()
-    vim.env["foo"] = "old"
-    vim.env["baz"] = "old"
-    ctx:set("bar", "old")
-    local script = [[
-    client.global.set("foo", "new")
-    request.variables.set("bar", "new")
-    request.variables.set("baz", "new")
-    ]]
-    local h = require("rest-nvim.script.lua").load_post_req_hook(script, ctx)
-    ---@diagnostic disable-next-line: missing-parameter
-    h()
-    assert.same("new", vim.env["foo"])
-    assert.same("new", ctx:resolve("bar"))
-    assert.same(nil, vim.env["bar"])
-    assert.same("old", vim.env["baz"])
-  end)
-  it("capture all request names", function ()
-    local source = [[
-### first named request
-GET http://localhost:80
-### request separator that isn't a request name
-###
-# @name=second named request
-# additional comments
-GET http://localhost:80
-]]
-    local names = parser.get_request_names(source)
-    assert.same({"first named request", "second named request"}, names)
-  end)
-  it("parse request with host header", function ()
-    local source = [[
-GET /some/path
-HOST: localhost:8000
-]]
-    local _, tree = utils.ts_parse_source(source)
-    local req_node = assert(tree:root():child(0))
-    local req = assert(parser.parse(req_node, source))
-    assert.same("http://localhost:8000/some/path", req.url)
-  end)
-end)
diff --git a/spec/client/curl/command_spec.lua b/spec/client/curl/command_spec.lua
index 3e1c0f5..83ab2c0 100644
--- a/spec/client/curl/command_spec.lua
+++ b/spec/client/curl/command_spec.lua
@@ -6,6 +6,7 @@ local function open(path)
   vim.cmd.edit(path)
   -- FIXME: why runtimepath is not working?
   vim.bo.filetype = "http"
+  vim.cmd.source("ftplugin/http.lua")
   return 0
 end
 
diff --git a/spec/examples_spec.lua b/spec/examples/examples_spec.lua
similarity index 54%
rename from spec/examples_spec.lua
rename to spec/examples/examples_spec.lua
index 6822e76..a884eb2 100644
--- a/spec/examples_spec.lua
+++ b/spec/examples/examples_spec.lua
@@ -52,3 +52,39 @@ describe("url without host", function ()
     assert.same("https://example.com/api", req.url)
   end)
 end)
+
+describe("in-place variables", function ()
+  local source = open("spec/examples/variables/in_place_variables.http")
+  local req_nodes = parser.get_all_request_nodes(source)
+  assert.same(3, #req_nodes)
+  local ctx = parser.create_context(source)
+  it("example 1", function ()
+    local req1 = assert(parser.parse(req_nodes[1], source, ctx))
+    ---@cast req1 rest.Request
+    assert.same("example.org/users", req1.url)
+  end)
+  it("example 2", function ()
+    local req2 = assert(parser.parse(req_nodes[2], source, ctx))
+    assert.same("example.net/users", req2.url)
+  end)
+  it("example 3", function ()
+    local req3 = assert(parser.parse(req_nodes[3], source, ctx))
+    assert.same("example.net/stats", req3.url)
+  end)
+end)
+
+describe("pre-request script", function ()
+  local source = open("spec/examples/script/pre_request_script.http")
+  local req_nodes = parser.get_all_request_nodes(source)
+  assert.same(2, #req_nodes)
+  local ctx = parser.create_context(source)
+  it("set local variable from pre-request script", function ()
+    local req1 = assert(parser.parse(req_nodes[1], source, ctx))
+    assert.same("https://jsonplaceholder.typicode.com/posts/3", req1.url)
+  end)
+  -- TODO:
+  -- it("local variables don't affect other requests", function ()
+  --   local req2 = assert(parser.parse(req_nodes[2], source, ctx))
+  --   assert.same("https://jsonplaceholder.typicode.com/posts/", req2.url)
+  -- end)
+end)
diff --git a/spec/examples/script/pre_request_script.http b/spec/examples/script/pre_request_script.http
index eebf33e..ee9b5e2 100644
--- a/spec/examples/script/pre_request_script.http
+++ b/spec/examples/script/pre_request_script.http
@@ -1,7 +1,11 @@
 ### run request with variable set via pre-request script
 
+# @lang lua
 < {%
 request.variables.set("postId", "3")
 %}
 
 GET https://jsonplaceholder.typicode.com/posts/{{postId}}
+
+### pre-request script result is local to each requests
+GET https://jsonplaceholder.typicode.com/posts/{{postId}}
diff --git a/spec/examples/variables/in_place_variables.http b/spec/examples/variables/in_place_variables.http
new file mode 100644
index 0000000..5edcc7b
--- /dev/null
+++ b/spec/examples/variables/in_place_variables.http
@@ -0,0 +1,21 @@
+# this is a valid in-place variable syntax
+@myhost = example.org
+
+# this is not a in-place variable
+# @myhost = example.org
+
+###
+GET {{myhost}}/users
+
+###
+# you can also define variables here
+# all in-place variables are global
+@myhost=example.net
+GET {{myhost}}/users
+
+###
+# so `myhost` here is now `example.net`
+GET {{myhost}}/stats
+
+###
+# check `spec/examples/script/pre_request_script.http` for variable local to each requests
diff --git a/spec/parser/http_parser_spec.lua b/spec/parser/http_parser_spec.lua
index b599b75..954e288 100644
--- a/spec/parser/http_parser_spec.lua
+++ b/spec/parser/http_parser_spec.lua
@@ -4,6 +4,10 @@ require("spec.minimum_init")
 
 local parser = require("rest-nvim.parser")
 local utils = require("rest-nvim.utils")
+local context = require("rest-nvim.context").Context
+local logger = require("rest-nvim.logger")
+
+local spy = require("luassert.spy")
 
 local function open(path)
   vim.cmd.edit(path)
@@ -11,54 +15,260 @@ local function open(path)
   return 0
 end
 
-describe("parser", function ()
-  it("parse form-urlencoded body", function ()
-    local source = [[
-POST https://ijhttp-examples.jetbrains.com/post
-Content-Type: application/x-www-form-urlencoded
+---@diagnostic disable-next-line: duplicate-set-field
+vim.notify = function() end
 
-key1 = value1 &
-key2 = value2 &
-key3 = value3 &
-key4 = value4 &
-key5 = value5
-]]
+describe("http parser", function()
+  it("validate http parser", function()
+    assert.same("http", vim.treesitter.language.get_lang("http"))
+  end)
+  it("parse from http string", function()
+    local source = "GET https://github.com\n"
     local _, tree = utils.ts_parse_source(source)
     local req_node = assert(tree:root():child(0))
     assert.same({
-      method = "POST",
-      url = "https://ijhttp-examples.jetbrains.com/post",
-      headers = {
-        ["content-type"] = { "application/x-www-form-urlencoded" },
-      },
+      method = "GET",
+      url = "https://github.com",
+      headers = {},
       cookies = {},
       handlers = {},
-      body = {
-        __TYPE = "raw",
-        data = "key1=value1&key2=value2&key3=value3&key4=value4&key5=value5",
-      },
     }, parser.parse(req_node, source))
   end)
-  it("parse external body", function ()
-    -- external body can be only sourced when
-    local source = open("spec/examples/post_with_external_body.http")
+  it("parse from http file", function()
+    local source = open("spec/examples/basic_get.http")
     local _, tree = utils.ts_parse_source(source)
     local req_node = assert(tree:root():child(0))
     assert.same({
-      method = "POST",
-      url = "https://example.com:8080/api/html/post",
+      name = "basic_get#1",
+      method = "GET",
+      url = "https://api.github.com/users/boltlessengineer",
       headers = {
-        ["content-type"] = { "application/json" },
+        ["user-agent"] = { "neovim" },
       },
       cookies = {},
       handlers = {},
-      name = "post_with_external_body#1",
-      body = {
-        __TYPE = "external",
-        data = {
-          path = "spec/examples/input.json"
-        }
-      },
     }, parser.parse(req_node, source))
   end)
+  it("capture all request names", function ()
+    local source = [[
+### first named request
+GET http://localhost:80
+### request separator that isn't a request name
+###
+# @name=second named request
+# additional comments
+GET http://localhost:80
+]]
+    local names = parser.get_request_names(source)
+    assert.same({"first named request", "second named request"}, names)
+  end)
+  it("parse request with host header", function ()
+    local source = [[
+GET /some/path
+HOST: localhost:8000
+]]
+    local _, tree = utils.ts_parse_source(source)
+    local req_node = assert(tree:root():child(0))
+    local req = assert(parser.parse(req_node, source))
+    assert.same("http://localhost:8000/some/path", req.url)
+  end)
+
+  describe("parse body", function()
+    it("json body", function()
+      local source = 'POST https://example.com\n\n{\n\t"blah": 1}\n'
+      local _, tree = utils.ts_parse_source(source)
+      local req_node = assert(tree:root():child(0))
+      assert.same({
+        method = "POST",
+        url = "https://example.com",
+        headers = {},
+        cookies = {},
+        handlers = {},
+        body = {
+          __TYPE = "json",
+          data = '{\n\t"blah": 1}',
+        },
+      }, parser.parse(req_node, source))
+    end)
+    it("invalid json body", function()
+      local source = 'POST https://example.com\n\n{\n\t"blah": 1\n'
+      local _, tree = utils.ts_parse_source(source)
+      local req_node = assert(tree:root():child(0))
+      local spy_log_warn = spy.on(logger, "warn")
+      parser.parse(req_node, source)
+      ---@diagnostic disable-next-line: undefined-field
+      assert.spy(spy_log_warn).called_with("invalid json: '{\n\t\"blah\": 1'")
+    end)
+    it("parse xml", function()
+      local source = [[POST https://example.com
+
+<?xml version="1.0" encoding="utf-8"?>
+<Request>
+  <Login>login</Login>
+  <Password>password</Password>
+</Request>
+]]
+      local _, tree = utils.ts_parse_source(source)
+      local req_node = assert(tree:root():child(0))
+      assert.same({
+        method = "POST",
+        url = "https://example.com",
+        headers = {},
+        cookies = {},
+        handlers = {},
+        body = {
+          __TYPE = "xml",
+          data = [[<?xml version="1.0" encoding="utf-8"?>
+<Request>
+  <Login>login</Login>
+  <Password>password</Password>
+</Request>]],
+        },
+      }, parser.parse(req_node, source))
+    end)
+    it("parse invalid xml", function()
+      logger.info("hi")
+      local source = "POST https://example.com\n\n<?xml\n"
+      local _, tree = utils.ts_parse_source(source)
+      local req_node = assert(tree:root():child(0))
+      local spy_log_warn = spy.on(logger, "warn")
+      parser.parse(req_node, source)
+      ---@diagnostic disable-next-line: undefined-field
+      assert.spy(spy_log_warn).called_with("invalid xml: '<?xml'")
+    end)
+    it("parse form-urlencoded body", function()
+      local source = [[
+POST https://ijhttp-examples.jetbrains.com/post
+Content-Type: application/x-www-form-urlencoded
+
+key1 = value1 &
+key2 = value2 &
+key3 = value3 &
+key4 = value4 &
+key5 = value5
+]]
+      local _, tree = utils.ts_parse_source(source)
+      local req_node = assert(tree:root():child(0))
+      assert.same({
+        method = "POST",
+        url = "https://ijhttp-examples.jetbrains.com/post",
+        headers = {
+          ["content-type"] = { "application/x-www-form-urlencoded" },
+        },
+        cookies = {},
+        handlers = {},
+        body = {
+          __TYPE = "raw",
+          data = "key1=value1&key2=value2&key3=value3&key4=value4&key5=value5",
+        },
+      }, parser.parse(req_node, source))
+    end)
+    it("parse external body", function()
+      -- external body can be only sourced when
+      local source = open("spec/examples/post_with_external_body.http")
+      local _, tree = utils.ts_parse_source(source)
+      local req_node = assert(tree:root():child(0))
+      assert.same({
+        method = "POST",
+        url = "https://example.com:8080/api/html/post",
+        headers = {
+          ["content-type"] = { "application/json" },
+        },
+        cookies = {},
+        handlers = {},
+        name = "post_with_external_body#1",
+        body = {
+          __TYPE = "external",
+          data = {
+            path = "spec/examples/input.json",
+          },
+        },
+      }, parser.parse(req_node, source))
+    end)
+  end)
+
+  describe("variables", function()
+    it("parse with variables in header", function()
+      local source = [[POST https://example.com
+X-DATE: {{$date}}
+]]
+      local _, tree = utils.ts_parse_source(source)
+      local req_node = assert(tree:root():child(0))
+      local req = parser.parse(req_node, source)
+      assert.not_nil(req)
+      ---@cast req rest.Request
+      assert.same({
+        ["x-date"] = { os.date("%Y-%m-%d") },
+      }, req.headers)
+    end)
+    it("parse with variables in body", function()
+      vim.env["DATE"] = "2024-07-28"
+      local source = [[POST https://example.com
+
+{
+  "name": "{{DATE}}"
+}
+]]
+      local _, tree = utils.ts_parse_source(source)
+      local req_node = assert(tree:root():child(0))
+      local req = parser.parse(req_node, source)
+      assert.not_nil(req)
+      ---@cast req rest.Request
+      assert.same({
+        __TYPE = "json",
+        data = [[{
+  "name": "2024-07-28"
+}]],
+      }, req.body)
+    end)
+    it("parse variable declaration", function()
+      local source = "@foo = bar\n"
+      local _, tree = utils.ts_parse_source(source)
+      local vd_node = assert(tree:root():child(0):child(0))
+      assert.same("variable_declaration", vd_node:type())
+      local c = context:new()
+      parser.parse_variable_declaration(vd_node, source, c)
+      assert.same({
+        foo = "bar",
+      }, c.vars)
+    end)
+    it("parse variable declaration with other variable", function()
+      local source = "@foo = bar\n@baz = {{foo}} {{$date}}\n"
+      local _, tree = utils.ts_parse_source(source)
+      local c = context:new()
+      parser.parse_variable_declaration(assert(tree:root():child(0):child(0)), source, c)
+      parser.parse_variable_declaration(assert(tree:root():child(0):child(1)), source, c)
+      assert.same({
+        foo = "bar",
+        baz = "bar " .. os.date("%Y-%m-%d"),
+      }, c.vars)
+    end)
+  end)
+
+  it("parse pre-request script", function()
+    local source = "# @lang=lua\n< {%\nrequest.variables.set('foo', 'bar')\n%}\n"
+    local _, tree = utils.ts_parse_source(source)
+    local c = context:new()
+    local script_node = tree:root():child(0):child(1)
+    assert(script_node)
+    assert.same("pre_request_script", script_node:type())
+    parser.parse_pre_request_script(script_node, source, c)
+    assert.same({
+      foo = "bar",
+    }, c.vars)
+  end)
+
+  it("create context from source", function()
+    local source = [[
+@foo=bar
+@bar=1234
+@baz={{foo}} adsf
+]]
+    local ctx = parser.create_context(source)
+    assert.same({
+      foo = "bar",
+      bar = "1234",
+      baz = "bar adsf",
+    }, ctx.vars)
+  end)
 end)
diff --git a/spec/script_spec.lua b/spec/script_spec.lua
index 35aaaf3..d74d16c 100644
--- a/spec/script_spec.lua
+++ b/spec/script_spec.lua
@@ -7,6 +7,24 @@ local utils = require("rest-nvim.utils")
 local Context = require("rest-nvim.context").Context
 
 describe("handler script", function ()
+  it("load handler script", function ()
+    local ctx = Context:new()
+    vim.env["foo"] = "old"
+    vim.env["baz"] = "old"
+    ctx:set("bar", "old")
+    local script = [[
+    client.global.set("foo", "new")
+    request.variables.set("bar", "new")
+    request.variables.set("baz", "new")
+    ]]
+    local h = require("rest-nvim.script.lua").load_post_req_hook(script, ctx)
+    ---@diagnostic disable-next-line: missing-parameter
+    h()
+    assert.same("new", vim.env["foo"])
+    assert.same("new", ctx:resolve("bar"))
+    assert.same(nil, vim.env["bar"])
+    assert.same("old", vim.env["baz"])
+  end)
   it("alter response body", function ()
     local source = [[
 http://localhost:8000
@@ -21,6 +39,7 @@ response.body = vim.json.encode(json)
     local _, tree = utils.ts_parse_source(source)
     local req_node = assert(tree:root():child(0))
     local req = assert(parser.parse(req_node, source))
+    ---fake response
     ---@type rest.Response
     ---@diagnostic disable-next-line: missing-fields
     local res = {
diff --git a/spec/utils_spec.lua b/spec/utils_spec.lua
index 7575621..2373357 100644
--- a/spec/utils_spec.lua
+++ b/spec/utils_spec.lua
@@ -13,9 +13,37 @@ json.data = "overwritten"
 response.body = vim.json.encode(json)
 %}
 ]]
-  local _, tree = utils.ts_parse_source(source)
-  local script_node = assert(tree:root():child(0):child(1))
-  assert.same("res_handler_script", script_node:type())
-  local comment_node = assert(utils.ts_upper_node(script_node))
-  assert.same("comment", comment_node:type())
+  local script_node
+  it("ts_parse_source", function ()
+    local _, tree = utils.ts_parse_source(source)
+    script_node = assert(tree:root():child(0):child(1))
+    assert.same("res_handler_script", script_node:type())
+  end)
+  it("ts_find", function ()
+    local section_node = assert(utils.ts_find(script_node, "section"))
+    assert.same("section", section_node:type())
+    local sr, sc, er, ec = section_node:range()
+    assert.same({ 0, 0, 8, 0 }, { sr, sc, er, ec })
+  end)
+  it("ts_upper_node", function ()
+    local comment_node = assert(utils.ts_upper_node(script_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."
+    }
+    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, ""))
+  end)
 end)
-- 
2.51.2