add some tests (#360)

1 files changed, 4 insertions, 4 deletions

diff --git a/lua/nvim-lsp-installer/dispatcher.lua b/lua/nvim-lsp-installer/dispatcher.lua
index 0ad41897..762da3e3 100644
--- a/lua/nvim-lsp-installer/dispatcher.lua
+++ b/lua/nvim-lsp-installer/dispatcher.lua
@@ -4,6 +4,7 @@ local M = {}
 
 local registered_callbacks = {}
 
+---@param server Server
 M.dispatch_server_ready = function(server)
     for _, callback in pairs(registered_callbacks) do
         local ok, err = pcall(callback, server)
@@ -13,12 +14,11 @@ M.dispatch_server_ready = function(server)
     end
 end
 
-local idx = 0
+---@param callback fun(server: Server)
 function M.register_server_ready_callback(callback)
-    local key = idx + 1
-    registered_callbacks[("%d"):format(key)] = callback
+    registered_callbacks[callback] = callback
     return function()
-        table.remove(registered_callbacks, key)
+        registered_callbacks[callback] = nil
     end
 end