lib/rspec.rb in whirlwind-0.1.6 vs lib/rspec.rb in whirlwind-0.1.7

@@ -1,18 +1,25 @@
 module RSpec
   module Core
     module DSL
+
+      def self.set_browser(browser)
+          @browser = browser
+          class <<
+            def self.browser
+              @browser
+            end
+          end
+      end
+
       module_eval do
         BASE = File.expand_path(File.dirname(__FILE__) + '/..')
         define_method('screen_shot') { |i='No Description Provided'|
-          file = "#{BASE}/images/#{Token.provide}.jpg"
+          file = "#{BASE}/images/#{Token.provide}.png"
           FileUtils.mkdir File.dirname(file) unless File.directory?(File.dirname(file))
           @__cucumber_step_mother.add_image(:src => file, :label => i)
-          
-          if RUBY_PLATFORM =~ /win/i or RUBY_PLATFORM =~ /mingw/i
-            Watir::Screen.shot.write(file)
-          else
-            Watir::Screen.shot.resize(800, 600).write(file)
+          File.open(file, 'w+') do |fd|
+            fd.write RSpec::Core::DSL.browser.driver.screenshot_as(:png)
           end
         }
         define_method('site_requires') {
           Find.find(BASE + '/sites').each { |i|
            require i if i =~ /\.rb$/ && !File.directory?(i) && i !~ /\/features\//