lualine.nvim/lua/tests/helpers.lua

local eq = assert.are.same

local M = {}

M.meths = setmetatable({}, {
  __index = function(_, key)
    return vim.api['nvim_' .. key]
  end,
})

-- Checks ouput of a component
M.assert_component = function(component, opts, result)
  -- for testing global options
  if component == nil then
    component = 'special.function_component'
  end
  local comp = require('lualine.components.' .. component):new(opts)
  eq(result, comp:draw(opts.hl))
end

-- sets defaults for component options
M.build_component_opts = function(opts)
  if not opts then
    opts = {}
  end
  if opts[1] == nil then
    opts[1] = function()
      return 'test'
    end
  end
  if not opts.self then
    opts.self = { section = 'lualine_c' }
  end
  if not opts.theme then
    opts.theme = 'gruvbox'
  end
  if not opts.hl then
    opts.hl = ''
  end
  if opts.icons_enabled == nil then
    opts.icons_enabled = true
  end
  if not opts.component_separators then
    opts.component_separators = { '', '' }
  end
  if not opts.section_separators then
    opts.section_separators = { '', '' }
  end
  return opts
end

M.P = function(t)
  print(vim.inspect(t))
end

return M