cp.nvim/spec/test_render_spec.lua

describe('cp.test_render', function()
  local test_render = require('cp.test_render')

  describe('get_status_info', function()
    it('returns AC for pass status', function()
      local test_case = { status = 'pass' }
      local result = test_render.get_status_info(test_case)
      assert.equals('AC', result.text)
      assert.equals('CpTestAC', result.highlight_group)
    end)

    it('returns WA for fail status with normal exit codes', function()
      local test_case = { status = 'fail', code = 1 }
      local result = test_render.get_status_info(test_case)
      assert.equals('WA', result.text)
      assert.equals('CpTestError', result.highlight_group)
    end)

    it('returns TLE for timeout status', function()
      local test_case = { status = 'timeout' }
      local result = test_render.get_status_info(test_case)
      assert.equals('TLE', result.text)
      assert.equals('CpTestError', result.highlight_group)
    end)

    it('returns TLE for timed out fail status', function()
      local test_case = { status = 'fail', timed_out = true }
      local result = test_render.get_status_info(test_case)
      assert.equals('TLE', result.text)
      assert.equals('CpTestError', result.highlight_group)
    end)

    it('returns RTE for fail with signal codes (>= 128)', function()
      local test_case = { status = 'fail', code = 139 }
      local result = test_render.get_status_info(test_case)
      assert.equals('RTE', result.text)
      assert.equals('CpTestError', result.highlight_group)
    end)

    it('returns empty for pending status', function()
      local test_case = { status = 'pending' }
      local result = test_render.get_status_info(test_case)
      assert.equals('', result.text)
      assert.equals('CpTestPending', result.highlight_group)
    end)

    it('returns running indicator for running status', function()
      local test_case = { status = 'running' }
      local result = test_render.get_status_info(test_case)
      assert.equals('...', result.text)
      assert.equals('CpTestPending', result.highlight_group)
    end)
  end)

  describe('render_test_list', function()
    it('renders test cases with CP terminology', function()
      local test_state = {
        test_cases = {
          { status = 'pass', input = '5' },
          { status = 'fail', code = 1, input = '3' },
        },
        current_index = 1,
      }
      local result = test_render.render_test_list(test_state)
      assert.equals('> 1. AC', result[1])
      assert.equals('  2. WA', result[3])
    end)

    it('shows current test with > prefix', function()
      local test_state = {
        test_cases = {
          { status = 'pass', input = '' },
          { status = 'pass', input = '' },
        },
        current_index = 2,
      }
      local result = test_render.render_test_list(test_state)
      assert.equals('  1. AC', result[1])
      assert.equals('> 2. AC', result[2])
    end)

    it('displays input only for current test', function()
      local test_state = {
        test_cases = {
          { status = 'pass', input = '5 3' },
          { status = 'pass', input = '2 4' },
        },
        current_index = 1,
      }
      local result = test_render.render_test_list(test_state)
      assert.equals('> 1. AC', result[1])
      assert.equals('  5 3', result[2])
      assert.equals('  2. AC', result[3])
    end)

    it('handles empty test cases', function()
      local test_state = { test_cases = {}, current_index = 1 }
      local result = test_render.render_test_list(test_state)
      assert.equals(0, #result)
    end)

    it('preserves input line breaks', function()
      local test_state = {
        test_cases = {
          { status = 'pass', input = '5\n3\n1' },
        },
        current_index = 1,
      }
      local result = test_render.render_test_list(test_state)
      assert.equals('> 1. AC', result[1])
      assert.equals('  5', result[2])
      assert.equals('  3', result[3])
      assert.equals('  1', result[4])
    end)
  end)

  describe('render_status_bar', function()
    it('formats time and exit code', function()
      local test_case = { time_ms = 45.7, code = 0 }
      local result = test_render.render_status_bar(test_case)
      assert.equals('46ms │ Exit: 0', result)
    end)

    it('handles missing time', function()
      local test_case = { code = 0 }
      local result = test_render.render_status_bar(test_case)
      assert.equals('Exit: 0', result)
    end)

    it('handles missing exit code', function()
      local test_case = { time_ms = 123 }
      local result = test_render.render_status_bar(test_case)
      assert.equals('123ms', result)
    end)

    it('returns empty for nil test case', function()
      local result = test_render.render_status_bar(nil)
      assert.equals('', result)
    end)
  end)

  describe('setup_highlights', function()
    it('sets up all highlight groups', function()
      local mock_set_hl = spy.on(vim.api, 'nvim_set_hl')
      test_render.setup_highlights()

      assert.spy(mock_set_hl).was_called(5)
      assert.spy(mock_set_hl).was_called_with(0, 'CpTestAC', { fg = '#10b981', bold = true })
      assert.spy(mock_set_hl).was_called_with(0, 'CpTestError', { fg = '#ef4444', bold = true })

      mock_set_hl:revert()
    end)
  end)
end)