From aba28d69619285354beb67d11c6d0f5f7ab639f0 Mon Sep 17 00:00:00 2001 From: John Bintz Date: Wed, 25 Jul 2012 14:34:49 -0400 Subject: [PATCH] make sure only valid drivers take screenshots --- lib/cuke-pack/support/screenshots.rb | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) diff --git a/lib/cuke-pack/support/screenshots.rb b/lib/cuke-pack/support/screenshots.rb index af57ef8..2b7d6c7 100644 --- a/lib/cuke-pack/support/screenshots.rb +++ b/lib/cuke-pack/support/screenshots.rb @@ -7,16 +7,19 @@ module CukePack end def take_screenshot(name, options = {}) - options = CukePack.screenshot_options.merge(options) - selenium = Capybara.current_session.driver.browser - selenium.manage.window.resize_to(options[:width], options[:height]) - target = options[:directory] - target = File.join(target, Capybara.current_driver.to_s) - target = File.join(target, name + ".png") + if selenium.respond_to?(:manage) + options = CukePack.screenshot_options.merge(options) - FileUtils.mkdir_p File.dirname(target) + selenium.manage.window.resize_to(options[:width], options[:height]) - selenium.save_screenshot(target) + target = options[:directory] + target = File.join(target, Capybara.current_driver.to_s) + target = File.join(target, name + ".png") + + FileUtils.mkdir_p File.dirname(target) + + selenium.save_screenshot(target) + end end