add TSamFailedOnly command and change test output within the floating window

tests/test_samurai_failed_only_spec.lua

@@ -0,0 +1,225 @@
+local test_samurai = require("test-samurai")
+local core = require("test-samurai.core")
+
+local function mkbuf(path, ft, lines)
+  local bufnr = vim.api.nvim_create_buf(false, true)
+  vim.api.nvim_buf_set_name(bufnr, path)
+  vim.bo[bufnr].filetype = ft
+  if lines then
+    vim.api.nvim_buf_set_lines(bufnr, 0, -1, false, lines)
+  end
+  return bufnr
+end
+
+local function stub_jobstart(opts_config)
+  local calls = {}
+  local orig = vim.fn.jobstart
+  local idx = 0
+  local config = opts_config or {}
+  vim.fn.jobstart = function(cmd, opts)
+    idx = idx + 1
+    table.insert(calls, { cmd = cmd, opts = opts })
+    local code = 0
+    if type(config.exit_codes) == "table" then
+      code = config.exit_codes[idx] or 0
+    elseif type(config.exit_codes) == "number" then
+      code = config.exit_codes
+    end
+    local out = config.stdout and config.stdout[idx] or nil
+    if out and opts and opts.on_stdout then
+      if type(out) == "string" then
+        out = { out }
+      end
+      opts.on_stdout(1, out, nil)
+    end
+    local err = config.stderr and config.stderr[idx] or nil
+    if err and opts and opts.on_stderr then
+      if type(err) == "string" then
+        err = { err }
+      end
+      opts.on_stderr(1, err, nil)
+    end
+    if opts and opts.on_exit then
+      opts.on_exit(1, code, nil)
+    end
+    return 1
+  end
+  return calls, orig
+end
+
+describe("TSamFailedOnly", function()
+  before_each(function()
+    test_samurai.setup()
+  end)
+
+  it("reruns failed jest tests with --onlyFailures", function()
+    local json = vim.json.encode({
+      testResults = {
+        {
+          assertionResults = {
+            { status = "passed", title = "inner 1", fullName = "outer inner 1" },
+            { status = "failed", title = "inner 2", fullName = "outer inner 2" },
+          },
+        },
+      },
+    })
+
+    local calls, orig_jobstart = stub_jobstart({
+      exit_codes = { 1, 0 },
+      stdout = { { json } },
+    })
+
+    local bufnr = mkbuf("/tmp/project/foo_failed_only.test.ts", "typescript", {
+      'describe("outer", function() {',
+      '  it("inner 1", function() {',
+      "    -- inside 1",
+      "  })",
+      "",
+      '  it("inner 2", function() {',
+      "    -- inside 2",
+      "  })",
+      "})",
+    })
+
+    vim.api.nvim_set_current_buf(bufnr)
+    vim.api.nvim_win_set_cursor(0, { 7, 0 })
+
+    core.run_nearest()
+    core.run_failed_only()
+
+    vim.fn.jobstart = orig_jobstart
+
+    assert.equals(2, #calls)
+    assert.are.same(
+      { "npx", "jest", "--json", "/tmp/project/foo_failed_only.test.ts", "-t", "inner 2" },
+      calls[1].cmd
+    )
+    assert.are.same(
+      { "npx", "jest", "--json", "-t", "outer inner 2", "/tmp/project/foo_failed_only.test.ts" },
+      calls[2].cmd
+    )
+  end)
+
+  it("falls back to TSamLast when last run had no failures", function()
+    local json = vim.json.encode({
+      testResults = {
+        {
+          assertionResults = {
+            { status = "passed", title = "inner 1", fullName = "outer inner 1" },
+          },
+        },
+      },
+    })
+
+    local calls, orig_jobstart = stub_jobstart({
+      exit_codes = { 0, 0 },
+      stdout = { { json } },
+    })
+
+    local bufnr = mkbuf("/tmp/project/foo_failed_only_pass.test.ts", "typescript", {
+      'describe("outer", function() {',
+      '  it("inner 1", function() {',
+      "    -- inside 1",
+      "  })",
+      "",
+      '  it("inner 2", function() {',
+      "    -- inside 2",
+      "  })",
+      "})",
+    })
+
+    vim.api.nvim_set_current_buf(bufnr)
+    vim.api.nvim_win_set_cursor(0, { 7, 0 })
+
+    core.run_nearest()
+    core.run_failed_only()
+
+    vim.fn.jobstart = orig_jobstart
+
+    assert.equals(2, #calls)
+    assert.are.same(calls[1].cmd, calls[2].cmd)
+  end)
+
+  it("reruns failed go tests with -run regex", function()
+    local json_lines = {
+      vim.json.encode({ Action = "fail", Test = "TestFoo/first" }),
+      vim.json.encode({ Action = "fail", Test = "TestBar" }),
+    }
+
+    local calls, orig_jobstart = stub_jobstart({
+      exit_codes = { 1, 0 },
+      stdout = { json_lines },
+    })
+
+    local bufnr = mkbuf("/tmp/project/foo_failed_only_test.go", "go", {
+      "package main",
+      "import \"testing\"",
+      "",
+      "func TestFoo(t *testing.T) {",
+      "  t.Run(\"first\", func(t *testing.T) {",
+      "    -- inside first",
+      "  })",
+      "}",
+      "",
+      "func TestBar(t *testing.T) {",
+      "  -- inside bar",
+      "}",
+    })
+
+    vim.api.nvim_set_current_buf(bufnr)
+
+    core.run_all()
+    core.run_failed_only()
+
+    vim.fn.jobstart = orig_jobstart
+
+    assert.equals(2, #calls)
+    assert.are.same({ "go", "test", "-json", "./..." }, calls[1].cmd)
+    assert.are.same({ "go", "test", "-json", "./...", "-run", "^(TestFoo/first|TestBar)$" }, calls[2].cmd)
+  end)
+
+  it("does not affect TSamLast history", function()
+    local json = vim.json.encode({
+      testResults = {
+        {
+          assertionResults = {
+            { status = "failed", title = "inner 2", fullName = "outer inner 2" },
+          },
+        },
+      },
+    })
+
+    local calls, orig_jobstart = stub_jobstart({
+      exit_codes = { 1, 1, 1 },
+      stdout = { { json } },
+    })
+
+    local bufnr = mkbuf("/tmp/project/foo_failed_only_last.test.ts", "typescript", {
+      'describe("outer", function() {',
+      '  it("inner 1", function() {',
+      "    -- inside 1",
+      "  })",
+      "",
+      '  it("inner 2", function() {',
+      "    -- inside 2",
+      "  })",
+      "})",
+    })
+
+    vim.api.nvim_set_current_buf(bufnr)
+    vim.api.nvim_win_set_cursor(0, { 7, 0 })
+
+    core.run_nearest()
+    core.run_failed_only()
+    core.run_last()
+
+    vim.fn.jobstart = orig_jobstart
+
+    assert.equals(3, #calls)
+    assert.are.same(calls[1].cmd, calls[3].cmd)
+    assert.are.same(
+      { "npx", "jest", "--json", "-t", "outer inner 2", "/tmp/project/foo_failed_only_last.test.ts" },
+      calls[2].cmd
+    )
+  end)
+end)