capture.rb in daddy-0.3.27

- old
+ new

@@ -1,9 +1,9 @@
 module Daddy
   module Cucumber
     module Capture
-      feature_dir = 'feature'
+      feature_dir = 'features'
       ARGV.each_with_index do |arg, i|
         if arg == '-r'
           feature_dir = ARGV[i + 1]
           break
         end
@@ -14,23 +14,28 @@
     
       @@_screen_count = 0
       @@_images = []
     
       def capture(options = {})
+        options ||= {}
+        options = {:title => options} if options.is_a?(String)
+
         pause
         return if ENV['FORMAT'] == 'junit'
 
         url = Rack::Utils.unescape(current_url)
     
         @@_screen_count += 1
 
         image = "#{IMAGE_DIR}/#{@@_screen_count}.png"
         page.driver.save_screenshot("#{REPORT_DIR}/#{image}", :full => true)
 
+        image_tag = "<img class=\"screenshot\" src=\"#{image}\" title=\"#{options[:title]}\" alt=\"#{url}\"/>"
+
         if options[:flash]
-          puts "<img class=\"screenshot\" src=\"#{image}\" alt=\"#{url}\"/>"
+          puts image_tag
         else
-          @@_images << "<img class=\"screenshot\" src=\"#{image}\" alt=\"#{url}\"/>"
+          @@_images << image_tag
         end
       end
 
       def resize_window(width, height)
         case Capybara.current_driver