Embedded png screenshots in your Cucumber HTML output

gistfile1.rb

After do |scenario|
  if scenario.failed?
    embed_screenshot
  end
  @browser.close
end

private

def embed_screenshot
  encoded_img = @browser.driver.screenshot_as(:base64)
  embed("data:image/png;base64,#{encoded_img}", 'image/png')
end