gitsigns.nvim/test/actions_spec.lua

local helpers = require('test.gs_helpers')

local setup_gitsigns  = helpers.setup_gitsigns
local feed            = helpers.feed
local test_file       = helpers.test_file
local edit            = helpers.edit
local command         = helpers.command
local check           = helpers.check
local exec_lua        = helpers.exec_lua
local fn              = helpers.funcs
local system          = fn.system
local test_config     = helpers.test_config
local clear           = helpers.clear
local setup_test_repo = helpers.setup_test_repo
local eq              = helpers.eq
local expectf         = helpers.expectf

helpers.env()

--- @param exp_hunks string[]
local function expect_hunks(exp_hunks)
  expectf(function()
    --- @type table[]
    local hunks = exec_lua"return require('gitsigns').get_hunks()"
    if #exp_hunks~= #hunks then
      local msg = {} --- @type string[]
      msg[#msg+1] = ''
      msg[#msg+1] = string.format(
              'Number of hunks do not match. Expected: %d, passed in: %d',
              #exp_hunks, #hunks)
      msg[#msg+1] = ''
      msg[#msg+1] = 'Expected hunks:'
      for _, h in ipairs(exp_hunks) do
        msg[#msg+1] = h.head
      end
      msg[#msg+1] = ''
      msg[#msg+1] = 'Passed in hunks:'
      for _, h in ipairs(hunks) do
        msg[#msg+1] = h.head
      end
      error(table.concat(msg, '\n'))
    end
    for i, hunk in ipairs(hunks) do
      eq(exp_hunks[i], hunk.head)
    end
  end)
end

describe('actions', function()
  local config

  before_each(function()
    clear()
    -- Make gitisigns available
    exec_lua('package.path = ...', package.path)
    config = vim.deepcopy(test_config)
    command('cd '..system{"dirname", os.tmpname()})
    setup_gitsigns(config)
  end)

  after_each(function()
    -- cleanup()
  end)

  it('works with commands', function()
    setup_test_repo()
    edit(test_file)

    feed("jjjccEDIT<esc>")
    check {
      status = {head='master', added=0, changed=1, removed=0},
      signs  = {changed=1}
    }

    -- command 'Gitsigns stage_hunk'
    -- check {
    --   status = {head='master', added=0, changed=0, removed=0},
    --   signs  = {}
    -- }

    -- command 'Gitsigns undo_stage_hunk'
    -- check {
    --   status = {head='master', added=0, changed=1, removed=0},
    --   signs  = {changed=1}
    -- }

    -- -- Add multiple edits
    -- feed('ggccThat<esc>')

    -- check {
    --   status = {head='master', added=0, changed=2, removed=0},
    --   signs  = {changed=2}
    -- }

    -- command 'Gitsigns stage_buffer'
    -- check {
    --   status = {head='master', added=0, changed=0, removed=0},
    --   signs  = {}
    -- }

    -- command 'Gitsigns reset_buffer_index'
    -- check {
    --   status = {head='master', added=0, changed=2, removed=0},
    --   signs  = {changed=2}
    -- }

    -- command 'Gitsigns reset_hunk'
    -- check {
    --   status = {head='master', added=0, changed=1, removed=0},
    --   signs  = {changed=1}
    -- }
  end)

  describe('staging partial hunks', function()
    before_each(function()
      setup_test_repo{test_file_text={'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'}}
      edit(test_file)
    end)

    local function set_lines(start, dend, lines)
      exec_lua([[
        local start, dend, lines = ...
        vim.api.nvim_buf_set_lines(0, start, dend, false, lines)
      ]], start, dend, lines)
      -- command('write')
    end

    describe('can stage add hunks', function()
      before_each(function()
        set_lines(2, 2, {'c1', 'c2', 'c3', 'c4'})
        expect_hunks{ '@@ -2 +3,4 @@' }
      end)

      it('contained in range', function()
        command[[1,7 Gitsigns stage_hunk]]
        expect_hunks{ }
      end)

      it('containing range', function()
        command[[4,5 Gitsigns stage_hunk]]
        expect_hunks{
          '@@ -2 +3,1 @@',
          '@@ -4 +6,1 @@'
        }
      end)

      it('from top range', function()
        command[[1,4 Gitsigns stage_hunk]]
        expect_hunks{ '@@ -4 +5,2 @@' }
      end)

      it('from bottom range', function()
        command[[4,7 Gitsigns stage_hunk]]
        expect_hunks{ '@@ -2 +3,1 @@' }

        command[[Gitsigns reset_buffer_index]]
        expect_hunks{ '@@ -2 +3,4 @@' }

        command[[4,10 Gitsigns stage_hunk]]
        expect_hunks{ '@@ -2 +3,1 @@' }
      end)
    end)

    describe('can stage modified-add hunks', function()
      before_each(function()
        set_lines(2, 4, {'c1', 'c2', 'c3', 'c4', 'c5'})
        expect_hunks{ '@@ -3,2 +3,5 @@' }
      end)

      it('from top range containing mod', function()
        command[[2,3 Gitsigns stage_hunk]]
        expect_hunks{ '@@ -4,1 +4,4 @@' }
      end)

      it('from top range containing mod-add', function()
        command[[2,5 Gitsigns stage_hunk]]
        expect_hunks{ '@@ -5 +6,2 @@' }
      end)

      it('from bottom range containing add', function()
        command[[6,8 Gitsigns stage_hunk]]
        expect_hunks{ '@@ -3,2 +3,3 @@' }
      end)

      it('containing range containing add', function()
        command'write'
        command[[5,6 Gitsigns stage_hunk]]
        expect_hunks{
          '@@ -3,2 +3,2 @@',
          '@@ -6 +7,1 @@'
        }
      end)

    end)

    describe('can stage modified-remove hunks', function()
      before_each(function()
        set_lines(2, 7, {'c1', 'c2', 'c3'})
        command('write')
        expect_hunks{ '@@ -3,5 +3,3 @@' }
      end)

      it('from top range', function()
        expect_hunks{ '@@ -3,5 +3,3 @@' }

        command[[2,3 Gitsigns stage_hunk]]
        expect_hunks{ '@@ -4,4 +4,2 @@' }

        command[[2,3 Gitsigns reset_buffer_index]]
        expect_hunks{ '@@ -3,5 +3,3 @@' }

        command[[2,4 Gitsigns stage_hunk]]
        expect_hunks{ '@@ -5,3 +5,1 @@' }
      end)

      it('from bottom range', function()
        expect_hunks{ '@@ -3,5 +3,3 @@' }

        command[[4,6 Gitsigns stage_hunk]]
        expect_hunks{ '@@ -3,1 +3,1 @@' }

        command[[2,3 Gitsigns reset_buffer_index]]
        expect_hunks{ '@@ -3,5 +3,3 @@' }

        command[[5,6 Gitsigns stage_hunk]]
        expect_hunks{ '@@ -3,2 +3,2 @@' }
      end)
    end)

    it('can stage remove hunks', function()
      set_lines(2, 5, {})
      expect_hunks{ '@@ -3,3 +2 @@' }

      command[[2 Gitsigns stage_hunk]]
      expect_hunks{}
    end)

  end)

end)