nvim-treesitter/tests/indent/t32_spec.lua

local Runner = require("tests.indent.common").Runner
local XFAIL = require("tests.indent.common").XFAIL

local runner = Runner:new(it, "tests/indent/t32", {
  tabstop = 2,
  shiftwidth = 2,
  softtabstop = 0,
  expandtab = true,
})

describe("indent t32:", function()
  describe("whole file:", function()
    runner:whole_file "."
  end)

  describe("new line:", function()
    runner:new_line("if_block.cmm", { on_line = 2, text = "GOTO start", indent = 0 }, "command after IF", XFAIL)

    runner:new_line("if_block.cmm", { on_line = 5, text = "GOTO start", indent = 2 }, "command in IF then block")

    runner:new_line("if_block.cmm", { on_line = 4, text = "(", indent = 0 }, "block after IF")

    for ii, test in ipairs {
      { 1, 2 },
      { 14, 2 },
      { 19, 2 },
      { 21, 2 },
      { 41, 2 },
      { 42, 4 },
    } do
      runner:new_line(
        "if_block.cmm",
        { on_line = test[1], text = "&x=1.", indent = test[2] },
        "command in IF then[" .. ii .. "]"
      )
    end

    -- Insertion of a command right before the existing block results in
    -- incorrect syntax. The parse either detect an error or incorrectly
    -- assumes "ELSE IF" is a command.
    for ii, test in ipairs {
      { 26, 2 },
      { 30, 2 },
    } do
      runner:new_line(
        "if_block.cmm",
        { on_line = test[1], text = 'PRINT ""', indent = test[2] },
        "displace block in IF then[" .. ii .. "]",
        XFAIL
      )
    end

    runner:new_line("if_block.cmm", { on_line = 45, text = "&x=1.", indent = 6 }, "command in IF then", XFAIL)

    for ii, test in ipairs {
      { 16, 2 },
      { 21, 2 },
      { 23, 2 },
      { 44, 4 },
    } do
      runner:new_line(
        "if_block.cmm",
        { on_line = test[1], text = "CONTinue\n", indent = test[2] },
        "command in IF else[" .. ii .. "]"
      )
    end

    runner:new_line("while_block.cmm", { on_line = 2, text = "&x=1.", indent = 2 }, "command after WHILE")

    runner:new_line("while_block.cmm", { on_line = 4, text = "&x=1.", indent = 0 }, "command after WHILE")

    runner:new_line("while_block.cmm", { on_line = 1, text = "(\n", indent = 0 }, "block in WHILE then")

    for ii, test in ipairs {
      { 5, 2 },
      { 12, 2 },
    } do
      runner:new_line(
        "while_block.cmm",
        { on_line = test[1], text = "&x=1.", indent = test[2] },
        "command in WHILE then block[" .. ii .. "]"
      )
    end

    for ii, test in ipairs {
      { 1, 0 },
      { 4, 2 },
    } do
      runner:new_line(
        "repeat_block.cmm",
        { on_line = test[1], text = "&x=1.", indent = test[2] },
        "command after RePeaT[" .. ii .. "]"
      )
    end

    runner:new_line("repeat_block.cmm", { on_line = 3, text = "(\n", indent = 0 }, "block in RePeaT then")

    for ii, test in ipairs {
      { 7, 2 },
      { 18, 2 },
      { 24, 2 },
    } do
      runner:new_line(
        "repeat_block.cmm",
        { on_line = test[1], text = "&x=1.", indent = test[2] },
        "command in RePeaT then block [" .. ii .. "]"
      )
    end

    runner:new_line("subroutine_block.cmm", { on_line = 1, text = "(\n", indent = 0 }, "block after call label")

    for ii, test in ipairs {
      { 2, 2 },
      { 3, 2 },
      { 8, 2 },
      { 12, 2 },
      { 19, 2 },
    } do
      runner:new_line(
        "subroutine_block.cmm",
        { on_line = test[1], text = "&x=1.", indent = test[2] },
        "command in subroutine block[" .. ii .. "]"
      )
    end

    for ii, test in ipairs {
      { 5, 2 },
      { 13, 2 },
      { 23, 2 },
    } do
      runner:new_line(
        "subroutine_block.cmm",
        { on_line = test[1], text = "&x=1.", indent = test[2] },
        "command after subroutine block[" .. ii .. "]"
      )
    end
  end)
end)