ci: better wait condition for screenshots

Author: ibhagwan

lua/fzf-lua/test/helpers.lua

@@ -133,7 +133,10 @@ M.new_child_neovim = function()
         %s
         winopts = {
           on_create = function() _G._fzf_lua_on_create = true end,
-          on_close = function() _G._fzf_lua_on_create = nil end,
+          on_close = function()
+            _G._fzf_lua_on_create = nil
+            _G._fzf_postprocess_called = nil
+          end,
         },
         keymap = { fzf = {
           true,

tests/file/ui_spec.lua

@@ -24,16 +24,26 @@ T["files()"] = new_set()
 T["files()"]["start and abort"] = new_set({ parametrize = { { "<esc>" }, { "<c-c>" } } }, {
   function(key)
     -- sort output and remove cwd in prompt as will be different on CI
-    child.lua([[FzfLua.files({
+    eq(child.lua_get([[_G._fzf_postprocess_called]]), vim.NIL)
+    child.lua([==[FzfLua.files({
       hidden = false,
       previewer = false,
       cwd_prompt = false,
       cmd = "rg --files --sort=path",
-    })]])
+      requires_processing = true, -- for __mt_postprocess
+      __mt_postprocess = [[return function()
+        local chan_id = vim.fn.sockconnect("pipe", _G._fzf_lua_server, { rpc = true })
+        vim.rpcrequest(chan_id, "nvim_exec_lua", "_G._fzf_postprocess_called=true", {})
+        vim.fn.chanclose(chan_id)
+      end]],
+    })]==])
     eq(child.lua_get([[_G._fzf_lua_on_create]]), true)
     child.wait_until(function()
       return child.lua_get([[_G._fzf_load_called]]) == true
     end)
+    child.wait_until(function()
+      return child.lua_get([[_G._fzf_postprocess_called]]) == true
+    end)
     -- Ignore last "-- TERMINAL --" line and paths on Windows (separator is "\")
     local screen_opts = { ignore_lines = { 28 }, normalize_paths = helpers.IS_WIN() }
     -- NOTE: we compare screen lines without "attrs"