1 files changed, 71 insertions, 0 deletions
diff --git a/lua/99/test/in_flight_spec.lua b/lua/99/test/in_flight_spec.lua
new file mode 100644
index 0000000..c9ebc6b
--- /dev/null
+++ b/lua/99/test/in_flight_spec.lua
@@ -0,0 +1,71 @@
+-- luacheck: globals describe it assert after_each
+local _99 = require("99")
+local Prompt = require("99.prompt")
+local Window = require("99.window")
+local test_utils = require("99.test.test_utils")
+local eq = assert.are.same
+
+local content = {
+  "local function foo()",
+  "  return 1",
+  "end",
+}
+
+--- You have to override this or else things will crash since the ui itself
+--- does not exist.  this is a headless test so i fake it by returning a very
+--- simple ui of 120x40
+local original_nvim_list_uis = vim.api.nvim_list_uis
+local function nvim_list_uis()
+  return {
+    { width = 120, height = 40 },
+  }
+end
+
+describe("in_flight window", function()
+  local WAIT_TIME = 10
+  before_each(function()
+    vim.api.nvim_list_uis = nvim_list_uis
+  end)
+  after_each(function()
+    vim.api.nvim_list_uis = original_nvim_list_uis
+  end)
+  it("shows active requests and clears when request completes", function()
+    local provider = test_utils.test_setup(content, 2, 4)
+    local state = _99.__get_state()
+    local context = Prompt.search(state)
+
+    context:start_request()
+    vim.wait(WAIT_TIME * 2, function() end)
+
+    eq(1, #Window.active_windows)
+
+    local win = Window.active_windows[1]
+    vim.api.nvim_win_close(win.win_id, true)
+
+    vim.wait(WAIT_TIME * 2, function() end)
+    local next_win = Window.active_windows[1]
+
+    eq(true, win.win_id ~= next_win.win_id)
+
+    provider:resolve("success", "results are in")
+
+    vim.wait(WAIT_TIME * 2, function() end)
+    eq(0, #Window.active_windows)
+  end)
+
+  it("enable false == do not show in flight", function()
+    local provider = test_utils.test_setup(content, 2, 4, "lua", {
+      in_flight_options = { enable = false },
+    })
+    local state = _99.__get_state()
+    local context = Prompt.search(state)
+
+    context:start_request()
+    vim.wait(WAIT_TIME * 2, function() end)
+
+    eq(0, #Window.active_windows)
+    provider:resolve("success", "results are in")
+    vim.wait(WAIT_TIME * 2, function() end)
+    eq(0, #Window.active_windows)
+  end)
+end)