SpaceVim/bundle/nvim-treesitter/tests/indent/tiger_spec.lua

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

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

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

  describe("new line:", function()
    runner:new_line("classes.tig", { on_line = 1, text = "var a := 0", indent = 2 }, "class declaration beginning")
    runner:new_line("classes.tig", { on_line = 2, text = "var a := 0", indent = 2 }, "class declaration after field")
    runner:new_line("classes.tig", { on_line = 4, text = "var a := 0", indent = 2 }, "class declaration after method")
    runner:new_line("classes.tig", { on_line = 5, text = "var a := 0", indent = 0 }, "after class declaration")
    runner:new_line("classes.tig", { on_line = 7, text = "var a := 0", indent = 2 }, "class type beginning")
    runner:new_line("classes.tig", { on_line = 8, text = "var a := 0", indent = 2 }, "class type after field")
    runner:new_line("classes.tig", { on_line = 10, text = "self.a := 0", indent = 4 }, "inside method")
    runner:new_line("classes.tig", { on_line = 13, text = "var a := 0", indent = 2 }, "class type after method")
    runner:new_line("classes.tig", { on_line = 14, text = "var a := 0", indent = 0 }, "after class type")

    runner:new_line("control-flow.tig", { on_line = 2, text = "true", indent = 4 }, "if condition")
    runner:new_line("control-flow.tig", { on_line = 4, text = "true", indent = 4 }, "if consequence")
    runner:new_line("control-flow.tig", { on_line = 4, text = "true", indent = 4 }, "if alternative")
    runner:new_line("control-flow.tig", { on_line = 10, text = "start := 0", indent = 4 }, "for index start")
    runner:new_line("control-flow.tig", { on_line = 12, text = "the_end", indent = 4 }, "for index end")
    runner:new_line("control-flow.tig", { on_line = 14, text = "break", indent = 4 }, "for body")
    runner:new_line("control-flow.tig", { on_line = 18, text = "true", indent = 4 }, "while condition")
    runner:new_line("control-flow.tig", { on_line = 20, text = "break", indent = 4 }, "while body")

    runner:new_line("functions.tig", { on_line = 1, text = "parameter: int,", indent = 2 }, "parameter list beginning")
    runner:new_line("functions.tig", { on_line = 2, text = "parameter: int,", indent = 2 }, "parameter list middle")
    runner:new_line("functions.tig", { on_line = 4, text = ",parameter: int", indent = 2 }, "parameter list end")
    runner:new_line("functions.tig", { on_line = 5, text = "var a := 0", indent = 0 }, "after parameter list")
    runner:new_line("functions.tig", { on_line = 7, text = "print(a)", indent = 2 }, "function body")
    runner:new_line("functions.tig", { on_line = 9, text = "a,", indent = 6 }, "function call beginning")
    runner:new_line("functions.tig", { on_line = 10, text = "a,", indent = 6 }, "function call middle")
    runner:new_line("functions.tig", { on_line = 12, text = ",a", indent = 6 }, "function call end")
    runner:new_line("functions.tig", { on_line = 13, text = "; print(a)", indent = 4 }, "after function call")
    runner:new_line("functions.tig", { on_line = 14, text = "var a := 12", indent = 0 }, "after function declaration")

    runner:new_line("groupings.tig", { on_line = 2, text = "var b := 0", indent = 2 }, "let declarations")
    runner:new_line("groupings.tig", { on_line = 3, text = "a := a + 1", indent = 2 }, "after 'in'")
    runner:new_line("groupings.tig", { on_line = 4, text = "a := a + 1;", indent = 4 }, "sequence")
    runner:new_line("groupings.tig", { on_line = 8, text = "a := a + 1;", indent = 2 }, "after sequence")
    runner:new_line("groupings.tig", { on_line = 10, text = "+ 1", indent = 0 }, "after 'end'")

    runner:new_line(
      "values-and-expressions.tig",
      { on_line = 4, text = "field: record,", indent = 4 },
      "record type beginning"
    )
    runner:new_line(
      "values-and-expressions.tig",
      { on_line = 5, text = "field: record,", indent = 4 },
      "record type middle"
    )
    runner:new_line(
      "values-and-expressions.tig",
      { on_line = 7, text = ",field: record", indent = 4 },
      "record type end"
    )
    runner:new_line("values-and-expressions.tig", { on_line = 8, text = "var a := 0", indent = 2 }, "after record type")
    runner:new_line(
      "values-and-expressions.tig",
      { on_line = 10, text = "0", indent = 4 },
      "variable declaration init value"
    )
    runner:new_line(
      "values-and-expressions.tig",
      { on_line = 11, text = "+ a", indent = 4 },
      "variable declaration init follow-up"
    )
    runner:new_line("values-and-expressions.tig", { on_line = 13, text = "a", indent = 4 }, "array index")
    runner:new_line("values-and-expressions.tig", { on_line = 14, text = "+ a", indent = 4 }, "array index follow-up")
    runner:new_line("values-and-expressions.tig", { on_line = 15, text = "+ a", indent = 2 }, "after array value")
    runner:new_line("values-and-expressions.tig", { on_line = 18, text = "a", indent = 4 }, "array expression size")
    runner:new_line(
      "values-and-expressions.tig",
      { on_line = 20, text = "of", indent = 4 },
      "array expression after size"
    )
    runner:new_line(
      "values-and-expressions.tig",
      { on_line = 21, text = "a", indent = 4 },
      "array expression init value"
    )
    runner:new_line(
      "values-and-expressions.tig",
      { on_line = 25, text = "field = 0,", indent = 4 },
      "record expression beginning"
    )
    runner:new_line(
      "values-and-expressions.tig",
      { on_line = 26, text = "field = 0,", indent = 4 },
      "record expression middle"
    )
    runner:new_line(
      "values-and-expressions.tig",
      { on_line = 28, text = ",field = 0", indent = 4 },
      "record expression end"
    )
    runner:new_line(
      "values-and-expressions.tig",
      { on_line = 29, text = "a := 0", indent = 2 },
      "after record expression"
    )
  end)
end)