From 5ec7f9200e1fa402266b9cb3ef5332d3170b3f17 Mon Sep 17 00:00:00 2001 From: mpaulson Date: Wed, 29 Nov 2023 14:50:13 -0700 Subject: [PATCH] feat: adding and removing buffers from list --- lua/harpoon2/test/ui_spec.lua | 17 ++++++++++++++++- lua/harpoon2/ui.lua | 1 + 2 files changed, 17 insertions(+), 1 deletion(-) diff --git a/lua/harpoon2/test/ui_spec.lua b/lua/harpoon2/test/ui_spec.lua index 4051b89..c84f675 100644 --- a/lua/harpoon2/test/ui_spec.lua +++ b/lua/harpoon2/test/ui_spec.lua @@ -28,7 +28,7 @@ describe("harpoon", function() eq(harpoon.ui.win_id, nil) end) - it("delete file from list via ui", function() + it("delete file from ui contents and save", function() local created_files = utils.fill_list_with_files(3, harpoon:list()) eq(harpoon:list():length(), 3) @@ -40,4 +40,19 @@ describe("harpoon", function() eq(harpoon:list():length(), 2) eq(harpoon:list():display(), created_files) end) + + it("add file from ui contents and save", function() + local list = harpoon:list() + local created_files = utils.fill_list_with_files(3, list) + table.insert(created_files, os.tmpname()) + + eq(list:length(), 3) + + harpoon.ui:toggle_quick_menu(list) + Buffer.set_contents(harpoon.ui.bufnr, created_files) + harpoon.ui:save() + + eq(list:length(), 4) + eq(list:display(), created_files) + end) end) diff --git a/lua/harpoon2/ui.lua b/lua/harpoon2/ui.lua index 2f652b8..de16f8a 100644 --- a/lua/harpoon2/ui.lua +++ b/lua/harpoon2/ui.lua @@ -109,6 +109,7 @@ end function HarpoonUI:save() local list = Buffer.get_contents(self.bufnr) self.active_list:resolve_displayed(list) + self:close_menu() end ---@param settings HarpoonSettings