3scale/porta

# frozen_string_literal: true

module TableHelpers

  # Does the same thing as #tableish from cucumber-rails.
  #
  # +table+ is a table selector
  # +rows+ is rows selector
  # +cells+ is cells selector
  #
  # returns array with table content
  #
  def extract_table(table, rows, cells = nil)
    table = find(*table)
    ActiveSupport::Deprecation.warn '[Cucumber] Table not implemented with Patternfly' unless table[:class].include?('pf-c-table')
    table.all(*rows).map do |row|
      if cells.respond_to?(:call)
        cells.call(row)
      elsif block_given?
        yield row
      else
        row.all(*cells).map { |cell| cell.text.strip }
      end
    end
  end

  def extract_pf4_table
    header = all('.pf-c-table thead th').map(&:text)

    body = all('.pf-c-table tbody tr').map do |row|
      row.all('td:not(.pf-c-table__action)').map(&:text)
    end

    [header] + body
  end

  def assert_plans_table(plans, headers: false, sort: false)
    table = extract_pf4_table
    table.shift unless headers

    rows = plans.pluck(:name, :contracts_count, :state).map { |r| r.map(&:to_s) }
    if sort
      assert_equal rows, table
    else
      assert_same_elements rows, table
    end
  end
end

World(TableHelpers)