refactor: add a convenience wrapper for setting up and tearing down m…

…ocks

lua/wincent/commandt/private/mocks/vim/init.lua

@@ -0,0 +1,31 @@
+-- SPDX-FileCopyrightText: Copyright 2025-present Greg Hurrell and contributors.
+-- SPDX-License-Identifier: BSD-2-Clause
+
+return function(spec)
+  if type(spec) == 'table' then
+    assert(_G.vim == nil)
+    _G.vim = {}
+    for key, value in pairs(spec) do
+      if key == 'inspect' then
+        if value then
+          require('wincent.commandt.private.mocks.vim.inspect').setup()
+        else
+          vim.inspect = nil
+        end
+      elseif key == 'iter' then
+        if value then
+          require('wincent.commandt.private.mocks.vim.iter').setup()
+        else
+          vim.iter = nil
+        end
+      else
+        error('unsupported key: ' .. key)
+      end
+    end
+  elseif spec == false then
+    -- Remove mock.
+    _G.vim = nil
+  else
+    error('unsupported spec: ' .. type(spec))
+  end
+end

lua/wincent/commandt/private/test/validate.lua

@@ -2,16 +2,18 @@
 -- SPDX-License-Identifier: BSD-2-Clause
 
 local validate = require('wincent.commandt.private.validate')
+local mocks = require('wincent.commandt.private.mocks.vim')
 
 describe('validate()', function()
   before(function()
-    assert(_G.vim == nil)
-    require('wincent.commandt.private.mocks.vim.inspect').setup()
-    require('wincent.commandt.private.mocks.vim.iter').setup()
+    mocks({
+      inspect = true,
+      iter = true,
+    })
   end)
 
   after(function()
-    _G.vim = nil
+    mocks(false)
   end)
 
   local spec = {