ci: tests

spec/sources_spec.lua

@@ -0,0 +1,186 @@
+vim.opt.runtimepath:prepend(vim.fn.getcwd())
+
+package.preload['fzf-lua.utils'] = function()
+  return {
+    ansi_from_hl = function(_, text)
+      return text
+    end,
+  }
+end
+
+describe('github', function()
+  local gh = require('forge.github')
+
+  it('has correct metadata', function()
+    assert.equals('gh', gh.cli)
+    assert.equals('github', gh.name)
+    assert.equals('pr', gh.kinds.pr)
+    assert.equals('issue', gh.kinds.issue)
+  end)
+
+  it('builds list_pr_json_cmd', function()
+    local cmd = gh:list_pr_json_cmd('open')
+    assert.equals('gh', cmd[1])
+    assert.truthy(vim.tbl_contains(cmd, '--state'))
+    assert.truthy(vim.tbl_contains(cmd, 'open'))
+    assert.truthy(vim.tbl_contains(cmd, '--json'))
+  end)
+
+  it('builds merge_cmd with method flag', function()
+    assert.same({ 'gh', 'pr', 'merge', '42', '--squash' }, gh:merge_cmd('42', 'squash'))
+    assert.same({ 'gh', 'pr', 'merge', '10', '--rebase' }, gh:merge_cmd('10', 'rebase'))
+  end)
+
+  it('builds create_pr_cmd', function()
+    local cmd = gh:create_pr_cmd('title', 'body', 'main', false, nil)
+    assert.truthy(vim.tbl_contains(cmd, '--title'))
+    assert.truthy(vim.tbl_contains(cmd, '--base'))
+    assert.falsy(vim.tbl_contains(cmd, '--draft'))
+  end)
+
+  it('adds draft flag to create_pr_cmd', function()
+    assert.truthy(vim.tbl_contains(gh:create_pr_cmd('t', 'b', 'main', true, nil), '--draft'))
+  end)
+
+  it('adds reviewers to create_pr_cmd', function()
+    local cmd = gh:create_pr_cmd('t', 'b', 'main', false, { 'alice', 'bob' })
+    local count = 0
+    for _, v in ipairs(cmd) do
+      if v == '--reviewer' then
+        count = count + 1
+      end
+    end
+    assert.equals(2, count)
+  end)
+
+  it('builds checkout_cmd', function()
+    assert.same({ 'gh', 'pr', 'checkout', '5' }, gh:checkout_cmd('5'))
+  end)
+
+  it('builds close/reopen commands', function()
+    assert.same({ 'gh', 'pr', 'close', '3' }, gh:close_cmd('3'))
+    assert.same({ 'gh', 'pr', 'reopen', '3' }, gh:reopen_cmd('3'))
+  end)
+
+  it('returns correct pr_json_fields', function()
+    local f = gh:pr_json_fields()
+    assert.equals('number', f.number)
+    assert.equals('headRefName', f.branch)
+    assert.equals('createdAt', f.created_at)
+  end)
+
+  it('normalizes completed run to conclusion', function()
+    local run = gh:normalize_run({
+      databaseId = 123,
+      name = 'CI',
+      headBranch = 'main',
+      status = 'completed',
+      conclusion = 'success',
+      event = 'push',
+      url = 'https://example.com',
+      createdAt = '2025-01-01T00:00:00Z',
+    })
+    assert.equals('123', run.id)
+    assert.equals('success', run.status)
+    assert.equals('main', run.branch)
+  end)
+
+  it('preserves in_progress status in normalize_run', function()
+    assert.equals('in_progress', gh:normalize_run({ databaseId = 1, status = 'in_progress' }).status)
+  end)
+end)
+
+describe('gitlab', function()
+  local gl = require('forge.gitlab')
+
+  it('has correct metadata', function()
+    assert.equals('glab', gl.cli)
+    assert.equals('gitlab', gl.name)
+    assert.equals('mr', gl.kinds.pr)
+  end)
+
+  it('builds list_pr_json_cmd with state variants', function()
+    local cmd = gl:list_pr_json_cmd('open')
+    assert.equals('glab', cmd[1])
+    assert.equals('mr', cmd[2])
+    assert.truthy(vim.tbl_contains(gl:list_pr_json_cmd('closed'), '--closed'))
+    assert.truthy(vim.tbl_contains(gl:list_pr_json_cmd('all'), '--all'))
+  end)
+
+  it('builds merge_cmd with method flags', function()
+    assert.same({ 'glab', 'mr', 'merge', '5', '--squash' }, gl:merge_cmd('5', 'squash'))
+    assert.same({ 'glab', 'mr', 'merge', '5', '--rebase' }, gl:merge_cmd('5', 'rebase'))
+    assert.same({ 'glab', 'mr', 'merge', '5' }, gl:merge_cmd('5', 'merge'))
+  end)
+
+  it('builds create_pr_cmd with --description and --target-branch', function()
+    local cmd = gl:create_pr_cmd('title', 'desc', 'develop', false, nil)
+    assert.truthy(vim.tbl_contains(cmd, '--description'))
+    assert.truthy(vim.tbl_contains(cmd, '--target-branch'))
+    assert.truthy(vim.tbl_contains(cmd, '--yes'))
+  end)
+
+  it('returns correct pr_json_fields', function()
+    local f = gl:pr_json_fields()
+    assert.equals('iid', f.number)
+    assert.equals('source_branch', f.branch)
+    assert.equals('created_at', f.created_at)
+  end)
+
+  it('extracts MR number from ref in normalize_run', function()
+    local run = gl:normalize_run({
+      id = 456,
+      ref = 'refs/merge-requests/10/head',
+      status = 'success',
+      source = 'push',
+      web_url = 'https://example.com',
+      created_at = '2025-01-01T00:00:00Z',
+    })
+    assert.equals('456', run.id)
+    assert.equals('!10', run.name)
+  end)
+
+  it('uses ref as name for non-MR refs', function()
+    assert.equals('main', gl:normalize_run({ id = 1, ref = 'main', status = 'running' }).name)
+  end)
+end)
+
+describe('codeberg', function()
+  local cb = require('forge.codeberg')
+
+  it('has correct metadata', function()
+    assert.equals('tea', cb.cli)
+    assert.equals('codeberg', cb.name)
+    assert.equals('pulls', cb.kinds.pr)
+    assert.equals('issues', cb.kinds.issue)
+  end)
+
+  it('builds list_pr_json_cmd with --fields', function()
+    local cmd = cb:list_pr_json_cmd('open')
+    assert.equals('tea', cmd[1])
+    assert.truthy(vim.tbl_contains(cmd, '--fields'))
+  end)
+
+  it('builds merge_cmd with --style', function()
+    assert.same({ 'tea', 'pr', 'merge', '7', '--style', 'squash' }, cb:merge_cmd('7', 'squash'))
+  end)
+
+  it('ignores draft and reviewers in create_pr_cmd', function()
+    local cmd = cb:create_pr_cmd('title', 'body', 'main', true, { 'alice' })
+    assert.falsy(vim.tbl_contains(cmd, '--draft'))
+    assert.falsy(vim.tbl_contains(cmd, '--reviewer'))
+    assert.truthy(vim.tbl_contains(cmd, '--base'))
+  end)
+
+  it('returns correct pr_json_fields', function()
+    local f = cb:pr_json_fields()
+    assert.equals('index', f.number)
+    assert.equals('head', f.branch)
+    assert.equals('poster', f.author)
+  end)
+
+  it('returns nil from draft_toggle_cmd', function()
+    assert.is_nil(cb:draft_toggle_cmd('1', true))
+    assert.is_nil(cb:draft_toggle_cmd('1', false))
+  end)
+end)