module Selenium::WebDriver::DriverExtensions::TakesScreenshot

Public Instance Methods

save_screenshot(png_path) click to toggle source

Save a PNG screenshot to the given path

@api public

# File lib/selenium/webdriver/common/driver_extensions/takes_screenshot.rb, line 33
def save_screenshot(png_path)
  extension = File.extname(png_path).downcase
  if extension != '.png'
    WebDriver.logger.warn "name used for saved screenshot does not match file type. "\
                          "It should end with .png extension"
  end
  File.open(png_path, 'wb') { |f| f << screenshot_as(:png) }
end
screenshot_as(format) click to toggle source

Return a PNG screenshot in the given format as a string

@param [:base64, :png] format @return String screenshot

@api public

# File lib/selenium/webdriver/common/driver_extensions/takes_screenshot.rb, line 50
def screenshot_as(format)
  case format
  when :base64
    bridge.screenshot
  when :png
    bridge.screenshot.unpack('m')[0]
  else
    raise Error::UnsupportedOperationError, "unsupported format: #{format.inspect}"
  end
end