class Selenium::WebDriver::Script

def add_console_message_handler(&block)
  @log_handler.add_message_handler('console', &block)
end

def add_javascript_error_handler(&block)
  @log_handler.add_message_handler('javascript', &block)
end

def initialize(bridge)
  @log_handler = BiDi::LogHandler.new(bridge.bidi)
end

def remove_console_message_handler(id)
  @log_handler.remove_message_handler(id)
end