module Rails::Dom::Testing

def choose_html_parser(parser_classes, html_version: nil)

def choose_html_parser(parser_classes, html_version: nil)
  html_version ||= Rails::Dom::Testing.default_html_version
  case html_version
  when :html4
    parser_classes[:html4]
  when :html5
    unless Rails::Dom::Testing.html5_support?
      raise NotImplementedError, "html5 parser is not supported on this platform"
    end
    parser_classes[:html5]
  else
    raise ArgumentError, "html_version must be :html4 or :html5, received #{html_version.inspect}"
  end
end