lib/janus/command/record.rb in janus-cli-0.1.0 vs lib/janus/command/record.rb in janus-cli-0.2.0

@@ -1,22 +1,53 @@
+# coding: utf-8
+
+require 'colorize'
+require 'sauce/connect'
+require 'janus/io/directory'
+require 'janus/io/selenium'
 require 'janus/screenshot'
 
 module Janus
   module Command
     class Record
       def initialize(configuration)
         @configuration = configuration
       end
 
       def execute
+        Sauce::Connect.connect!(quiet: true) if @configuration.tunnel?
+
+        puts 'Recording screenshots...'
+        puts ''
+
+        @configuration.browsers.each do |browser|
+          record_screenshots_for_browser(browser)
+        end
+      end
+
+      def record_screenshots_for_browser(browser)
+        puts "#{browser}"
+        puts ''
+
         @configuration.tests.each do |test|
-          record_screenshot(test)
+          record_screenshot(browser, test)
         end
+
+        puts ''
       end
 
-      def record_screenshot(test)
-        screenshot = Janus::Screenshot.capture(test, username: @configuration.username, access_key: @configuration.access_key)
-        screenshot.save('output')
+      def record_screenshot(browser, test)
+        selenium = Janus::IO::Selenium.new(@configuration.username, @configuration.access_key, browser)
+        screenshot = selenium.read(test)
+
+        directory = Janus::IO::Directory.new(@configuration.directory, browser)
+        directory.write(test, screenshot)
+
+        print '✔ '.green
+      rescue
+        print '✖ '.red
+      ensure
+        puts test.name
       end
     end
   end
 end