tests: add process_spec.lua (#419)

1 files changed, 136 insertions, 0 deletions

diff --git a/tests/core/process_spec.lua b/tests/core/process_spec.lua
new file mode 100644
index 00000000..17ff0822
--- /dev/null
+++ b/tests/core/process_spec.lua
@@ -0,0 +1,136 @@
+local spy = require "luassert.spy"
+local process = require "nvim-lsp-installer.process"
+local async = require "plenary.async"
+
+describe("process.attempt", function()
+    it("should attempt job variants until first successful one", function()
+        local on_finish = spy.new()
+        local on_iterate = spy.new()
+        local success = function(cb)
+            cb(true)
+        end
+        local fail = function(cb)
+            cb(false)
+        end
+
+        local job1 = spy.new(fail)
+        local job2 = spy.new(fail)
+        local job3 = spy.new(success)
+        local job4 = spy.new(success)
+        local job5 = spy.new(fail)
+
+        process.attempt {
+            jobs = { job1, job2, job3, job4, job5 },
+            on_iterate = on_iterate,
+            on_finish = on_finish,
+        }
+
+        assert.spy(job1).was_called(1)
+        assert.spy(job2).was_called(1)
+        assert.spy(job3).was_called(1)
+        assert.spy(job4).was_called(0)
+        assert.spy(job5).was_called(0)
+        assert.spy(on_iterate).was_called(2)
+        assert.spy(on_finish).was_called(1)
+        assert.spy(on_finish).was_called_with(true)
+    end)
+
+    it("should call on_finish with false if all variants fail", function()
+        local on_finish = spy.new()
+        local on_iterate = spy.new()
+        local fail = function(cb)
+            cb(false)
+        end
+
+        local job1 = spy.new(fail)
+        local job2 = spy.new(fail)
+        local job3 = spy.new(fail)
+
+        process.attempt {
+            jobs = { job1, job2, job3 },
+            on_iterate = on_iterate,
+            on_finish = on_finish,
+        }
+
+        assert.spy(job1).was_called(1)
+        assert.spy(job2).was_called(1)
+        assert.spy(job3).was_called(1)
+        assert.spy(on_iterate).was_called(2)
+        assert.spy(on_finish).was_called(1)
+        assert.spy(on_finish).was_called_with(false)
+    end)
+end)
+
+describe("process.spawn", function()
+    -- Unix only
+    async.tests.it("should spawn command and feed output to sink", function()
+        local stdio = process.in_memory_sink()
+        local callback = spy.new()
+        process.spawn("env", {
+            args = {},
+            env = {
+                "HELLO=world",
+                "MY_ENV=var",
+            },
+            stdio_sink = stdio.sink,
+        }, callback)
+
+        assert.wait_for(function()
+            assert.spy(callback).was_called(1)
+            assert.spy(callback).was_called_with(true)
+            assert.equal(table.concat(stdio.buffers.stdout, ""), "HELLO=world\nMY_ENV=var\n")
+        end)
+    end)
+end)
+
+describe("process.chain", function()
+    -- Unix only
+    async.tests.it("should chain commands", function()
+        local stdio = process.in_memory_sink()
+        local callback = spy.new()
+
+        local c = process.chain {
+            env = {
+                "HELLO=world",
+                "MY_ENV=var",
+            },
+            stdio_sink = stdio.sink,
+        }
+
+        c.run("env", {})
+        c.run("bash", { "-c", "echo Hello $HELLO" })
+        c.spawn(callback)
+
+        assert.wait_for(function()
+            assert.spy(callback).was_called(1)
+            assert.spy(callback).was_called_with(true)
+            assert.equal(table.concat(stdio.buffers.stdout, ""), "HELLO=world\nMY_ENV=var\nHello world\n")
+        end)
+    end)
+
+    -- Unix only
+    async.tests.it("should abort chain commands should one fail", function()
+        local stdio = process.in_memory_sink()
+        local callback = spy.new()
+
+        local c = process.chain {
+            env = {
+                "HELLO=world",
+                "MY_ENV=var",
+            },
+            stdio_sink = stdio.sink,
+        }
+
+        c.run("env", {})
+        c.run("bash", { "-c", ">&2 echo Uh oh; exit 1" })
+        c.run("bash", { "-c", "echo Hello $HELLO" })
+        c.spawn(callback)
+
+        assert.wait_for(function()
+            assert.spy(callback).was_called(1)
+            assert.spy(callback).was_called_with(false)
+            assert.equal(table.concat(stdio.buffers.stdout, ""), "HELLO=world\nMY_ENV=var\n")
+            assert.equal(table.concat(stdio.buffers.stderr, ""), "Uh oh\n")
+        end)
+    end)
+end)