test(nut): refactor tests for git.status

MunifTanjim · Jan 4, 2024 · aa3dbf2 · aa3dbf2
1 parent 1a034be
commit aa3dbf2
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/tests/nougat/nut/git/status_spec.lua b/tests/nougat/nut/git/status_spec.lua
@@ -28,9 +28,15 @@ describe("nut.git.status", function()
   end)
 
   describe("w/ gitsigns", function()
-    it("works", function()
+    before_each(function()
       package.loaded["gitsigns"] = {}
+    end)
+
+    after_each(function()
+      package.loaded["gitsigns"] = nil
+    end)
 
+    it("works", function()
       bar:add_item(nut.create({
         prefix = " ",
         content = {
@@ -40,8 +46,6 @@ describe("nut.git.status", function()
         },
       }))
 
-      package.loaded["gitsigns"] = nil
-
       vim.b[ctx.bufnr].gitsigns_status_dict = {
         added = 1,
       }
@@ -62,17 +66,13 @@ describe("nut.git.status", function()
     end)
 
     it("can handle missing gitstatus", function()
-      package.loaded["gitsigns"] = {}
-
       bar:add_item(nut.create({
         prefix = " ",
         content = {
           nut.count("added", { prefix = "+", suffix = " " }),
         },
       }))
 
-      package.loaded["gitsigns"] = nil
-
       vim.b[ctx.bufnr].gitsigns_status_dict = nil
       vim.api.nvim_exec_autocmds("User", { pattern = "GitSignsUpdate" })
       vim.wait(0)