Take a screenshot after a failure in Ruby, using Capybara

screenshot_after_failure.rb

After do |scenario|
  diff = (Time.new - time)
  timings << { scenario: scenario.name, timing: diff }

  if scenario.failed?
    begin
      Capybara.using_session(Capybara::Screenshot.final_session_name) do
       filename_prefix = Capybara::Screenshot.filename_prefix_for(:cucumber, scenario)

       saver = Capybara::Screenshot::Saver.new(Capybara, Capybara.page, true, filename_prefix)
       saver.save
       saver.output_screenshot_path

       # Trying to embed the screenshot into our output."
       if File.exist?(saver.screenshot_path)
         require "base64"
         #encode the image into it's base64 representation
         image = open(saver.screenshot_path, 'rb') {|io|io.read}
         encoded_img = Base64.encode64(image)
         #this will embed the image in the HTML report, embed() is defined in cucumber
         embed(encoded_img, 'image/png;base64', "Screenshot of the error")
       end
     end
    rescue Exception => e
      puts "Failed to save screenshot"
      puts "#{e.message}\n #{e.backtrace.join("\n ")}"
    end
  end

  if ($config.get('FAIL_ON_FIRST_ERROR') == 'true')
    Cucumber.wants_to_quit = true if scenario.failed?
  end

  unless $config.get('TEST_MODE') == 'acceptance'
    CookieRemover.delete_all($config.get('LOGIN_URL'))
  end

end