harpoon/lua/harpoon2/test/harpoon_spec.lua

local utils = require("harpoon2.test.utils")
local Data = require("harpoon2.data")
local harpoon = require("harpoon2")

local eq = assert.are.same

describe("harpoon", function()

    before_each(function()
        Data.set_data_path("/tmp/harpoon2.json")
        Data.__dangerously_clear_data()
        require("plenary.reload").reload_module("harpoon2")
        Data = require("harpoon2.data")
        Data.set_data_path("/tmp/harpoon2.json")
        harpoon = require("harpoon2")
    end)

    it("full harpoon add sync cycle", function()
        local file_name = "/tmp/harpoon-test"
        local row = 3
        local col = 1
        local default_key = harpoon:info().default_key
        local bufnr = utils.create_file(file_name, {
            "foo",
            "bar",
            "baz",
            "qux"
        }, row, col)

        local list = harpoon:list():append()
        harpoon:sync()

        eq(harpoon:dump(), {
            [default_key] = list:encode()
        })
    end)

    it("prepend/append double add", function()
        local default_key = harpoon:info().default_key
        local file_name_1 = "/tmp/harpoon-test"
        local row_1 = 3
        local col_1 = 1

        local file_name_2 = "/tmp/harpoon-test-2"
        local row_2 = 1
        local col_2 = 2

        local contents = { "foo", "bar", "baz", "qux" }

        local bufnr_1 = utils.create_file(file_name_1, contents, row_1, col_1)
        local list = harpoon:list():append()

        utils.create_file(file_name_2, contents, row_2, col_2)
        harpoon:list():prepend()

        harpoon:sync()

        eq(harpoon:dump(), {
            [default_key] = list:encode()
        })

        eq(list.items, {
            {value = file_name_2, context = {row = row_2, col = col_2}},
            {value = file_name_1, context = {row = row_1, col = col_1}},
        })

        harpoon:list():append()
        vim.api.nvim_set_current_buf(bufnr_1)
        harpoon:list():prepend()

        eq(list.items, {
            {value = file_name_2, context = {row = row_2, col = col_2}},
            {value = file_name_1, context = {row = row_1, col = col_1}},
        })

    end)

    it("ui - display resolve", function()
        harpoon:setup({
            default = {
                display = function(item)
                    -- split string on /
                    local parts = vim.split(item.value, "/")
                    return parts[#parts]
                end
            }
        })

        local file_names = {
            "/tmp/harpoon-test-1",
            "/tmp/harpoon-test-2",
            "/tmp/harpoon-test-3",
            "/tmp/harpoon-test-4",
        }

        local contents = { "foo", "bar", "baz", "qux" }

        local bufnrs = {}
        local list = harpoon:list()
        for _, v in ipairs(file_names) do
            table.insert(bufnrs, utils.create_file(v, contents))
            harpoon:list():append()
        end

        local displayed = list:display()
        eq(displayed, {
            "harpoon-test-1",
            "harpoon-test-2",
            "harpoon-test-3",
            "harpoon-test-4",
        })

        table.remove(displayed, 3)
        table.remove(displayed, 2)

        list:resolve_displayed(displayed)

        eq(list.items, {
            {value = file_names[1], context = {row = 4, col = 2}},
            {value = file_names[4], context = {row = 4, col = 2}},
        })
    end)

    it("ui - display resolve", function()
        local file_names = {
            "/tmp/harpoon-test-1",
            "/tmp/harpoon-test-2",
            "/tmp/harpoon-test-3",
            "/tmp/harpoon-test-4",
        }

        local contents = { "foo", "bar", "baz", "qux" }

        local bufnrs = {}
        local list = harpoon:list()
        for _, v in ipairs(file_names) do
            table.insert(bufnrs, utils.create_file(v, contents))
            harpoon:list():append()
        end

        local displayed = list:display()
        eq(displayed, {
            "/tmp/harpoon-test-1",
            "/tmp/harpoon-test-2",
            "/tmp/harpoon-test-3",
            "/tmp/harpoon-test-4",
        })

        table.remove(displayed, 3)
        table.remove(displayed, 2)

        table.insert(displayed, "/tmp/harpoon-test-other-file-1")
        table.insert(displayed, "/tmp/harpoon-test-other-file-2")

        list:resolve_displayed(displayed)

        eq(list.items, {
            {value = file_names[1], context = {row = 4, col = 2}},
            {value = file_names[4], context = {row = 4, col = 2}},
            {value = "/tmp/harpoon-test-other-file-1", context = {row = 1, col = 0}},
            {value = "/tmp/harpoon-test-other-file-2", context = {row = 1, col = 0}},
        })

        table.remove(displayed, 3)
        table.insert(displayed, "/tmp/harpoon-test-4")
        list:resolve_displayed(displayed)

        eq(list.items, {
            {value = file_names[1], context = {row = 4, col = 2}},
            {value = file_names[4], context = {row = 4, col = 2}},
            {value = "/tmp/harpoon-test-other-file-2", context = {row = 1, col = 0}},
        })
    end)
end)