1 files changed, 40 insertions, 5 deletions
diff --git a/lua/mason-test/helpers.lua b/lua/mason-test/helpers.lua
index 2348e9df..88354046 100644
--- a/lua/mason-test/helpers.lua
+++ b/lua/mason-test/helpers.lua
@@ -1,7 +1,8 @@
 local InstallContext = require "mason-core.installer.context"
-local InstallHandle = require "mason-core.installer.handle"
-local InstallLocation = require "mason-core.installer.location"
+local InstallHandle = require "mason-core.installer.InstallHandle"
+local InstallLocation = require "mason-core.installer.InstallLocation"
 local Result = require "mason-core.result"
+local a = require "mason-core.async"
 local registry = require "mason-registry"
 local spy = require "luassert.spy"
 
@@ -10,9 +11,8 @@ local M = {}
 ---@param opts? { install_opts?: PackageInstallOpts, package?: string }
 function M.create_context(opts)
     local pkg = registry.get_package(opts and opts.package or "dummy")
-    local handle = InstallHandle:new(pkg)
-    local location = InstallLocation.global()
-    local context = InstallContext:new(handle, location, opts and opts.install_opts or {})
+    local handle = InstallHandle:new(pkg, InstallLocation.global())
+    local context = InstallContext:new(handle, opts and opts.install_opts or {})
     context.spawn = setmetatable({}, {
         __index = function(s, cmd)
             s[cmd] = spy.new(function()
@@ -25,4 +25,39 @@ function M.create_context(opts)
     return context
 end
 
+---@param pkg AbstractPackage
+---@param opts? PackageInstallOpts
+function M.sync_install(pkg, opts)
+    return a.run_blocking(function()
+        return a.wait(function(resolve, reject)
+            pkg:install(opts, function(success, result)
+                (success and resolve or reject)(result)
+            end)
+        end)
+    end)
+end
+
+---@param pkg AbstractPackage
+---@param opts? PackageUninstallOpts
+function M.sync_uninstall(pkg, opts)
+    return a.run_blocking(function()
+        return a.wait(function(resolve, reject)
+            pkg:uninstall(opts, function(success, result)
+                (success and resolve or reject)(result)
+            end)
+        end)
+    end)
+end
+
+---@param runner InstallRunner
+---@param opts PackageInstallOpts
+function M.sync_runner_execute(runner, opts)
+    local callback = spy.new()
+    runner:execute(opts, callback)
+    assert.wait(function()
+        assert.spy(callback).was_called()
+    end)
+    return callback
+end
+
 return M