Capybara convenience module for finding elements by `data-X` values

dom_helper.rb

# frozen_string_literal: true
module DomHelper
  # Executes the given block within the context of a specific node identified
  # using its data-xxx attributes. It will take any number of arguments.
  #
  # eg:
  #   user = create_user()
  #   within_element(id: user.id) do
  #     click_on 'Edit'
  #   end
  #
  def within_element(**opts)
    within(:xpath, "//*[#{data_attrs(opts)}]") do
      yield
    end
  end
  
  private

  def data_attrs(opts)
    opts
      .map{|k,v| "@data-#{k}=\"#{v}\"" }
      .join(' and ')
  end
end